Hulk did something

[bytom/vapor.git] / vendor / gonum.org / v1 / gonum / lapack / testlapack / dgeev_bench.go

diff --git a/vendor/gonum.org/v1/gonum/lapack/testlapack/dgeev_bench.go b/vendor/gonum.org/v1/gonum/lapack/testlapack/dgeev_bench.go
new file mode 100644 (file)
index 0000000..01007e4
--- /dev/null
+++ b/vendor/gonum.org/v1/gonum/lapack/testlapack/dgeev_bench.go
@@ -0,0 +1,62 @@
+// Copyright ©2016 The Gonum Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package testlapack
+
+import (
+       "testing"
+
+       "golang.org/x/exp/rand"
+
+       "gonum.org/v1/gonum/blas/blas64"
+       "gonum.org/v1/gonum/lapack"
+)
+
+var resultGeneral blas64.General
+
+func DgeevBenchmark(b *testing.B, impl Dgeever) {
+       rnd := rand.New(rand.NewSource(1))
+       benchmarks := []struct {
+               name string
+               a    blas64.General
+       }{
+               {"AntisymRandom3", NewAntisymRandom(3, rnd).Matrix()},
+               {"AntisymRandom4", NewAntisymRandom(4, rnd).Matrix()},
+               {"AntisymRandom5", NewAntisymRandom(5, rnd).Matrix()},
+               {"AntisymRandom10", NewAntisymRandom(10, rnd).Matrix()},
+               {"AntisymRandom50", NewAntisymRandom(50, rnd).Matrix()},
+               {"AntisymRandom100", NewAntisymRandom(100, rnd).Matrix()},
+               {"AntisymRandom200", NewAntisymRandom(200, rnd).Matrix()},
+               {"AntisymRandom500", NewAntisymRandom(500, rnd).Matrix()},
+               {"Circulant3", Circulant(3).Matrix()},
+               {"Circulant4", Circulant(4).Matrix()},
+               {"Circulant5", Circulant(5).Matrix()},
+               {"Circulant10", Circulant(10).Matrix()},
+               {"Circulant50", Circulant(50).Matrix()},
+               {"Circulant100", Circulant(100).Matrix()},
+               {"Circulant200", Circulant(200).Matrix()},
+               {"Circulant500", Circulant(500).Matrix()},
+       }
+       for _, bm := range benchmarks {
+               n := bm.a.Rows
+               a := zeros(n, n, n)
+               vl := zeros(n, n, n)
+               vr := zeros(n, n, n)
+               wr := make([]float64, n)
+               wi := make([]float64, n)
+               work := make([]float64, 1)
+               impl.Dgeev(lapack.ComputeLeftEV, lapack.ComputeRightEV, n, nil, n, nil, nil, nil, n, nil, n, work, -1)
+               work = make([]float64, int(work[0]))
+               b.Run(bm.name, func(b *testing.B) {
+                       for i := 0; i < b.N; i++ {
+                               b.StopTimer()
+                               copyGeneral(a, bm.a)
+                               b.StartTimer()
+                               impl.Dgeev(lapack.ComputeLeftEV, lapack.ComputeRightEV, n, a.Data, a.Stride, wr, wi,
+                                       vl.Data, vl.Stride, vr.Data, vr.Stride, work, len(work))
+                       }
+                       resultGeneral = a
+               })
+       }
+}