inducer / pycuda

CUDA integration for Python, plus shiny features
http://mathema.tician.de/software/pycuda
Other
1.85k stars 288 forks source link

Combine Cuda IPC wth Direct Access #312

Closed eedalong closed 3 years ago

eedalong commented 3 years ago

I have 2 GPUs and they are under the same PCIe Switch. When I enable P2P access and allocate memory blocks on device1 and device0, I can access device1's memory block in kernels launched on device 0 within current process. I wander whether a cuda IPC memory handle, allocated on device1 from another process, can somehow be directly accessed in kernel launched on device0 within current process.

inducer commented 3 years ago

Isn't that what's controlled by the LAZY_ENABLE_PEER_ACCESS flag passed to the IPCMemoryHandle constructor?

See

eedalong commented 3 years ago

Yes, It works, Thx! @inducer