| // 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}, |
| }, |
| }, |
| { |
| s: blas.Right, |
| ul: blas.Lower, |
| tA: blas.Trans, |
| d: blas.Unit, |
| m: 2, |
| n: 3, |
| alpha: 0, |
| a: [][]float64{ |
| {1, 0, 0}, |
| {2, 4, 0}, |
| {3, 5, 6}, |
| }, |
| b: [][]float64{ |
| {10, 11, 12}, |
| {13, 14, 15}, |
| }, |
| ans: [][]float64{ |
| {0, 0, 0}, |
| {0, 0, 0}, |
| }, |
| }, |
| } { |
| 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) |
| } |
| } |
| } |
