From bdd4e2e29c4b011cf455a72f85c92010656d2722 Mon Sep 17 00:00:00 2001 From: Chunlin Date: Mon, 1 Jun 2020 14:17:57 +0800 Subject: ENH: ARM Neon implementation with intrinsic for np.argmax. (#16375) * Neon implementation with intrinsic for bool argmax --- benchmarks/benchmarks/bench_reduce.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'benchmarks') diff --git a/benchmarks/benchmarks/bench_reduce.py b/benchmarks/benchmarks/bench_reduce.py index 76c573a51..7b05f4fcc 100644 --- a/benchmarks/benchmarks/bench_reduce.py +++ b/benchmarks/benchmarks/bench_reduce.py @@ -58,6 +58,15 @@ class MinMax(Benchmark): def time_max(self, dtype): np.max(self.d) +class ArgMax(Benchmark): + params = [np.float32, bool] + param_names = ['dtype'] + + def setup(self, dtype): + self.d = np.zeros(200000, dtype=dtype) + + def time_argmax(self, dtype): + np.argmax(self.d) class SmallReduction(Benchmark): def setup(self): -- cgit v1.2.1