Google Git
Sign in
fuchsia / third_party / github.com / gonum / gonum / refs/heads/upstream/joss-paper / . / blas / testblas / dtrsm.go
blob: 9623397cf9962de982f145ad133fbe03eda626d0 [file] [log] [blame]
package testblas
import (
"testing"
"gonum.org/v1/gonum/blas"
"gonum.org/v1/gonum/floats"
)
type Dtrsmer interface {
Dtrsm(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 DtrsmTest(t *testing.T, blasser Dtrsmer) {
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: 3,
n: 2,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 5},
},
b: [][]float64{
{3, 6},
{4, 7},
{5, 8},
},
ans: [][]float64{
{1, 3.4},
{-0.5, -0.5},
{2, 3.2},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.Unit,
m: 3,
n: 2,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 5},
},
b: [][]float64{
{3, 6},
{4, 7},
{5, 8},
},
ans: [][]float64{
{60, 96},
{-42, -66},
{10, 16},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 3,
n: 4,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 5},
},
b: [][]float64{
{3, 6, 2, 9},
{4, 7, 1, 3},
{5, 8, 9, 10},
},
ans: [][]float64{
{1, 3.4, 1.2, 13},
{-0.5, -0.5, -4, -3.5},
{2, 3.2, 3.6, 4},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.Unit,
m: 3,
n: 4,
alpha: 2,
a: [][]float64{
{1, 2, 3},
{0, 4, 5},
{0, 0, 5},
},
b: [][]float64{
{3, 6, 2, 9},
{4, 7, 1, 3},
{5, 8, 9, 10},
},
ans: [][]float64{
{60, 96, 126, 146},
{-42, -66, -88, -94},
{10, 16, 18, 20},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 3,
n: 2,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 4, 0},
{5, 6, 7},
},
b: [][]float64{
{3, 6},
{4, 7},
{5, 8},
},
ans: [][]float64{
{4.5, 9},
{-0.375, -1.5},
{-0.75, -12.0 / 7},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.Unit,
m: 3,
n: 2,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 4, 0},
{5, 6, 7},
},
b: [][]float64{
{3, 6},
{4, 7},
{5, 8},
},
ans: [][]float64{
{9, 18},
{-15, -33},
{60, 132},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 3,
n: 4,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 4, 0},
{5, 6, 7},
},
b: [][]float64{
{3, 6, 2, 9},
{4, 7, 1, 3},
{5, 8, 9, 10},
},
ans: [][]float64{
{4.5, 9, 3, 13.5},
{-0.375, -1.5, -1.5, -63.0 / 8},
{-0.75, -12.0 / 7, 3, 39.0 / 28},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.Unit,
m: 3,
n: 4,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 4, 0},
{5, 6, 7},
},
b: [][]float64{
{3, 6, 2, 9},
{4, 7, 1, 3},
{5, 8, 9, 10},
},
ans: [][]float64{
{9, 18, 6, 27},
{-15, -33, -15, -72},
{60, 132, 87, 327},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.Trans,
d: blas.NonUnit,
m: 3,
n: 2,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 7},
},
b: [][]float64{
{3, 6},
{4, 7},
{5, 8},
},
ans: [][]float64{
{4.5, 9},
{-0.30, -1.2},
{-6.0 / 35, -24.0 / 35},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.Trans,
d: blas.Unit,
m: 3,
n: 2,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 7},
},
b: [][]float64{
{3, 6},
{4, 7},
{5, 8},
},
ans: [][]float64{
{9, 18},
{-15, -33},
{69, 150},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.Trans,
d: blas.NonUnit,
m: 3,
n: 4,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 7},
},
b: [][]float64{
{3, 6, 6, 7},
{4, 7, 8, 9},
{5, 8, 10, 11},
},
ans: [][]float64{
{4.5, 9, 9, 10.5},
{-0.3, -1.2, -0.6, -0.9},
{-6.0 / 35, -24.0 / 35, -12.0 / 35, -18.0 / 35},
},
},
{
s: blas.Left,
ul: blas.Upper,
tA: blas.Trans,
d: blas.Unit,
m: 3,
n: 4,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 7},
},
b: [][]float64{
{3, 6, 6, 7},
{4, 7, 8, 9},
{5, 8, 10, 11},
},
ans: [][]float64{
{9, 18, 18, 21},
{-15, -33, -30, -36},
{69, 150, 138, 165},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.Trans,
d: blas.NonUnit,
m: 3,
n: 2,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 4, 0},
{5, 6, 8},
},
b: [][]float64{
{3, 6},
{4, 7},
{5, 8},
},
ans: [][]float64{
{-0.46875, 0.375},
{0.1875, 0.75},
{1.875, 3},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.Trans,
d: blas.Unit,
m: 3,
n: 2,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 4, 0},
{5, 6, 8},
},
b: [][]float64{
{3, 6},
{4, 7},
{5, 8},
},
ans: [][]float64{
{168, 267},
{-78, -123},
{15, 24},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.Trans,
d: blas.NonUnit,
m: 3,
n: 4,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 4, 0},
{5, 6, 8},
},
b: [][]float64{
{3, 6, 2, 3},
{4, 7, 4, 5},
{5, 8, 6, 7},
},
ans: [][]float64{
{-0.46875, 0.375, -2.0625, -1.78125},
{0.1875, 0.75, -0.375, -0.1875},
{1.875, 3, 2.25, 2.625},
},
},
{
s: blas.Left,
ul: blas.Lower,
tA: blas.Trans,
d: blas.Unit,
m: 3,
n: 4,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 4, 0},
{5, 6, 8},
},
b: [][]float64{
{3, 6, 2, 3},
{4, 7, 4, 5},
{5, 8, 6, 7},
},
ans: [][]float64{
{168, 267, 204, 237},
{-78, -123, -96, -111},
{15, 24, 18, 21},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 7},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{15, -2.4, -48.0 / 35},
{19.5, -3.3, -66.0 / 35},
{24, -4.2, -2.4},
{28.5, -5.1, -102.0 / 35},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{30, -57, 258},
{39, -75, 339},
{48, -93, 420},
{57, -111, 501},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 7},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{15, -2.4, -48.0 / 35},
{19.5, -3.3, -66.0 / 35},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.NoTrans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{30, -57, 258},
{39, -75, 339},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 5, 0},
{4, 6, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{4.2, 1.2, 4.5},
{5.775, 1.65, 5.625},
{7.35, 2.1, 6.75},
{8.925, 2.55, 7.875},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 5, 0},
{4, 6, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{435, -183, 36},
{543, -228, 45},
{651, -273, 54},
{759, -318, 63},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 5, 0},
{4, 6, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{4.2, 1.2, 4.5},
{5.775, 1.65, 5.625},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.NoTrans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 5, 0},
{4, 6, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{435, -183, 36},
{543, -228, 45},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.Trans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{4.2, 1.2, 4.5},
{5.775, 1.65, 5.625},
{7.35, 2.1, 6.75},
{8.925, 2.55, 7.875},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.Trans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{435, -183, 36},
{543, -228, 45},
{651, -273, 54},
{759, -318, 63},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.Trans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{4.2, 1.2, 4.5},
{5.775, 1.65, 5.625},
},
},
{
s: blas.Right,
ul: blas.Upper,
tA: blas.Trans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 3,
a: [][]float64{
{2, 3, 4},
{0, 5, 6},
{0, 0, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{435, -183, 36},
{543, -228, 45},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.Trans,
d: blas.NonUnit,
m: 4,
n: 3,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 5, 0},
{4, 6, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{15, -2.4, -1.2},
{19.5, -3.3, -1.65},
{24, -4.2, -2.1},
{28.5, -5.1, -2.55},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.Trans,
d: blas.Unit,
m: 4,
n: 3,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 5, 0},
{4, 6, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
{16, 17, 18},
{19, 20, 21},
},
ans: [][]float64{
{30, -57, 258},
{39, -75, 339},
{48, -93, 420},
{57, -111, 501},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.Trans,
d: blas.NonUnit,
m: 2,
n: 3,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 5, 0},
{4, 6, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{15, -2.4, -1.2},
{19.5, -3.3, -1.65},
},
},
{
s: blas.Right,
ul: blas.Lower,
tA: blas.Trans,
d: blas.Unit,
m: 2,
n: 3,
alpha: 3,
a: [][]float64{
{2, 0, 0},
{3, 5, 0},
{4, 6, 8},
},
b: [][]float64{
{10, 11, 12},
{13, 14, 15},
},
ans: [][]float64{
{30, -57, 258},
{39, -75, 339},
},
},
} {
aFlat := flatten(test.a)
bFlat := flatten(test.b)
ansFlat := flatten(test.ans)
var lda int
if test.s == blas.Left {
lda = test.m
} else {
lda = test.n
}
blasser.Dtrsm(test.s, test.ul, test.tA, test.d, test.m, test.n, test.alpha, aFlat, lda, bFlat, test.n)
if !floats.EqualApprox(ansFlat, bFlat, 1e-13) {
t.Errorf("Case %v: Want %v, got %v.", i, ansFlat, bFlat)
}
}
}