use nogil and cache in numba decorator

louis-richard · May 15, 2023 · 623a0ac · 623a0ac
1 parent 70954c1
commit 623a0ac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyrfu/pyrf/int_sph_dist.py b/pyrfu/pyrf/int_sph_dist.py
@@ -230,7 +230,7 @@ def int_sph_dist(vdf, speed, phi, theta, speed_grid, **kwargs):
     return pst
 
 
-@numba.jit(nopython=True, parallel=True, fastmath=True)
+@numba.jit(cache=True, nogil=True, parallel=True, nopython=True)
 def mc_pol_1d(
     vdf,
     v,
@@ -456,7 +456,7 @@ def mc_cart_3d(
     return f_g
 
 
-@numba.jit(nopython=True, parallel=True, fastmath=True)
+@numba.jit(cache=True, nogil=True, parallel=True, nopython=True)
 def mc_cart_2d(
     vdf,
     v,