BENCH: fix bench_reduce.py and bench_shape_base.py

rgommers · Aug 27, 2023 · 2cdbc42 · 2cdbc42
1 parent f8bab52
commit 2cdbc42
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/benchmarks/benchmarks/bench_reduce.py b/benchmarks/benchmarks/bench_reduce.py
@@ -46,18 +46,11 @@ def time_any_slow(self):
 
 
 class StatsReductions(Benchmark):
-    # Not testing, but in array_api (redundant)
-    # 8, 16, 32 bit variants, and 128 complexes
-    params = ['int64', 'uint64', 'float64', 'intp',
-               'complex64', 'bool', 'float', 'int',
-               'complex', 'complex256'],
+    params = ['int64', 'uint64', 'float32', 'float64', 'complex64', 'bool_'],
     param_names = ['dtype']
 
     def setup(self, dtype):
-        try:
-            self.data = np.ones(200, dtype=getattr(np, dtype))
-        except AttributeError:  # builtins throw AttributeError after 1.20
-            self.data = np.ones(200, dtype=dtype)
+        self.data = np.ones(200, dtype=dtype)
         if dtype.startswith('complex'):
             self.data = self.data * self.data.T*1j
 

diff --git a/benchmarks/benchmarks/bench_shape_base.py b/benchmarks/benchmarks/bench_shape_base.py
@@ -68,7 +68,7 @@ def time_no_lists(self, n):
 
 
 class Block2D(Benchmark):
-    params = [[(16, 16), (32, 32), (64, 64), (128, 128), (256, 256), (512, 512), (1024, 1024)],
+    params = [[(16, 16), (64, 64), (256, 256), (1024, 1024)],
               ['uint8', 'uint16', 'uint32', 'uint64'],
               [(2, 2), (4, 4)]]
     param_names = ['shape', 'dtype', 'n_chunks']