replace njit with cuda.jit

CEMeNT-PSAAP · Aug 15, 2024 · d8a3e00 · d8a3e00
1 parent ccbfbe8
commit d8a3e00
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/mcdc/code_factory.py b/mcdc/code_factory.py
@@ -1,6 +1,6 @@
 import numpy as np
 
-from numba import njit
+from numba import cuda, njit
 
 import mcdc.local as local
 import mcdc.type_ as type_
@@ -36,7 +36,7 @@ def local_array(dtype, size, target):
     def cpu():
         return np.zeros(1, dtype=struct)[0]
 
-    @njit
+    @cuda.jit(device=True)
     def gpu():
         return cuda.local.array(1, dtype=struct)[0]
 
@@ -49,7 +49,7 @@ def local_object(dtype, target):
     def cpu():
         return np.zeros(1, dtype=dtype)[0]
 
-    @njit
+    @cuda.jit(device=True)
     def gpu():
         return cuda.local.array(1, dtype=dtype)[0]
 

diff --git a/mcdc/main.py b/mcdc/main.py
@@ -384,7 +384,7 @@ def prepare():
     type_.make_type_global(input_deck)
     kernel.adapt_rng(nb.config.DISABLE_JIT)
 
-    input_deck.setting['target'] = target
+    input_deck.setting["target"] = target
     code_factory.make_locals(input_deck)
 
     # =========================================================================