BENCH: Adds benchmarks for arctan2 and power

1 files changed, 17 insertions, 0 deletions

diff --git a/benchmarks/benchmarks/bench_ufunc.py b/benchmarks/benchmarks/bench_ufunc.py
index 858dcccfc..afe6118b1 100644
--- a/benchmarks/benchmarks/bench_ufunc.py
+++ b/benchmarks/benchmarks/bench_ufunc.py
@@ -294,3 +294,20 @@ class ArgParsingReduce(Benchmark):
 
     def time_add_reduce_arg_parsing(self, arg_pack):
         np.add.reduce(*arg_pack.args, **arg_pack.kwargs)
+
+class BinaryBench(Benchmark):
+    def setup(self):
+        N = 1000000
+        self.a32 = np.random.rand(N).astype(np.float32)
+        self.b32 = np.random.rand(N).astype(np.float32)
+        self.a64 = np.random.rand(N).astype(np.float64)
+        self.b64 = np.random.rand(N).astype(np.float64)
+    
+    def time_pow_32(self):
+        np.power(self.a32, self.b32)
+    def time_pow_64(self):
+        np.power(self.a64, self.b64)
+    def time_atan2_32(self):
+        np.arctan2(self.a32, self.b32)
+    def time_atan2_64(self):
+        np.arctan2(self.a64, self.b64)
\ No newline at end of file