| // 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" |
| ) |
| |
| type DgemvCase struct { |
| Name string |
| m int |
| n int |
| A [][]float64 |
| tA blas.Transpose |
| x []float64 |
| incX int |
| y []float64 |
| incY int |
| |
| Subcases []DgemvSubcase |
| } |
| |
| type DgemvSubcase struct { |
| mulXNeg1 bool |
| mulYNeg1 bool |
| alpha float64 |
| beta float64 |
| ans []float64 |
| } |
| |
| var DgemvCases = []DgemvCase{ |
| { |
| Name: "M_gt_N_Inc1_NoTrans", |
| tA: blas.NoTrans, |
| m: 5, |
| n: 3, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| {1, 1, 2}, |
| {9, 2, 5}, |
| }, |
| incX: 1, |
| incY: 1, |
| x: []float64{1, 2, 3}, |
| y: []float64{7, 8, 9, 10, 11}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 0, |
| beta: 0, |
| ans: []float64{0, 0, 0, 0, 0}, |
| }, |
| { |
| alpha: 0, |
| beta: 1, |
| ans: []float64{7, 8, 9, 10, 11}, |
| }, |
| { |
| alpha: 1, |
| beta: 0, |
| ans: []float64{40.8, 43.9, 33, 9, 28}, |
| }, |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{284.4, 303.2, 210, 12, 158}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_gt_N_Inc1_Trans", |
| tA: blas.Trans, |
| m: 5, |
| n: 3, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| {1, 1, 2}, |
| {9, 2, 5}, |
| }, |
| incX: 1, |
| incY: 1, |
| x: []float64{1, 2, 3, -4, 5}, |
| y: []float64{7, 8, 9}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 0, |
| beta: 0, |
| ans: []float64{0, 0, 0}, |
| }, |
| { |
| alpha: 0, |
| beta: 1, |
| ans: []float64{7, 8, 9}, |
| }, |
| { |
| alpha: 1, |
| beta: 0, |
| ans: []float64{94.3, 40.2, 52.3}, |
| }, |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{712.4, 273.6, 364.4}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_eq_N_Inc1_NoTrans", |
| tA: blas.NoTrans, |
| m: 3, |
| n: 3, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| }, |
| incX: 1, |
| incY: 1, |
| x: []float64{1, 2, 3}, |
| y: []float64{7, 2, 2}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 0, |
| beta: 0, |
| ans: []float64{0, 0, 0}, |
| }, |
| { |
| alpha: 0, |
| beta: 1, |
| ans: []float64{7, 2, 2}, |
| }, |
| { |
| alpha: 1, |
| beta: 0, |
| ans: []float64{40.8, 43.9, 33}, |
| }, |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{40.8*8 - 6*7, 43.9*8 - 6*2, 33*8 - 6*2}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_eq_N_Inc1_Trans", |
| tA: blas.Trans, |
| m: 3, |
| n: 3, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| }, |
| incX: 1, |
| incY: 1, |
| x: []float64{1, 2, 3}, |
| y: []float64{7, 2, 2}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{384.4, 261.6, 270.4}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_lt_N_Inc1_NoTrans", |
| tA: blas.NoTrans, |
| m: 3, |
| n: 5, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1, 10, 7}, |
| {9.6, 3.5, 9.1, -2, 9}, |
| {10, 7, 3, 1, -5}, |
| }, |
| incX: 1, |
| incY: 1, |
| x: []float64{1, 2, 3, -7.6, 8.1}, |
| y: []float64{7, 2, 2}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 0, |
| beta: 0, |
| ans: []float64{0, 0, 0}, |
| }, |
| { |
| alpha: 0, |
| beta: 1, |
| ans: []float64{7, 2, 2}, |
| }, |
| { |
| alpha: 1, |
| beta: 0, |
| ans: []float64{21.5, 132, -15.1}, |
| }, |
| |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{21.5*8 - 6*7, 132*8 - 6*2, -15.1*8 - 6*2}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_lt_N_Inc1_Trans", |
| tA: blas.Trans, |
| m: 3, |
| n: 5, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1, 10, 7}, |
| {9.6, 3.5, 9.1, -2, 9}, |
| {10, 7, 3, 1, -5}, |
| }, |
| incX: 1, |
| incY: 1, |
| x: []float64{1, 2, 3}, |
| y: []float64{7, 2, 2, -3, 5}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{384.4, 261.6, 270.4, 90, 50}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_gt_N_Part1_NoTrans", |
| tA: blas.NoTrans, |
| m: 5, |
| n: 3, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| {1, 1, 2}, |
| {9, 2, 5}, |
| }, |
| incX: 1, |
| incY: 2, |
| x: []float64{1, 2, 3}, |
| y: []float64{7, 100, 8, 101, 9, 102, 10, 103, 11}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 0, |
| beta: 0, |
| ans: []float64{0, 100, 0, 101, 0, 102, 0, 103, 0}, |
| }, |
| { |
| alpha: 0, |
| beta: 1, |
| ans: []float64{7, 100, 8, 101, 9, 102, 10, 103, 11}, |
| }, |
| { |
| alpha: 1, |
| beta: 0, |
| ans: []float64{40.8, 100, 43.9, 101, 33, 102, 9, 103, 28}, |
| }, |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{284.4, 100, 303.2, 101, 210, 102, 12, 103, 158}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_gt_N_Part1_Trans", |
| tA: blas.Trans, |
| m: 5, |
| n: 3, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| {1, 1, 2}, |
| {9, 2, 5}, |
| }, |
| incX: 1, |
| incY: 2, |
| x: []float64{1, 2, 3, -4, 5}, |
| y: []float64{7, 100, 8, 101, 9}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 0, |
| beta: 0, |
| ans: []float64{0, 100, 0, 101, 0}, |
| }, |
| { |
| alpha: 0, |
| beta: 1, |
| ans: []float64{7, 100, 8, 101, 9}, |
| }, |
| { |
| alpha: 1, |
| beta: 0, |
| ans: []float64{94.3, 100, 40.2, 101, 52.3}, |
| }, |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{712.4, 100, 273.6, 101, 364.4}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_gt_N_IncNot1_NoTrans", |
| tA: blas.NoTrans, |
| m: 5, |
| n: 3, |
| |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| {1, 1, 2}, |
| {9, 2, 5}, |
| }, |
| incX: 2, |
| incY: 3, |
| x: []float64{1, 15, 2, 150, 3}, |
| y: []float64{7, 2, 6, 8, -4, -5, 9, 1, 1, 10, 19, 22, 11}, |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{284.4, 2, 6, 303.2, -4, -5, 210, 1, 1, 12, 19, 22, 158}, |
| }, |
| { |
| mulXNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{220.4, 2, 6, 311.2, -4, -5, 322, 1, 1, -4, 19, 22, 222}, |
| }, |
| { |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{182, 2, 6, 24, -4, -5, 210, 1, 1, 291.2, 19, 22, 260.4}, |
| }, |
| { |
| mulXNeg1: true, |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{246, 2, 6, 8, -4, -5, 322, 1, 1, 299.2, 19, 22, 196.4}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_gt_N_IncNot1_Trans", |
| tA: blas.Trans, |
| m: 5, |
| n: 3, |
| |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| {1, 1, 2}, |
| {9, 2, 5}, |
| }, |
| incX: 2, |
| incY: 3, |
| x: []float64{1, 15, 2, 150, 3, 8, -3, 6, 5}, |
| y: []float64{7, 2, 6, 8, -4, -5, 9}, |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{720.4, 2, 6, 281.6, -4, -5, 380.4}, |
| }, |
| { |
| mulXNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{219.6, 2, 6, 316, -4, -5, 195.6}, |
| }, |
| { |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{392.4, 2, 6, 281.6, -4, -5, 708.4}, |
| }, |
| { |
| mulXNeg1: true, |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{207.6, 2, 6, 316, -4, -5, 207.6}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_eq_N_IncNot1_NoTrans", |
| tA: blas.NoTrans, |
| m: 3, |
| n: 3, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| }, |
| incX: 2, |
| incY: 3, |
| x: []float64{1, 15, 2, 150, 3}, |
| y: []float64{7, 2, 6, 8, -4, -5, 9}, |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{284.4, 2, 6, 303.2, -4, -5, 210}, |
| }, |
| { |
| mulXNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{220.4, 2, 6, 311.2, -4, -5, 322}, |
| }, |
| { |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{222, 2, 6, 303.2, -4, -5, 272.4}, |
| }, |
| { |
| mulXNeg1: true, |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{334, 2, 6, 311.2, -4, -5, 208.4}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_eq_N_IncNot1_Trans", |
| tA: blas.Trans, |
| m: 3, |
| n: 3, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1}, |
| {9.6, 3.5, 9.1}, |
| {10, 7, 3}, |
| }, |
| incX: 2, |
| incY: 3, |
| x: []float64{1, 15, 2, 150, 3}, |
| y: []float64{7, 2, 6, 8, -4, -5, 9}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{384.4, 2, 6, 225.6, -4, -5, 228.4}, |
| }, |
| { |
| mulXNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{290, 2, 6, 212.8, -4, -5, 310}, |
| }, |
| { |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{240.4, 2, 6, 225.6, -4, -5, 372.4}, |
| }, |
| { |
| mulXNeg1: true, |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{322, 2, 6, 212.8, -4, -5, 278}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_lt_N_IncNot1_NoTrans", |
| tA: blas.NoTrans, |
| m: 3, |
| n: 5, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1, 10, 11}, |
| {9.6, 3.5, 9.1, -3, -2}, |
| {10, 7, 3, -7, -4}, |
| }, |
| incX: 2, |
| incY: 3, |
| x: []float64{1, 15, 2, 150, 3, -2, -4, 8, -9}, |
| y: []float64{7, 2, 6, 8, -4, -5, 9}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{-827.6, 2, 6, 543.2, -4, -5, 722}, |
| }, |
| { |
| mulXNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{-93.2, 2, 6, -696.8, -4, -5, -1070}, |
| }, |
| { |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{734, 2, 6, 543.2, -4, -5, -839.6}, |
| }, |
| { |
| mulXNeg1: true, |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{-1058, 2, 6, -696.8, -4, -5, -105.2}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_lt_N_IncNot1_Trans", |
| tA: blas.Trans, |
| m: 3, |
| n: 5, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1, 10, 11}, |
| {9.6, 3.5, 9.1, -3, -2}, |
| {10, 7, 3, -7, -4}, |
| }, |
| incX: 2, |
| incY: 3, |
| x: []float64{1, 15, 2, 150, 3}, |
| y: []float64{7, 2, 6, 8, -4, -5, 9, -4, -1, -9, 1, 1, 2}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{384.4, 2, 6, 225.6, -4, -5, 228.4, -4, -1, -82, 1, 1, -52}, |
| }, |
| { |
| mulXNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{290, 2, 6, 212.8, -4, -5, 310, -4, -1, 190, 1, 1, 188}, |
| }, |
| { |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{-82, 2, 6, -184, -4, -5, 228.4, -4, -1, 327.6, 1, 1, 414.4}, |
| }, |
| { |
| mulXNeg1: true, |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{158, 2, 6, 88, -4, -5, 310, -4, -1, 314.8, 1, 1, 320}, |
| }, |
| }, |
| }, |
| { |
| Name: "M_eq_N_Lg_IncNot1_Trans", |
| tA: blas.Trans, |
| m: 7, |
| n: 7, |
| A: [][]float64{ |
| {4.1, 6.2, 8.1, 2.5, 3.3, 7.4, 9.3}, |
| {9.6, 3.5, 9.1, 1.2, 5.4, 4.8, 8.7}, |
| {10, 7, 3, 2, 4, 1, 12}, |
| {9.6, 3.5, 9.1, 1.2, 5.4, 4.8, 8.7}, |
| {4.1, 6.2, 8.1, 2.5, 3.3, 7.4, 9.3}, |
| {10, 7, 3, 2, 4, 1, 12}, |
| {9.6, 3.5, 9.1, 1.2, 5.4, 4.8, 8.7}, |
| }, |
| incX: 2, |
| incY: 3, |
| x: []float64{1, 105, 2, 150, 3, 200, 4, 300, 5, 600, 6, 700, 7}, |
| y: []float64{7, 200, 600, 8, -400, -500, 9, 600, 700, 10, 500, 400, 11, 200, 300, 12, 100, 200, 13, 300, 400, 14}, |
| |
| Subcases: []DgemvSubcase{ |
| { |
| alpha: 8, |
| beta: -6, |
| ans: []float64{1873.1999999999998, 200, 600, 1117.6, -400, -500, 1497.1999999999998, 600, 700, 328.8, 500, 400, 942, 200, 300, 854.4000000000001, 100, 200, 2137.2, 300, 400, 14}, |
| }, |
| { |
| mulXNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{1690.8, 200, 600, 1148, -400, -500, 1562.8, 600, 700, 357.6, 500, 400, 897.2, 200, 300, 998.4, 100, 200, 2103.6000000000004, 300, 400, 14}, |
| }, |
| { |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{2173.2, 200, 600, 878.4000000000001, -400, -500, 954, 600, 700, 328.8, 500, 400, 1485.1999999999998, 200, 300, 1093.6, 100, 200, 1837.1999999999998, 300, 400, 14}, |
| }, |
| { |
| mulXNeg1: true, |
| mulYNeg1: true, |
| alpha: 8, |
| beta: -6, |
| ans: []float64{2139.6, 200, 600, 1022.4, -400, -500, 909.2, 600, 700, 357.6, 500, 400, 1550.8, 200, 300, 1124, 100, 200, 1654.8, 300, 400, 14}, |
| }, |
| }, |
| }, |
| |
| // TODO: A can be longer than mxn. Add cases where it is longer |
| // TODO: x and y can also be longer. Add tests for these |
| // TODO: Add tests for dimension mismatch |
| // TODO: Add places with a "submatrix view", where lda != m |
| } |
| |
| type Dgemver interface { |
| Dgemv(tA blas.Transpose, m, n int, alpha float64, a []float64, lda int, x []float64, incX int, beta float64, y []float64, incY int) |
| } |
| |
| func DgemvTest(t *testing.T, blasser Dgemver) { |
| for _, test := range DgemvCases { |
| for i, cas := range test.Subcases { |
| // Test that it passes with row-major |
| dgemvcomp(t, test, cas, i, blasser) |
| |
| // Test the bad inputs |
| dgemvbad(t, test, cas, i, blasser) |
| } |
| } |
| } |
| |
| func dgemvcomp(t *testing.T, test DgemvCase, cas DgemvSubcase, i int, blasser Dgemver) { |
| x := sliceCopy(test.x) |
| y := sliceCopy(test.y) |
| a := sliceOfSliceCopy(test.A) |
| aFlat := flatten(a) |
| |
| lda := test.n |
| |
| incX := test.incX |
| if cas.mulXNeg1 { |
| incX *= -1 |
| } |
| incY := test.incY |
| if cas.mulYNeg1 { |
| incY *= -1 |
| } |
| |
| f := func() { |
| blasser.Dgemv(test.tA, test.m, test.n, cas.alpha, aFlat, lda, x, incX, cas.beta, y, incY) |
| } |
| if panics(f) { |
| t.Errorf("Test %v case %v: unexpected panic", test.Name, i) |
| if throwPanic { |
| blasser.Dgemv(test.tA, test.m, test.n, cas.alpha, aFlat, lda, x, incX, cas.beta, y, incY) |
| } |
| return |
| } |
| // Check that x and a are unchanged |
| if !dSliceEqual(x, test.x) { |
| t.Errorf("Test %v, case %v: x modified during call", test.Name, i) |
| } |
| aFlat2 := flatten(sliceOfSliceCopy(test.A)) |
| if !dSliceEqual(aFlat2, aFlat) { |
| t.Errorf("Test %v, case %v: a modified during call", test.Name, i) |
| } |
| |
| // Check that the answer matches |
| if !dSliceTolEqual(cas.ans, y) { |
| t.Errorf("Test %v, case %v: answer mismatch: Expected %v, Found %v", test.Name, i, cas.ans, y) |
| } |
| } |
| |
| func dgemvbad(t *testing.T, test DgemvCase, cas DgemvSubcase, i int, blasser Dgemver) { |
| x := sliceCopy(test.x) |
| y := sliceCopy(test.y) |
| a := sliceOfSliceCopy(test.A) |
| aFlatRow := flatten(a) |
| ldaRow := test.n |
| |
| f := func() { |
| blasser.Dgemv('X', test.m, test.n, cas.alpha, aFlatRow, ldaRow, x, test.incX, cas.beta, y, test.incY) |
| } |
| if !panics(f) { |
| t.Errorf("Test %v case %v: no panic for bad transpose", test.Name, i) |
| } |
| f = func() { |
| blasser.Dgemv(test.tA, -2, test.n, cas.alpha, aFlatRow, ldaRow, x, test.incX, cas.beta, y, test.incY) |
| } |
| if !panics(f) { |
| t.Errorf("Test %v case %v: no panic for m negative", test.Name, i) |
| } |
| f = func() { |
| blasser.Dgemv(test.tA, test.m, -4, cas.alpha, aFlatRow, ldaRow, x, test.incX, cas.beta, y, test.incY) |
| } |
| if !panics(f) { |
| t.Errorf("Test %v case %v: no panic for n negative", test.Name, i) |
| } |
| f = func() { |
| blasser.Dgemv(test.tA, test.m, test.n, cas.alpha, aFlatRow, ldaRow, x, 0, cas.beta, y, test.incY) |
| } |
| if !panics(f) { |
| t.Errorf("Test %v case %v: no panic for incX zero", test.Name, i) |
| } |
| f = func() { |
| blasser.Dgemv(test.tA, test.m, test.n, cas.alpha, aFlatRow, ldaRow, x, test.incX, cas.beta, y, 0) |
| } |
| if !panics(f) { |
| t.Errorf("Test %v case %v: no panic for incY zero", test.Name, i) |
| } |
| f = func() { |
| blasser.Dgemv(test.tA, test.m, test.n, cas.alpha, aFlatRow, ldaRow-1, x, test.incX, cas.beta, y, test.incY) |
| } |
| if !panics(f) { |
| t.Errorf("Test %v case %v: no panic for lda too small row major", test.Name, i) |
| } |
| } |