+++ /dev/null
-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)
- }
- }
-}
OSDN Git Service
