summaryrefslogtreecommitdiff
path: root/benchmarks
diff options
context:
space:
mode:
authorJohn Kirkham <kirkhamj@janelia.hhmi.org>2016-01-08 18:05:46 -0500
committerJohn Kirkham <kirkhamj@janelia.hhmi.org>2016-01-08 18:53:00 -0500
commit5a1d64e85724118714fa76bb67695c884c67ec92 (patch)
tree4703bb9aa81fe97a7da30eb877cdcaf0cad6c69e /benchmarks
parent8a76291c76aa9b33b18ceb06f6e8f37f990c9e27 (diff)
downloadnumpy-5a1d64e85724118714fa76bb67695c884c67ec92.tar.gz
BENCH: Perform benchmarking for the computation of `inner` on a matrix with itself.
Diffstat (limited to 'benchmarks')
-rw-r--r--benchmarks/benchmarks/bench_linalg.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/benchmarks/benchmarks/bench_linalg.py b/benchmarks/benchmarks/bench_linalg.py
index c230d985a..3d26b800c 100644
--- a/benchmarks/benchmarks/bench_linalg.py
+++ b/benchmarks/benchmarks/bench_linalg.py
@@ -8,6 +8,7 @@ import numpy as np
class Eindot(Benchmark):
def setup(self):
self.a = np.arange(60000.0).reshape(150, 400)
+ self.ac = self.a.copy()
self.at = self.a.T
self.atc = self.a.T.copy()
self.b = np.arange(240000.0).reshape(400, 600)
@@ -35,6 +36,12 @@ class Eindot(Benchmark):
def time_dot_trans_atc_a(self):
np.dot(self.atc, self.a)
+ def time_inner_trans_a_a(self):
+ np.inner(self.a, self.a)
+
+ def time_inner_trans_a_ac(self):
+ np.inner(self.a, self.ac)
+
def time_einsum_i_ij_j(self):
np.einsum('i,ij,j', self.d, self.b, self.c)