1
- using BatchedBLAS, Test, CUDA, LinearAlgebra, SymmetricFormats
1
+ using BatchedBLAS, Test, CUDA, LinearAlgebra, LinearAlgebra . BLAS, SymmetricFormats
2
2
3
3
L= 16 ; N= 4
4
4
A = reshape (1.0 : L* L* N, L,L,N);
@@ -43,7 +43,7 @@ test_types = (Float64, Int32)
43
43
for i= 1 : N
44
44
thisalpha = Talpha<: Vector ? alpha_vector[i] : alpha_scalar
45
45
thisbeta = Tbeta<: Vector ? beta_vector[i] : beta_scalar
46
- BLAS . gemv! (trans, thisalpha, A[:,:,i], x[:,i], thisbeta, @view y_cpu[:,i])
46
+ gemv! (trans, thisalpha, A[:,:,i], x[:,i], thisbeta, @view y_cpu[:,i])
47
47
end
48
48
y_gpu= copy (cuy);
49
49
thisalpha = Talpha<: Vector ? cualpha_vector : alpha_scalar
@@ -57,7 +57,7 @@ test_types = (Float64, Int32)
57
57
for i= 1 : N
58
58
thisalpha = Talpha<: Vector ? alpha_vector[i] : alpha_scalar
59
59
thisbeta = Tbeta<: Vector ? beta_vector[i] : beta_scalar
60
- BLAS . symv! (uplo, thisalpha, A[:,:,i], x[:,i], thisbeta, @view y_cpu[:,i])
60
+ symv! (uplo, thisalpha, A[:,:,i], x[:,i], thisbeta, @view y_cpu[:,i])
61
61
end
62
62
y_gpu= copy (cuy);
63
63
thisalpha = Talpha<: Vector ? cualpha_vector : alpha_scalar
@@ -69,7 +69,7 @@ test_types = (Float64, Int32)
69
69
for i= 1 : N
70
70
thisalpha = Talpha<: Vector ? alpha_vector[i] : alpha_scalar
71
71
thisbeta = Tbeta<: Vector ? beta_vector[i] : beta_scalar
72
- BLAS . spmv! (uplo, thisalpha, APU[:,i], x[:,i], thisbeta, @view y_cpu[:,i])
72
+ spmv! (uplo, thisalpha, APU[:,i], x[:,i], thisbeta, @view y_cpu[:,i])
73
73
end
74
74
y_gpu= copy (cuy);
75
75
thisalpha = Talpha<: Vector ? cualpha_vector : alpha_scalar
@@ -82,7 +82,7 @@ test_types = (Float64, Int32)
82
82
A_cpu= copy (A);
83
83
for i= 1 : N
84
84
thisalpha = Talpha<: Vector ? alpha_vector[i] : alpha_scalar
85
- BLAS . ger! (thisalpha, x[:,i], y[:,i], @view A_cpu[:,:,i])
85
+ ger! (thisalpha, x[:,i], y[:,i], @view A_cpu[:,:,i])
86
86
end
87
87
A_gpu= copy (cuA);
88
88
thisalpha = Talpha<: Vector ? cualpha_vector : alpha_scalar
@@ -94,7 +94,7 @@ test_types = (Float64, Int32)
94
94
A_cpu= copy (A);
95
95
for i= 1 : N
96
96
thisalpha = Talpha<: Vector ? alpha_vector[i] : alpha_scalar
97
- BLAS . syr! (uplo, thisalpha, x[:,i], @view A_cpu[:,:,i])
97
+ syr! (uplo, thisalpha, x[:,i], @view A_cpu[:,:,i])
98
98
end
99
99
A_gpu= copy (cuA);
100
100
thisalpha = Talpha<: Vector ? cualpha_vector : alpha_scalar
0 commit comments