package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-05 15:16:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
619455fee75d13f2da63dba3e1a60062b8f819c7
gonum/blas/testblas/dtrmm.go
Vladimir Chalupecky b96df58db9 all: add missing copyright headers
2018-06-22 17:32:53 +02:00

811 lines
13 KiB
Go
Raw Blame History

// Copyright ©2014 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 testblas
import (
"testing"
"gonum.org/v1/gonum/blas"
"gonum.org/v1/gonum/floats"
)
type Dtrmmer interface {
Dtrmm(s blas.Side, ul blas.Uplo, tA blas.Transpose, d blas.Diag, m, n int, alpha float64, a []float64, lda int, b []float64, ldb int)
}
func DtrmmTest(t *testing.T, blasser Dtrmmer) {
for i, test := range []struct {
s blas.Side
ul blas.Uplo
tA blas.Transpose
d blas.Diag
m int
n int
alpha float64
a [][]float64
b [][]float64
ans [][]float64
}{
{
s: blas.Left,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3, 4},
{0, 5, 6, 7},
{0, 0, 8, 9},
{0, 0, 0, 10},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{320, 340, 360},
{588, 624, 660},
{598, 632, 666},
{380, 400, 420},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2},
{0, 5},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{72, 78, 84},
{130, 140, 150},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3, 4},
{0, 5, 6, 7},
{0, 0, 8, 9},
{0, 0, 0, 10},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{320, 340, 360},
{484, 512, 540},
{374, 394, 414},
{38, 40, 42},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2},
{0, 5},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{72, 78, 84},
{26, 28, 30},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0, 0},
{2, 5, 0, 0},
{3, 6, 8, 0},
{4, 7, 9, 10},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{20, 22, 24},
{170, 184, 198},
{472, 506, 540},
{930, 990, 1050},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0},
{2, 5},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{20, 22, 24},
{170, 184, 198},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0, 0},
{2, 5, 0, 0},
{3, 6, 8, 0},
{4, 7, 9, 10},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{20, 22, 24},
{66, 72, 78},
{248, 268, 288},
{588, 630, 672},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0},
{2, 5},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{20, 22, 24},
{66, 72, 78},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.Trans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3, 4},
{0, 5, 6, 7},
{0, 0, 8, 9},
{0, 0, 0, 10},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{20, 22, 24},
{170, 184, 198},
{472, 506, 540},
{930, 990, 1050},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.Trans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2},
{0, 5},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{20, 22, 24},
{170, 184, 198},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.Trans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3, 4},
{0, 5, 6, 7},
{0, 0, 8, 9},
{0, 0, 0, 10},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{20, 22, 24},
{66, 72, 78},
{248, 268, 288},
{588, 630, 672},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.Trans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2},
{0, 5},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{20, 22, 24},
{66, 72, 78},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.Trans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0, 0},
{2, 5, 0, 0},
{3, 6, 8, 0},
{4, 7, 9, 10},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{320, 340, 360},
{588, 624, 660},
{598, 632, 666},
{380, 400, 420},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.Trans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0},
{2, 5},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{72, 78, 84},
{130, 140, 150},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.Trans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0, 0},
{2, 5, 0, 0},
{3, 6, 8, 0},
{4, 7, 9, 10},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{320, 340, 360},
{484, 512, 540},
{374, 394, 414},
{38, 40, 42},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.Trans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0},
{2, 5},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{72, 78, 84},
{26, 28, 30},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{20, 128, 314},
{26, 164, 398},
{32, 200, 482},
{38, 236, 566},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{20, 128, 314},
{26, 164, 398},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{20, 62, 194},
{26, 80, 248},
{32, 98, 302},
{38, 116, 356},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{20, 62, 194},
{26, 80, 248},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0},
{2, 4, 0},
{3, 5, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{136, 208, 144},
{172, 262, 180},
{208, 316, 216},
{244, 370, 252},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0},
{2, 4, 0},
{3, 5, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{136, 208, 144},
{172, 262, 180},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0},
{2, 4, 0},
{3, 5, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{136, 142, 24},
{172, 178, 30},
{208, 214, 36},
{244, 250, 42},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0},
{2, 4, 0},
{3, 5, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{136, 142, 24},
{172, 178, 30},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.Trans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{136, 208, 144},
{172, 262, 180},
{208, 316, 216},
{244, 370, 252},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.Trans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{136, 208, 144},
{172, 262, 180},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.Trans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{136, 142, 24},
{172, 178, 30},
{208, 214, 36},
{244, 250, 42},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.Trans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{136, 142, 24},
{172, 178, 30},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.Trans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0},
{2, 4, 0},
{3, 5, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{20, 128, 314},
{26, 164, 398},
{32, 200, 482},
{38, 236, 566},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.Trans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0},
{2, 4, 0},
{3, 5, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{20, 128, 314},
{26, 164, 398},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.Trans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0},
{2, 4, 0},
{3, 5, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{20, 62, 194},
{26, 80, 248},
{32, 98, 302},
{38, 116, 356},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.Trans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 2,
a: [][]float64{
{1, 0, 0},
{2, 4, 0},
{3, 5, 6},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{20, 62, 194},
{26, 80, 248},
},
},
} {
aFlat := flatten(test.a)
bFlat := flatten(test.b)
ansFlat := flatten(test.ans)
blasser.Dtrmm(test.s, test.ul, test.tA, test.d, test.m, test.n, test.alpha, aFlat, len(test.a[0]), bFlat, len(test.b[0]))
if !floats.EqualApprox(ansFlat, bFlat, 1e-14) {
t.Errorf("Case %v. Want %v, got %v.", i, ansFlat, bFlat)
}
}
}
Reference in New Issue View Git Blame Copy Permalink