From 31a82eac110437d10b74d234dff4ee5ed16f43e1 Mon Sep 17 00:00:00 2001 From: Katharine Hyatt Date: Tue, 27 Oct 2026 13:31:24 -0400 Subject: [PATCH] Fix sizes of allocated U, V for Jacobi on CUDA --- ext/MatrixAlgebraKitCUDAExt/yacusolver.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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))