1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
// 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"
)
func DgeevBenchmark(b *testing.B, impl Dgeever) {
var resultGeneral blas64.General
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.LeftEVCompute, lapack.RightEVCompute, n, a.Data, a.Stride, wr, wi, vl.Data, vl.Stride, vr.Data, vr.Stride, 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.LeftEVCompute, lapack.RightEVCompute, n, a.Data, a.Stride, wr, wi,
vl.Data, vl.Stride, vr.Data, vr.Stride, work, len(work))
}
resultGeneral = a
})
}
if resultGeneral.Data == nil {
b.Error("unexpected nil data")
}
}
|
