work in progress

pyccel · Jul 25, 2024 · 1aeb5a5 · 1aeb5a5
1 parent e9436a9
commit 1aeb5a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyccel/decorators.py b/pyccel/decorators.py
@@ -153,9 +153,10 @@ def internal_loop(*args, **kwargs):
                         if 'cuda' in self._f.__globals__:
                             self._f.__globals__['cuda'].threadIdx = cu.threadIdx
                             self._f.__globals__['cuda'].blockIdx = cu.blockIdx
-                            self._f.__globals__['cuda'].blockDim = num_threads
+                            self._f.__globals__['cuda'].blockDim = cu.blockDim
                         else:
                             self._f.__globals__['cuda'] = cu
+                        self._f(*args, **kwargs)
             return internal_loop
 
     return KernelAccessor(f)