diff --git a/test/libraries/cublas/level3.jl b/test/libraries/cublas/level3.jl index 459085c177..01f76b5b81 100644 --- a/test/libraries/cublas/level3.jl +++ b/test/libraries/cublas/level3.jl @@ -439,13 +439,14 @@ k = 13 end @testset "syr2k" begin alpha = rand(elty) + dalpha = CuArray{elty}([alpha]) A = rand(elty,m,k) B = rand(elty,m,k) # move to device d_A = CuArray(A) d_B = CuArray(B) C = alpha*(A*transpose(B) + B*transpose(A)) - d_C = CUBLAS.syr2k('U','N',alpha,d_A,d_B) + d_C = CUBLAS.syr2k('U','N',dalpha,d_A,d_B) # move back to host and compare C = triu(C) h_C = Array(d_C) @@ -455,12 +456,14 @@ k = 13 if elty <: Complex @testset "herk!" begin alpha = rand(elty) + dalpha = CuArray{real(elty)}([real(alpha)]) beta = rand(elty) + dbeta = CuArray{real(elty)}([real(beta)]) A = rand(elty,m,m) hA = A + A' d_A = CuArray(A) d_C = CuArray(hA) - CUBLAS.herk!('U','N',real(alpha),d_A,real(beta),d_C) + CUBLAS.herk!('U','N',dalpha,d_A,dbeta,d_C) C = real(alpha)*(A*A') + real(beta)*hA C = triu(C) # move to host and compare @@ -484,7 +487,9 @@ k = 13 elty2 = real(elty) # generate parameters α = rand(elty1) + dα = CuArray{elty2}(α) β = rand(elty2) + dβ = CuArray{elty2}(β) A = rand(elty,m,k) B = rand(elty,m,k) Bbad = rand(elty,m+1,k+1) @@ -496,7 +501,7 @@ k = 13 C = C + C' d_C = CuArray(C) C = α*(A*B') + conj(α)*(B*A') + β*C - CUBLAS.her2k!('U','N',α,d_A,d_B,β,d_C) + CUBLAS.her2k!('U','N',dα,d_A,d_B,dβ,d_C) # move back to host and compare C = triu(C) h_C = Array(d_C)