Keep only the Ellipsis workaround for Pythran

The serge-sans-paille/pythran#1951 PR has fixed nearly all issues with this benchmark pip install -e 'git+https://github.com/serge-sans-paille/pythran.git@fix/isoneutral#egg=pythran' Commit: serge-sans-paille/pythran@d305e4a
ashwinvis · Jan 1, 2022 · 1feac29 · 1feac29
1 parent 63d16f6
commit 1feac29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/isoneutral_mixing/isoneutral_transonic_pythran.py b/benchmarks/isoneutral_mixing/isoneutral_transonic_pythran.py
@@ -159,7 +159,7 @@ def dm_taper(sx):
     """
     Compute Ai_nz and K_22 on center of north face of T cell.
     """
-    diffloc[...] = 0
+    diffloc[:, :, :] = 0
     diffloc[2:-2, 1:-2, 1:] = 0.25 * (
         K_iso[2:-2, 1:-2, 1:]
         + K_iso[2:-2, 1:-2, :-1]