use cuda_types::{CUGLDeviceList, CUdevice}; use std::{ ffi::{c_void, CStr}, fmt::LowerHex, mem, ptr, slice, }; pub(crate) trait CudaDisplay { fn write( &self, fn_name: &'static str, index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()>; } impl CudaDisplay for cuda_types::CUuuid { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { let guid = self.bytes; write!(writer, "{{{:02X}{:02X}{:02X}{:02X}-{:02X}{:02X}-{:02X}{:02X}-{:02X}{:02X}-{:02X}{:02X}{:02X}{:02X}{:02X}{:02X}}}", guid[0], guid[1], guid[2], guid[3], guid[4], guid[5], guid[6], guid[7], guid[8], guid[9], guid[10], guid[11], guid[12], guid[13], guid[14], guid[15]) } } impl CudaDisplay for cuda_types::CUdevice { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{}", self.0) } } impl CudaDisplay for cuda_types::CUdeviceptr { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{:p}", self.0) } } impl CudaDisplay for cuda_types::CUdeviceptr_v1 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{:p}", self.0 as usize as *const ()) } } impl CudaDisplay for u8 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{}", *self) } } impl CudaDisplay for u16 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{}", *self) } } impl CudaDisplay for i32 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{}", *self) } } impl CudaDisplay for u32 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{}", *self) } } impl CudaDisplay for u64 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{}", *self) } } impl CudaDisplay for usize { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{}", *self) } } impl CudaDisplay for f32 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{}", *self) } } pub fn write_handle( this: &[T; 64], writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { writer.write_all(b"0x")?; for i in (0..64).rev() { write!(writer, "{:02x}", this[i])?; } Ok(()) } impl CudaDisplay for cuda_types::CUipcMemHandle { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write_handle(&self.reserved, writer) } } impl CudaDisplay for cuda_types::CUipcEventHandle { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write_handle(&self.reserved, writer) } } impl CudaDisplay for cuda_types::CUmemPoolPtrExportData_v1 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write_handle(&self.reserved, writer) } } impl CudaDisplay for *mut c_void { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{:p}", *self) } } impl CudaDisplay for *const c_void { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!(writer, "{:p}", *self) } } impl CudaDisplay for *const i8 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { write!( writer, "\"{}\"", unsafe { CStr::from_ptr(*self as _) }.to_string_lossy() ) } } #[repr(C)] #[derive(Copy, Clone)] struct Luid { low_part: u32, high_part: u32, } impl CudaDisplay for *mut i8 { fn write( &self, fn_name: &'static str, index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { if fn_name == "cuDeviceGetLuid" && index == 0 { let luid_ptr = *self as *mut Luid; let luid = unsafe { *luid_ptr }; write!(writer, "{{{:08X}-{:08X}}}", luid.low_part, luid.high_part) } else { write!( writer, "\"{}\"", unsafe { CStr::from_ptr(*self as _) }.to_string_lossy() ) } } } impl CudaDisplay for cuda_types::CUstreamBatchMemOpParams { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { unsafe { match self.operation { // The below is not a typo, `WAIT_VALUE` and `WRITE_VALUE` are // distinct operations with nominally distinct union variants, but // in reality they are structurally different, so we take a little // shortcut here cuda_types::CUstreamBatchMemOpType::CU_STREAM_MEM_OP_WAIT_VALUE_32 | cuda_types::CUstreamBatchMemOpType::CU_STREAM_MEM_OP_WRITE_VALUE_32 => { write_wait_value(&self.waitValue, writer, false) } cuda_types::CUstreamBatchMemOpType::CU_STREAM_MEM_OP_WAIT_VALUE_64 | cuda_types::CUstreamBatchMemOpType::CU_STREAM_MEM_OP_WRITE_VALUE_64 => { write_wait_value(&self.waitValue, writer, true) } cuda_types::CUstreamBatchMemOpType::CU_STREAM_MEM_OP_FLUSH_REMOTE_WRITES => { CudaDisplay::write(&self.flushRemoteWrites, "", 0, writer) } _ => { writer.write_all(b"{ operation: ")?; CudaDisplay::write(&self.operation, "", 0, writer)?; writer.write_all(b", ... }") } } } } } pub fn write_wait_value( this: &cuda_types::CUstreamBatchMemOpParams_union_CUstreamMemOpWaitValueParams_st, writer: &mut (impl std::io::Write + ?Sized), is_64_bit: bool, ) -> std::io::Result<()> { writer.write_all(b"{ operation: ")?; CudaDisplay::write(&this.operation, "", 0, writer)?; writer.write_all(b", address: ")?; CudaDisplay::write(&this.address, "", 0, writer)?; write_wait_value_32_or_64(&this.__bindgen_anon_1, writer, is_64_bit)?; writer.write_all(b", flags: ")?; CudaDisplay::write(&this.flags, "", 0, writer)?; writer.write_all(b", alias: ")?; CudaDisplay::write(&this.alias, "", 0, writer)?; writer.write_all(b" }") } pub fn write_wait_value_32_or_64( this: &cuda_types::CUstreamBatchMemOpParams_union_CUstreamMemOpWaitValueParams_st__bindgen_ty_1, writer: &mut (impl std::io::Write + ?Sized), is_64_bit: bool, ) -> std::io::Result<()> { if is_64_bit { writer.write_all(b", value64: ")?; CudaDisplay::write(unsafe { &this.value64 }, "", 0, writer) } else { writer.write_all(b", value: ")?; CudaDisplay::write(unsafe { &this.value }, "", 0, writer) } } impl CudaDisplay for cuda_types::CUDA_RESOURCE_DESC_st { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { writer.write_all(b"{ resType: ")?; CudaDisplay::write(&self.resType, "", 0, writer)?; match self.resType { cuda_types::CUresourcetype::CU_RESOURCE_TYPE_ARRAY => { writer.write_all(b", res: ")?; CudaDisplay::write(unsafe { &self.res.array }, "", 0, writer)?; writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0, writer)?; writer.write_all(b" }") } cuda_types::CUresourcetype::CU_RESOURCE_TYPE_MIPMAPPED_ARRAY => { writer.write_all(b", res: ")?; CudaDisplay::write(unsafe { &self.res.mipmap }, "", 0, writer)?; writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0, writer)?; writer.write_all(b" }") } cuda_types::CUresourcetype::CU_RESOURCE_TYPE_LINEAR => { writer.write_all(b", res: ")?; CudaDisplay::write(unsafe { &self.res.linear }, "", 0, writer)?; writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0, writer)?; writer.write_all(b" }") } cuda_types::CUresourcetype::CU_RESOURCE_TYPE_PITCH2D => { writer.write_all(b", res: ")?; CudaDisplay::write(unsafe { &self.res.pitch2D }, "", 0, writer)?; writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0, writer)?; writer.write_all(b" }") } _ => { writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0, writer)?; writer.write_all(b", ... }") } } } } impl CudaDisplay for cuda_types::CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { writer.write_all(b"{ type: ")?; CudaDisplay::write(&self.type_, "", 0, writer)?; match self.type_ { cuda_types::CUexternalMemoryHandleType::CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_FD => { writer.write_all(b", handle: ")?; CudaDisplay::write(unsafe { &self.handle.fd }, "", 0,writer)?; } cuda_types::CUexternalMemoryHandleType::CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32 | cuda_types::CUexternalMemoryHandleType::CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_HEAP | cuda_types::CUexternalMemoryHandleType::CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_RESOURCE |cuda_types::CUexternalMemoryHandleType::CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE => { write_win32_handle(unsafe { self.handle.win32 }, writer)?; } cuda_types::CUexternalMemoryHandleType::CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32_KMT | cuda_types::CUexternalMemoryHandleType::CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE_KMT => { writer.write_all(b", handle: ")?; CudaDisplay::write(unsafe { &self.handle.win32.handle }, "", 0,writer)?; } cuda_types::CUexternalMemoryHandleType::CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF => { writer.write_all(b", handle: ")?; CudaDisplay::write(unsafe { &self.handle.nvSciBufObject }, "", 0,writer)?; } _ => { writer.write_all(b", size: ")?; CudaDisplay::write(&self.size, "", 0,writer)?; writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0,writer)?; return writer.write_all(b", ... }") } } writer.write_all(b", size: ")?; CudaDisplay::write(&self.size, "", 0, writer)?; writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0, writer)?; writer.write_all(b" }") } } pub fn write_win32_handle( win32: cuda_types::CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st__bindgen_ty_1__bindgen_ty_1, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { if win32.handle != ptr::null_mut() { writer.write_all(b", handle: ")?; CudaDisplay::write(&win32.handle, "", 0, writer)?; } if win32.name != ptr::null_mut() { let name_ptr = win32.name as *const u16; let mut strlen = 0usize; while unsafe { *name_ptr.add(strlen) } != 0 { strlen += 1; } let text = String::from_utf16_lossy(unsafe { slice::from_raw_parts(name_ptr, strlen) }); write!(writer, ", name: \"{}\"", text)?; } Ok(()) } impl CudaDisplay for cuda_types::CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { writer.write_all(b"{ type: ")?; CudaDisplay::write(&self.type_, "", 0, writer)?; match self.type_ { cuda_types::CUexternalSemaphoreHandleType::CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_FD => { writer.write_all(b", handle: ")?; CudaDisplay::write(unsafe { &self.handle.fd }, "", 0,writer)?; } cuda_types::CUexternalSemaphoreHandleType::CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32 | cuda_types::CUexternalSemaphoreHandleType::CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D12_FENCE | cuda_types::CUexternalSemaphoreHandleType::CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_FENCE | cuda_types::CUexternalSemaphoreHandleType::CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX | cuda_types::CUexternalSemaphoreHandleType::CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX_KMT => { write_win32_handle(unsafe { mem::transmute(self.handle.win32) }, writer)?; } cuda_types::CUexternalSemaphoreHandleType::CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32_KMT => { writer.write_all(b", handle: ")?; CudaDisplay::write(unsafe { &self.handle.win32.handle }, "", 0,writer)?; } cuda_types::CUexternalSemaphoreHandleType::CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC => { writer.write_all(b", handle: ")?; CudaDisplay::write(unsafe { &self.handle.nvSciSyncObj }, "", 0,writer)?; } _ => { writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0,writer)?; return writer.write_all(b", ... }") } } writer.write_all(b", flags: ")?; CudaDisplay::write(&self.flags, "", 0, writer)?; writer.write_all(b" }") } } impl CudaDisplay for cuda_types::CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st__bindgen_ty_1__bindgen_ty_2 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { writer.write_all(b"{ fence: ")?; CudaDisplay::write(&unsafe { self.fence }, "", 0, writer)?; writer.write_all(b" }") } } impl CudaDisplay for cuda_types::CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st__bindgen_ty_1__bindgen_ty_2 { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { writer.write_all(b"{ fence: ")?; CudaDisplay::write(&unsafe { self.fence }, "", 0, writer)?; writer.write_all(b" }") } } impl CudaDisplay for cuda_types::CUgraphNodeParams_st { fn write( &self, _fn_name: &'static str, _index: usize, _writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { todo!() } } impl CudaDisplay for cuda_types::CUlaunchConfig_st { fn write( &self, _fn_name: &'static str, _index: usize, _writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { todo!() } } impl CudaDisplay for cuda_types::CUeglFrame_st { fn write( &self, _fn_name: &'static str, _index: usize, _writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { todo!() } } impl CudaDisplay for cuda_types::CUdevResource_st { fn write( &self, _fn_name: &'static str, _index: usize, _writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { todo!() } } impl CudaDisplay for cuda_types::CUlaunchAttribute_st { fn write( &self, _fn_name: &'static str, _index: usize, _writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { todo!() } } impl CudaDisplay for *mut T { fn write( &self, fn_name: &'static str, index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { if *self == ptr::null_mut() { writer.write_all(b"NULL") } else { let this: &T = unsafe { &**self }; this.write(fn_name, index, writer) } } } impl CudaDisplay for *const T { fn write( &self, fn_name: &'static str, index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { if *self == ptr::null() { writer.write_all(b"NULL") } else { let this: &T = unsafe { &**self }; this.write(fn_name, index, writer) } } } impl CudaDisplay for [T; N] { fn write( &self, _fn_name: &'static str, _index: usize, writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { writer.write_all(b"[")?; for i in 0..N { CudaDisplay::write(&self[i], "", 0, writer)?; if i != N - 1 { writer.write_all(b", ")?; } } writer.write_all(b"]") } } impl CudaDisplay for cuda_types::CUarrayMapInfo_st { fn write( &self, _fn_name: &'static str, _index: usize, _writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { todo!() } } impl CudaDisplay for cuda_types::CUexecAffinityParam_st { fn write( &self, _fn_name: &'static str, _index: usize, _writer: &mut (impl std::io::Write + ?Sized), ) -> std::io::Result<()> { todo!() } } #[allow(non_snake_case)] pub fn write_cuGraphKernelNodeGetAttribute( writer: &mut (impl std::io::Write + ?Sized), hNode: cuda_types::CUgraphNode, attr: cuda_types::CUkernelNodeAttrID, value_out: *mut cuda_types::CUkernelNodeAttrValue, ) -> std::io::Result<()> { writer.write_all(b"(hNode: ")?; CudaDisplay::write(&hNode, "cuGraphKernelNodeGetAttribute", 0, writer)?; writer.write_all(b", attr: ")?; CudaDisplay::write(&attr, "cuGraphKernelNodeGetAttribute", 1, writer)?; write_launch_attribute(writer, "cuGraphKernelNodeGetAttribute", 2, attr, value_out)?; writer.write_all(b") ") } #[allow(non_snake_case)] pub fn write_cuGraphKernelNodeSetAttribute( writer: &mut (impl std::io::Write + ?Sized), hNode: cuda_types::CUgraphNode, attr: cuda_types::CUkernelNodeAttrID, value_out: *const cuda_types::CUkernelNodeAttrValue, ) -> std::io::Result<()> { write_cuGraphKernelNodeGetAttribute(writer, hNode, attr, value_out as *mut _) } #[allow(non_snake_case)] pub fn write_cuStreamGetAttribute( writer: &mut (impl std::io::Write + ?Sized), hStream: cuda_types::CUstream, attr: cuda_types::CUstreamAttrID, value_out: *mut cuda_types::CUstreamAttrValue, ) -> std::io::Result<()> { writer.write_all(b"(hStream: ")?; CudaDisplay::write(&hStream, "cuStreamGetAttribute", 0, writer)?; writer.write_all(b", attr: ")?; CudaDisplay::write(&attr, "cuStreamGetAttribute", 1, writer)?; write_launch_attribute(writer, "cuStreamGetAttribute", 2, attr, value_out)?; writer.write_all(b") ") } fn write_launch_attribute( writer: &mut (impl std::io::Write + ?Sized), fn_name: &'static str, index: usize, attribute: cuda_types::CUlaunchAttributeID, value_out: *mut cuda_types::CUstreamAttrValue, ) -> std::io::Result<()> { match attribute { cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW => { writer.write_all(b", value_out: ")?; CudaDisplay::write( unsafe { &(*value_out).accessPolicyWindow }, fn_name, index, writer, ) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_COOPERATIVE => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).cooperative }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).syncPolicy }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).clusterDim }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).clusterSchedulingPolicyPreference }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).programmaticStreamSerializationAllowed }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).programmaticEvent }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_PRIORITY => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).priority }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).memSyncDomainMap }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).memSyncDomain }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).launchCompletionEvent }, fn_name, index, writer) } cuda_types::CUlaunchAttributeID::CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE => { writer.write_all(b", value_out: ")?; CudaDisplay::write(unsafe { &(*value_out).deviceUpdatableKernelNode }, fn_name, index, writer) } _ => writer.write_all(b", ... "), } } #[allow(non_snake_case)] pub fn write_cuStreamGetAttribute_ptsz( writer: &mut (impl std::io::Write + ?Sized), hStream: cuda_types::CUstream, attr: cuda_types::CUstreamAttrID, value_out: *mut cuda_types::CUstreamAttrValue, ) -> std::io::Result<()> { write_cuStreamGetAttribute(writer, hStream, attr, value_out) } #[allow(non_snake_case)] pub fn write_cuStreamSetAttribute( writer: &mut (impl std::io::Write + ?Sized), hStream: cuda_types::CUstream, attr: cuda_types::CUstreamAttrID, value_out: *const cuda_types::CUstreamAttrValue, ) -> std::io::Result<()> { write_cuStreamGetAttribute(writer, hStream, attr, value_out as *mut _) } #[allow(non_snake_case)] pub fn write_cuStreamSetAttribute_ptsz( writer: &mut (impl std::io::Write + ?Sized), hStream: cuda_types::CUstream, attr: cuda_types::CUstreamAttrID, value_out: *const cuda_types::CUstreamAttrValue, ) -> std::io::Result<()> { write_cuStreamSetAttribute(writer, hStream, attr, value_out) } #[allow(non_snake_case)] pub fn write_cuGLGetDevices( _writer: &mut (impl std::io::Write + ?Sized), _pCudaDeviceCount: *mut ::std::os::raw::c_uint, _pCudaDevices: *mut CUdevice, _cudaDeviceCount: ::std::os::raw::c_uint, _deviceList: CUGLDeviceList, ) -> std::io::Result<()> { todo!() } #[allow(non_snake_case)] pub fn write_cuGLGetDevices_v2( _writer: &mut (impl std::io::Write + ?Sized), _pCudaDeviceCount: *mut ::std::os::raw::c_uint, _pCudaDevices: *mut CUdevice, _cudaDeviceCount: ::std::os::raw::c_uint, _deviceList: CUGLDeviceList, ) -> std::io::Result<()> { todo!() } #[path = "format_generated.rs"] mod format_generated; pub(crate) use format_generated::*;