shallow water: initialize coord arrays with int64

IntelPython · Dec 16, 2024 · ed1ee9c · ed1ee9c
1 parent a609ea4
commit ed1ee9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/shallow_water.py b/examples/shallow_water.py
@@ -115,10 +115,10 @@ def ind_arr(shape, columns=False):
         """Construct an (nx, ny) array where each row/col is an arange"""
         nx, ny = shape
         if columns:
-            ind = np.arange(0, nx * ny, 1, dtype=np.int32) % nx
+            ind = np.arange(0, nx * ny, 1, dtype=np.int64) % nx
             ind = transpose(np.reshape(ind, (ny, nx)))
         else:
-            ind = np.arange(0, nx * ny, 1, dtype=np.int32) % ny
+            ind = np.arange(0, nx * ny, 1, dtype=np.int64) % ny
             ind = np.reshape(ind, (nx, ny))
         return ind.astype(dtype)