Skip to content

Commit 05392f3

Browse files
andfoyLegNeato
authored andcommitted
Add CUDA IPC functions
1 parent bac72cd commit 05392f3

File tree

2 files changed

+9
-0
lines changed

2 files changed

+9
-0
lines changed

crates/cuda_std/src/rt/driver_types_sys.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -387,3 +387,8 @@ pub struct CUevent_st {
387387
_unused: [u8; 0],
388388
}
389389
pub type cudaEvent_t = *mut CUevent_st;
390+
391+
#[repr(C)]
392+
pub struct cudaIpcMemHandle_t {
393+
pub reserved: [u8; 64],
394+
}

crates/cuda_std/src/rt/sys.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,4 +51,8 @@ extern "C" {
5151
shared_mem_size: c_uint,
5252
) -> *mut c_void;
5353
pub fn cudaLaunchDeviceV2(parameter_buffer: *mut c_void, stream: cudaStream_t) -> cudaError_t;
54+
55+
pub fn cudaIpcGetMemHandle(handle: *mut cudaIpcMemHandle_t, dev_ptr: *mut c_void) -> cudaError_t;
56+
pub fn cudaIpcOpenMemHandle(dev_ptr: *mut *mut c_void, handle: cudaIpcMemHandle_t, flags: u32) -> cudaError_t;
57+
pub fn cudaIpcCloseMemHandle(dev_ptr: *mut c_void) -> cudaError_t;
5458
}

0 commit comments

Comments
 (0)