github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-13 08:05:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: CUDA_P2P works (#4155)

Browse Source
This commit is contained in:
nimlgen
2024-04-12 18:20:12 +03:00
committed by GitHub
parent 5a57b48134
commit 24a27a01a9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/graph/cuda.py
View File

@@ -46,7 +46,7 @@ class CUDAGraph(MultiDeviceJITGraph):
node_from = cuda.CUgraphNode()
deps = self.access_resources(read=[src], write=[dest], new_dependency=node_from)
c_deps = (cuda.CUgraphNode*len(deps))(*deps) if deps else None
if getenv("CUDA_P2P", CUDADevice.peer_access):
if getenv("CUDA_P2P", int(CUDADevice.peer_access)):
cp_params = cuda.CUDA_MEMCPY3D_v2(srcMemoryType=cuda.CU_MEMORYTYPE_DEVICE, srcDevice=src._buf, srcPitch=src.nbytes, srcHeight=1,
dstMemoryType=cuda.CU_MEMORYTYPE_DEVICE, dstDevice=dest._buf, dstPitch=dest.nbytes, dstHeight=1,
WidthInBytes=dest.nbytes, Height=1, Depth=1)