diff --git a/ext/MatrixAlgebraKitCUDAExt/yacusolver.jl b/ext/MatrixAlgebraKitCUDAExt/yacusolver.jl index 934606c49..c91d1c66a 100644 --- a/ext/MatrixAlgebraKitCUDAExt/yacusolver.jl +++ b/ext/MatrixAlgebraKitCUDAExt/yacusolver.jl @@ -232,8 +232,8 @@ for (bname, fname, elty, relty) in length(S) == minmn || throw(DimensionMismatch("length mismatch between A and S")) - Ṽ = (jobz == 'V') ? similar(Vᴴ') : similar(Vᴴ, (n, minmn)) - Ũ = (jobz == 'V') ? U : similar(U, (m, minmn)) + Ṽ = (jobz == 'V') ? similar(Vᴴ') : similar(Vᴴ, (n, n)) + Ũ = (jobz == 'V') ? U : similar(U, (m, m)) lda = max(1, stride(A, 2)) ldu = max(1, stride(Ũ, 2)) ldv = max(1, stride(Ṽ, 2))