BENCH: Added floor divide benchmarks (#17727)

1 files changed, 14 insertions, 9 deletions

diff --git a/benchmarks/benchmarks/bench_ufunc.py b/benchmarks/benchmarks/bench_ufunc.py
index c388da5b5..ef3ebe75f 100644
--- a/benchmarks/benchmarks/bench_ufunc.py
+++ b/benchmarks/benchmarks/bench_ufunc.py
@@ -134,15 +134,20 @@ class CustomScalar(Benchmark):
         (self.d < 1)
 
 
-class CustomScalarInt(Benchmark):
-    params = [10**size for size in range(1, 8)]
-    param_names = ['size']
-
-    def setup(self, size):
-        self.x = np.arange(size)
-
-    def time_floor_divide(self, size):
-        self.x//8
+class CustomScalarFloorDivideInt(Benchmark):
+    params = ([np.int8, np.int16, np.int32, np.int64], [8, -8, 43, -43])
+    param_names = ['dtype', 'divisors']
+    max_value = 10**7
+    min_value = -10**7
+
+    def setup(self, dtype, divisor):
+        iinfo = np.iinfo(dtype)
+        self.x = np.arange(
+                max(iinfo.min, self.min_value),
+                min(iinfo.max, self.max_value))
+
+    def time_floor_divide_int(self, dtpye, divisor):
+        self.x // divisor
 
 
 class Scalar(Benchmark):