From 24a27a01a952485a80c36b487e58339cdf6bf6aa Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Fri, 12 Apr 2024 18:20:12 +0300 Subject: [PATCH] hotfix: CUDA_P2P works (#4155) --- tinygrad/runtime/graph/cuda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/runtime/graph/cuda.py b/tinygrad/runtime/graph/cuda.py index 10d0d4c137..6eee53d1ce 100644 --- a/tinygrad/runtime/graph/cuda.py +++ b/tinygrad/runtime/graph/cuda.py @@ -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)