diff --git a/benchmarks/benchmarks/bench_reduce.py b/benchmarks/benchmarks/bench_reduce.py index 040b5ca73ce2..53016f238b45 100644 --- 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 index 7d7195ed8187..72c2a6132e4e 100644 --- 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']