github.com/jingcheng-WU/gonum@v0.9.1-0.20210323123734-f1a2a11a8f7b/blas/testblas/zgemv.go (about)

     1  // Copyright ©2017 The Gonum Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package testblas
     6  
     7  import (
     8  	"testing"
     9  
    10  	"github.com/jingcheng-WU/gonum/blas"
    11  )
    12  
    13  type Zgemver interface {
    14  	Zgemv(trans blas.Transpose, m, n int, alpha complex128, a []complex128, lda int, x []complex128, incX int, beta complex128, y []complex128, incY int)
    15  }
    16  
    17  func ZgemvTest(t *testing.T, impl Zgemver) {
    18  	for tc, test := range []struct {
    19  		trans blas.Transpose
    20  		alpha complex128
    21  		a     []complex128
    22  		x     []complex128
    23  		beta  complex128
    24  		y     []complex128
    25  
    26  		want      []complex128
    27  		wantXNeg  []complex128
    28  		wantYNeg  []complex128
    29  		wantXYNeg []complex128
    30  	}{
    31  		{
    32  			trans: blas.NoTrans,
    33  			alpha: 1 + 2i,
    34  			beta:  3 + 4i,
    35  		},
    36  		{
    37  			trans: blas.NoTrans,
    38  			alpha: 1 + 2i,
    39  			a: []complex128{
    40  				9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
    41  				0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
    42  				10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
    43  				10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
    44  			},
    45  			x: []complex128{
    46  				4 - 9i,
    47  				8 + 5i,
    48  				-2 - 10i,
    49  				2 - 4i,
    50  				-6 + 6i,
    51  			},
    52  			beta: 3 + 4i,
    53  			y: []complex128{
    54  				-2 + 3i,
    55  				10 + 5i,
    56  				-8 - 5i,
    57  				-8 + 7i,
    58  			},
    59  			want: []complex128{
    60  				101 - 116i,
    61  				58 + 166i,
    62  				126 - 242i,
    63  				336 - 75i,
    64  			},
    65  			wantXNeg: []complex128{
    66  				98 + 128i,
    67  				374 - 252i,
    68  				-113 + 205i,
    69  				-60 - 312i,
    70  			},
    71  			wantYNeg: []complex128{
    72  				370 - 63i,
    73  				140 - 140i,
    74  				44 + 64i,
    75  				67 - 128i,
    76  			},
    77  			wantXYNeg: []complex128{
    78  				-26 - 300i,
    79  				-99 + 307i,
    80  				360 - 354i,
    81  				64 + 116i,
    82  			},
    83  		},
    84  		{
    85  			trans: blas.Trans,
    86  			alpha: 1 + 2i,
    87  			a: []complex128{
    88  				9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
    89  				0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
    90  				10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
    91  				10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
    92  			},
    93  			x: []complex128{
    94  				4 - 9i,
    95  				8 + 5i,
    96  				-2 - 10i,
    97  				2 - 4i,
    98  			},
    99  			beta: 3 + 4i,
   100  			y: []complex128{
   101  				8 - 6i,
   102  				-8 - 2i,
   103  				9 + 5i,
   104  				4 - 1i,
   105  				6 - 4i,
   106  			},
   107  			want: []complex128{
   108  				580 - 137i,
   109  				221 + 311i,
   110  				149 + 115i,
   111  				443 - 208i,
   112  				517 + 143i,
   113  			},
   114  			wantXNeg: []complex128{
   115  				387 + 152i,
   116  				109 - 433i,
   117  				225 - 53i,
   118  				-246 + 44i,
   119  				13 + 20i,
   120  			},
   121  			wantYNeg: []complex128{
   122  				531 + 145i,
   123  				411 - 259i,
   124  				149 + 115i,
   125  				253 + 362i,
   126  				566 - 139i,
   127  			},
   128  			wantXYNeg: []complex128{
   129  				27 + 22i,
   130  				-278 - 7i,
   131  				225 - 53i,
   132  				141 - 382i,
   133  				373 + 150i,
   134  			},
   135  		},
   136  		{
   137  			trans: blas.ConjTrans,
   138  			alpha: 1 + 2i,
   139  			a: []complex128{
   140  				9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
   141  				0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
   142  				10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
   143  				10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
   144  			},
   145  			x: []complex128{
   146  				4 - 9i,
   147  				8 + 5i,
   148  				-2 - 10i,
   149  				2 - 4i,
   150  			},
   151  			beta: 3 + 4i,
   152  			y: []complex128{
   153  				8 - 6i,
   154  				-8 - 2i,
   155  				9 + 5i,
   156  				4 - 1i,
   157  				6 - 4i,
   158  			},
   159  			want: []complex128{
   160  				472 - 133i,
   161  				-253 + 23i,
   162  				217 + 131i,
   163  				229 - 316i,
   164  				187 - 97i,
   165  			},
   166  			wantXNeg: []complex128{
   167  				289 + 276i,
   168  				499 + 47i,
   169  				237 + 91i,
   170  				54 + 504i,
   171  				251 + 196i,
   172  			},
   173  			wantYNeg: []complex128{
   174  				201 - 95i,
   175  				197 - 367i,
   176  				217 + 131i,
   177  				-221 + 74i,
   178  				458 - 135i,
   179  			},
   180  			wantXYNeg: []complex128{
   181  				265 + 198i,
   182  				22 + 453i,
   183  				237 + 91i,
   184  				531 + 98i,
   185  				275 + 274i,
   186  			},
   187  		},
   188  		{
   189  			trans: blas.ConjTrans,
   190  			alpha: 1 + 2i,
   191  			a: []complex128{
   192  				9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
   193  				0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
   194  				10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
   195  				10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
   196  			},
   197  			x: []complex128{
   198  				4 - 9i,
   199  				8 + 5i,
   200  				-2 - 10i,
   201  				2 - 4i,
   202  			},
   203  			beta: 0,
   204  			y: []complex128{
   205  				8 - 6i,
   206  				-8 - 2i,
   207  				9 + 5i,
   208  				4 - 1i,
   209  				6 - 4i,
   210  			},
   211  			want: []complex128{
   212  				424 - 147i,
   213  				-237 + 61i,
   214  				210 + 80i,
   215  				213 - 329i,
   216  				153 - 109i,
   217  			},
   218  			wantXNeg: []complex128{
   219  				241 + 262i,
   220  				515 + 85i,
   221  				230 + 40i,
   222  				38 + 491i,
   223  				217 + 184i,
   224  			},
   225  			wantYNeg: []complex128{
   226  				153 - 109i,
   227  				213 - 329i,
   228  				210 + 80i,
   229  				-237 + 61i,
   230  				424 - 147i,
   231  			},
   232  			wantXYNeg: []complex128{
   233  				217 + 184i,
   234  				38 + 491i,
   235  				230 + 40i,
   236  				515 + 85i,
   237  				241 + 262i,
   238  			},
   239  		},
   240  		{
   241  			trans: blas.ConjTrans,
   242  			alpha: 0,
   243  			a: []complex128{
   244  				9 + 5i, -2 + 6i, 5 + 1i, 9 + 2i, 10 + 4i,
   245  				0 - 7i, 9 - 9i, 5 + 3i, -8 - 1i, 7 - 7i,
   246  				10 - 7i, -1 + 3i, 2 + 2i, 7 + 6i, 9 + 1i,
   247  				10 + 0i, 8 - 6i, 4 - 6i, -2 - 10i, -5 + 0i,
   248  			},
   249  			x: []complex128{
   250  				4 - 9i,
   251  				8 + 5i,
   252  				-2 - 10i,
   253  				2 - 4i,
   254  			},
   255  			beta: 3 + 4i,
   256  			y: []complex128{
   257  				8 - 6i,
   258  				-8 - 2i,
   259  				9 + 5i,
   260  				4 - 1i,
   261  				6 - 4i,
   262  			},
   263  			want: []complex128{
   264  				48 + 14i,
   265  				-16 - 38i,
   266  				7 + 51i,
   267  				16 + 13i,
   268  				34 + 12i,
   269  			},
   270  			wantXNeg: []complex128{
   271  				48 + 14i,
   272  				-16 - 38i,
   273  				7 + 51i,
   274  				16 + 13i,
   275  				34 + 12i,
   276  			},
   277  			wantYNeg: []complex128{
   278  				48 + 14i,
   279  				-16 - 38i,
   280  				7 + 51i,
   281  				16 + 13i,
   282  				34 + 12i,
   283  			},
   284  			wantXYNeg: []complex128{
   285  				48 + 14i,
   286  				-16 - 38i,
   287  				7 + 51i,
   288  				16 + 13i,
   289  				34 + 12i,
   290  			},
   291  		},
   292  	} {
   293  		var m, n int
   294  		switch test.trans {
   295  		case blas.NoTrans:
   296  			m = len(test.y)
   297  			n = len(test.x)
   298  		case blas.Trans, blas.ConjTrans:
   299  			m = len(test.x)
   300  			n = len(test.y)
   301  		}
   302  		for _, incX := range []int{-11, -2, -1, 1, 2, 7} {
   303  			for _, incY := range []int{-11, -2, -1, 1, 2, 7} {
   304  				for _, lda := range []int{max(1, n), n + 11} {
   305  					alpha := test.alpha
   306  
   307  					a := makeZGeneral(test.a, m, n, lda)
   308  					aCopy := make([]complex128, len(a))
   309  					copy(aCopy, a)
   310  
   311  					x := makeZVector(test.x, incX)
   312  					xCopy := make([]complex128, len(x))
   313  					copy(xCopy, x)
   314  
   315  					y := makeZVector(test.y, incY)
   316  
   317  					impl.Zgemv(test.trans, m, n, alpha, a, lda, x, incX, test.beta, y, incY)
   318  
   319  					if !zsame(x, xCopy) {
   320  						t.Errorf("Case %v (incX=%v,incY=%v,lda=%v): unexpected modification of x", tc, incX, incY, lda)
   321  					}
   322  					if !zsame(a, aCopy) {
   323  						t.Errorf("Case %v (incX=%v,incY=%v,lda=%v): unexpected modification of A", tc, incX, incY, lda)
   324  					}
   325  
   326  					var want []complex128
   327  					switch {
   328  					case incX > 0 && incY > 0:
   329  						want = makeZVector(test.want, incY)
   330  					case incX < 0 && incY > 0:
   331  						want = makeZVector(test.wantXNeg, incY)
   332  					case incX > 0 && incY < 0:
   333  						want = makeZVector(test.wantYNeg, incY)
   334  					default:
   335  						want = makeZVector(test.wantXYNeg, incY)
   336  					}
   337  					if !zsame(y, want) {
   338  						t.Errorf("Case %v (incX=%v,incY=%v,lda=%v): unexpected result\nwant %v\ngot  %v", tc, incX, incY, lda, want, y)
   339  					}
   340  				}
   341  			}
   342  		}
   343  	}
   344  }