Cache jitted function.

mdbartos · Feb 5, 2024 · 91176eb · 91176eb
1 parent c234320
commit 91176eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysheds/_sgrid.py b/pysheds/_sgrid.py
@@ -1919,7 +1919,7 @@ def count(start=0, step=1):
         yield n
         n += step
 
-@njit
+@njit(cache=True)
 def _priority_flood(dem, dem_mask, tuple_type):
     open_cells = typedlist.List.empty_list(tuple_type)  # Priority queue
     pits = typedlist.List.empty_list(tuple_type)  # FIFO queue