mirror of
https://github.com/gonum/gonum.git
synced 2025-11-01 02:52:49 +08:00
345 lines
6.4 KiB
Go
345 lines
6.4 KiB
Go
// 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)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|