diff options
author | John Kirkham <kirkhamj@janelia.hhmi.org> | 2016-01-08 18:05:46 -0500 |
---|---|---|
committer | John Kirkham <kirkhamj@janelia.hhmi.org> | 2016-01-08 18:53:00 -0500 |
commit | 5a1d64e85724118714fa76bb67695c884c67ec92 (patch) | |
tree | 4703bb9aa81fe97a7da30eb877cdcaf0cad6c69e /benchmarks | |
parent | 8a76291c76aa9b33b18ceb06f6e8f37f990c9e27 (diff) | |
download | numpy-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.py | 7 |
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) |