From 5323bbfc15f320ac5b9c4b3bece6394f5bf74ee2 Mon Sep 17 00:00:00 2001
From: Raghuveer Devulapalli <raghuveer.devulapalli@intel.com>
Date: Mon, 1 Jul 2019 13:38:09 -0700
Subject: BENCH: adding benchmarks for avx based ufuncs

---
 benchmarks/benchmarks/bench_avx.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 benchmarks/benchmarks/bench_avx.py

(limited to 'benchmarks')

diff --git a/benchmarks/benchmarks/bench_avx.py b/benchmarks/benchmarks/bench_avx.py
new file mode 100644
index 000000000..f7b524e43
--- /dev/null
+++ b/benchmarks/benchmarks/bench_avx.py
@@ -0,0 +1,34 @@
+from __future__ import absolute_import, division, print_function
+
+from .common import Benchmark
+
+import numpy as np
+
+avx_ufuncs = ['sqrt',
+              'absolute',
+              'reciprocal',
+              'square',
+              'rint',
+              'floor',
+              'ceil' ,
+              'trunc']
+stride = [1, 2, 4]
+dtype  = ['f', 'd']
+
+class AVX_UFunc(Benchmark):
+    params = [avx_ufuncs, stride, dtype]
+    param_names = ['avx_based_ufunc', 'stride', 'dtype']
+    timeout = 10
+
+    def setup(self, ufuncname, stride, dtype):
+        np.seterr(all='ignore')
+        try:
+            self.f = getattr(np, ufuncname)
+        except AttributeError:
+            raise NotImplementedError()
+        N = 10000
+        self.arr = np.ones(stride*N, dtype)
+
+    def time_ufunc(self, ufuncname, stride, dtype):
+        self.f(self.arr[::stride])
+
-- 
cgit v1.2.1