From 0eebb8e9981060f9fdf6df1b0bbee1fda036732d Mon Sep 17 00:00:00 2001 From: Roelof van Dijk <3604013+roelofvandijk@users.noreply.github.com> Date: Sat, 8 Jun 2024 14:45:06 +0200 Subject: [PATCH] fix: _free should not return (#4880) --- tinygrad/runtime/ops_cuda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/runtime/ops_cuda.py b/tinygrad/runtime/ops_cuda.py index 2144cccedb..8cc265da28 100644 --- a/tinygrad/runtime/ops_cuda.py +++ b/tinygrad/runtime/ops_cuda.py @@ -121,7 +121,7 @@ class CUDAAllocator(LRUAllocator): if options.host: return init_c_var(ctypes.c_void_p(), lambda x: check(cuda.cuMemHostAlloc(ctypes.byref(x), size, 0x01))) return init_c_var(cuda.CUdeviceptr(), lambda x: check(cuda.cuMemAlloc_v2(ctypes.byref(x), size))) def _free(self, opaque, options:BufferOptions): - if options.host: return check(cuda.cuMemFreeHost(opaque)) + if options.host: check(cuda.cuMemFreeHost(opaque)) else: check(cuda.cuMemFree_v2(opaque)) def copyin(self, dest, src:memoryview): check(cuda.cuCtxSetCurrent(self.device.context))