+++ /dev/null
-// Copyright ©2017 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 Zgemver interface {
- Zgemv(trans blas.Transpose, m, n int, alpha complex128, a []complex128, lda int, x []complex128, incX int, beta complex128, y []complex128, incY int)
-}
-
-func ZgemvTest(t *testing.T, impl Zgemver) {
- for tc, test := range []struct {
- trans blas.Transpose
- alpha complex128
- a []complex128
- x []complex128
- beta complex128
- y []complex128
-
- want []complex128
- wantXNeg []complex128
- wantYNeg []complex128
- wantXYNeg []complex128
- }{
- {
- trans: blas.NoTrans,
- alpha: 1 + 2i,
- beta: 3 + 4i,
- },
- {
- trans: blas.NoTrans,
- alpha: 1 + 2i,
- a: []complex128{
- 9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
- 0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
- 10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
- 10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
- },
- x: []complex128{
- 4 - 9i,
- 8 + 5i,
- -2 - 10i,
- 2 - 4i,
- -6 + 6i,
- },
- beta: 3 + 4i,
- y: []complex128{
- -2 + 3i,
- 10 + 5i,
- -8 - 5i,
- -8 + 7i,
- },
- want: []complex128{
- 101 - 116i,
- 58 + 166i,
- 126 - 242i,
- 336 - 75i,
- },
- wantXNeg: []complex128{
- 98 + 128i,
- 374 - 252i,
- -113 + 205i,
- -60 - 312i,
- },
- wantYNeg: []complex128{
- 370 - 63i,
- 140 - 140i,
- 44 + 64i,
- 67 - 128i,
- },
- wantXYNeg: []complex128{
- -26 - 300i,
- -99 + 307i,
- 360 - 354i,
- 64 + 116i,
- },
- },
- {
- trans: blas.Trans,
- alpha: 1 + 2i,
- a: []complex128{
- 9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
- 0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
- 10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
- 10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
- },
- x: []complex128{
- 4 - 9i,
- 8 + 5i,
- -2 - 10i,
- 2 - 4i,
- },
- beta: 3 + 4i,
- y: []complex128{
- 8 - 6i,
- -8 - 2i,
- 9 + 5i,
- 4 - 1i,
- 6 - 4i,
- },
- want: []complex128{
- 580 - 137i,
- 221 + 311i,
- 149 + 115i,
- 443 - 208i,
- 517 + 143i,
- },
- wantXNeg: []complex128{
- 387 + 152i,
- 109 - 433i,
- 225 - 53i,
- -246 + 44i,
- 13 + 20i,
- },
- wantYNeg: []complex128{
- 531 + 145i,
- 411 - 259i,
- 149 + 115i,
- 253 + 362i,
- 566 - 139i,
- },
- wantXYNeg: []complex128{
- 27 + 22i,
- -278 - 7i,
- 225 - 53i,
- 141 - 382i,
- 373 + 150i,
- },
- },
- {
- trans: blas.ConjTrans,
- alpha: 1 + 2i,
- a: []complex128{
- 9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
- 0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
- 10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
- 10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
- },
- x: []complex128{
- 4 - 9i,
- 8 + 5i,
- -2 - 10i,
- 2 - 4i,
- },
- beta: 3 + 4i,
- y: []complex128{
- 8 - 6i,
- -8 - 2i,
- 9 + 5i,
- 4 - 1i,
- 6 - 4i,
- },
- want: []complex128{
- 472 - 133i,
- -253 + 23i,
- 217 + 131i,
- 229 - 316i,
- 187 - 97i,
- },
- wantXNeg: []complex128{
- 289 + 276i,
- 499 + 47i,
- 237 + 91i,
- 54 + 504i,
- 251 + 196i,
- },
- wantYNeg: []complex128{
- 201 - 95i,
- 197 - 367i,
- 217 + 131i,
- -221 + 74i,
- 458 - 135i,
- },
- wantXYNeg: []complex128{
- 265 + 198i,
- 22 + 453i,
- 237 + 91i,
- 531 + 98i,
- 275 + 274i,
- },
- },
- {
- trans: blas.ConjTrans,
- alpha: 1 + 2i,
- a: []complex128{
- 9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
- 0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
- 10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
- 10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
- },
- x: []complex128{
- 4 - 9i,
- 8 + 5i,
- -2 - 10i,
- 2 - 4i,
- },
- beta: 0,
- y: []complex128{
- 8 - 6i,
- -8 - 2i,
- 9 + 5i,
- 4 - 1i,
- 6 - 4i,
- },
- want: []complex128{
- 424 - 147i,
- -237 + 61i,
- 210 + 80i,
- 213 - 329i,
- 153 - 109i,
- },
- wantXNeg: []complex128{
- 241 + 262i,
- 515 + 85i,
- 230 + 40i,
- 38 + 491i,
- 217 + 184i,
- },
- wantYNeg: []complex128{
- 153 - 109i,
- 213 - 329i,
- 210 + 80i,
- -237 + 61i,
- 424 - 147i,
- },
- wantXYNeg: []complex128{
- 217 + 184i,
- 38 + 491i,
- 230 + 40i,
- 515 + 85i,
- 241 + 262i,
- },
- },
- {
- trans: blas.ConjTrans,
- alpha: 0,
- a: []complex128{
- 9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
- 0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
- 10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
- 10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
- },
- x: []complex128{
- 4 - 9i,
- 8 + 5i,
- -2 - 10i,
- 2 - 4i,
- },
- beta: 3 + 4i,
- y: []complex128{
- 8 - 6i,
- -8 - 2i,
- 9 + 5i,
- 4 - 1i,
- 6 - 4i,
- },
- want: []complex128{
- 48 + 14i,
- -16 - 38i,
- 7 + 51i,
- 16 + 13i,
- 34 + 12i,
- },
- wantXNeg: []complex128{
- 48 + 14i,
- -16 - 38i,
- 7 + 51i,
- 16 + 13i,
- 34 + 12i,
- },
- wantYNeg: []complex128{
- 48 + 14i,
- -16 - 38i,
- 7 + 51i,
- 16 + 13i,
- 34 + 12i,
- },
- wantXYNeg: []complex128{
- 48 + 14i,
- -16 - 38i,
- 7 + 51i,
- 16 + 13i,
- 34 + 12i,
- },
- },
- } {
- var m, n int
- switch test.trans {
- case blas.NoTrans:
- m = len(test.y)
- n = len(test.x)
- case blas.Trans, blas.ConjTrans:
- m = len(test.x)
- n = len(test.y)
- }
- for _, incX := range []int{-11, -2, -1, 1, 2, 7} {
- for _, incY := range []int{-11, -2, -1, 1, 2, 7} {
- for _, lda := range []int{max(1, n), n + 11} {
- alpha := test.alpha
-
- a := makeZGeneral(test.a, m, n, lda)
- aCopy := make([]complex128, len(a))
- copy(aCopy, a)
-
- x := makeZVector(test.x, incX)
- xCopy := make([]complex128, len(x))
- copy(xCopy, x)
-
- y := makeZVector(test.y, incY)
-
- impl.Zgemv(test.trans, m, n, alpha, a, lda, x, incX, test.beta, y, incY)
-
- if !zsame(x, xCopy) {
- t.Errorf("Case %v (incX=%v,incY=%v,lda=%v): unexpected modification of x", tc, incX, incY, lda)
- }
- if !zsame(a, aCopy) {
- t.Errorf("Case %v (incX=%v,incY=%v,lda=%v): unexpected modification of A", tc, incX, incY, lda)
- }
-
- var want []complex128
- switch {
- case incX > 0 && incY > 0:
- want = makeZVector(test.want, incY)
- case incX < 0 && incY > 0:
- want = makeZVector(test.wantXNeg, incY)
- case incX > 0 && incY < 0:
- want = makeZVector(test.wantYNeg, incY)
- default:
- want = makeZVector(test.wantXYNeg, incY)
- }
- if !zsame(y, want) {
- t.Errorf("Case %v (incX=%v,incY=%v,lda=%v): unexpected result\nwant %v\ngot %v", tc, incX, incY, lda, want, y)
- }
- }
- }
- }
- }
-}
OSDN Git Service
