shallow water: fixed integer division

bh107 · May 16, 2018 · 9cae4fd · 9cae4fd
1 parent bf0e09a
commit 9cae4fd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchpress/benchmarks/shallow_water/python_numpy/shallow_water.py b/benchpress/benchmarks/shallow_water/python_numpy/shallow_water.py
@@ -30,10 +30,10 @@ def model(height, width, dtype=np.float32):
 
     m = np.ones((height, width), dtype=dtype)
     D = droplet(8, 8)  # simulate a water drop
-    droploc = height / 4
+    droploc = height // 4
     (dropx, dropy) = D.shape
     m[droploc:droploc + dropx, droploc:droploc + dropy] += D
-    droploc = height / 2
+    droploc = height // 2
     (dropx, dropy) = D.shape
     m[droploc:droploc + dropx, droploc:droploc + dropy] += D