BENCH: fix issue with UnaryFP benchmark, and reduce parametrization

rgommers · Aug 27, 2023 · c459e95 · c459e95
1 parent 59ccb8c
commit c459e95
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/benchmarks/benchmarks/bench_ufunc_strides.py b/benchmarks/benchmarks/bench_ufunc_strides.py
@@ -100,7 +100,11 @@ def time_unary(self, ufunc, stride_in, stride_out, dtype):
         ufunc(*self.ufunc_args)
 
 class UnaryFP(_AbstractUnary):
-    params = [UFUNCS_UNARY, [1, 2, 4], [1, 2, 4], ['e', 'f', 'd']]
+    params = [[uf for uf in UFUNCS_UNARY if uf != np.invert],
+              [1, 4],
+              [1, 2],
+              ['e', 'f', 'd']
+             ]
 
     def setup(self, ufunc, stride_in, stride_out, dtype):
         _AbstractUnary.setup(self, ufunc, stride_in, stride_out, dtype)
@@ -115,7 +119,7 @@ class UnaryFPSpecial(UnaryFP):
 class BinaryFP(_AbstractBinary):
     params = [
         [np.maximum, np.minimum, np.fmax, np.fmin, np.ldexp],
-        [1, 2, 4], [1, 2, 4], [1, 2, 4], ['f', 'd']
+        [1, 2], [1, 4], [1, 2, 4], ['f', 'd']
     ]
 
 class BinaryFPSpecial(BinaryFP):