File: dtrmvbench_test.go

package info (click to toggle)
golang-gonum-v1-gonum 0.15.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 18,792 kB
  • sloc: asm: 6,252; fortran: 5,271; sh: 377; ruby: 211; makefile: 98
file content (52 lines) | stat: -rw-r--r-- 1,261 bytes parent folder | download
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
// Copyright ©2017 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 gonum

import (
	"strconv"
	"testing"

	"gonum.org/v1/gonum/blas"
	"gonum.org/v1/gonum/blas/testblas"
)

func BenchmarkDtrmv(b *testing.B) {
	for _, n := range []int{testblas.MediumMat, testblas.LargeMat} {
		for _, incX := range []int{1, 5} {
			for _, uplo := range []blas.Uplo{blas.Upper, blas.Lower} {
				for _, trans := range []blas.Transpose{blas.NoTrans, blas.Trans} {
					for _, unit := range []blas.Diag{blas.NonUnit, blas.Unit} {
						var str string
						if n == testblas.MediumMat {
							str += "Med"
						} else if n == testblas.LargeMat {
							str += "Large"
						}
						str += "_Inc" + strconv.Itoa(incX)
						if uplo == blas.Upper {
							str += "_UP"
						} else {
							str += "_LO"
						}
						if trans == blas.NoTrans {
							str += "_NT"
						} else {
							str += "_TR"
						}
						if unit == blas.NonUnit {
							str += "_NU"
						} else {
							str += "_UN"
						}
						lda := n
						b.Run(str, func(b *testing.B) {
							testblas.DtrmvBenchmark(b, Implementation{}, n, lda, incX, uplo, trans, unit)
						})
					}
				}
			}
		}
	}
}