From 51fb6aeb452973b41e2f012930fc8d52a21ced2d Mon Sep 17 00:00:00 2001 From: Martin Loretz <20306567+martinloretzzz@users.noreply.github.com> Date: Thu, 2 Mar 2023 15:52:34 +0100 Subject: [PATCH] Fix cuda runtime (#625) --- 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 7f8fc265ba..2c3d9e8781 100644 --- a/tinygrad/runtime/ops_cuda.py +++ b/tinygrad/runtime/ops_cuda.py @@ -24,7 +24,7 @@ class CUDAProgram: global_size = global_size + [1] * (3 - len(global_size)) assert all(x%y == 0 for x,y in zip(global_size, local_size)), f"local:{local_size} must divide global:{global_size}" global_size = [x//y for x,y in zip(global_size, local_size)] - self.prg(*args, block=tuple(local_size), grid=tuple(global_size)) + self.prg(*[x._cl for x in args], block=tuple(local_size), grid=tuple(global_size)) class CUDACodegen(GPUCodegen): lang = GPULanguage(