From 98d868d59129b9d2e60569690d86dffce1b75807 Mon Sep 17 00:00:00 2001 From: Alexis Montoison Date: Sun, 31 Mar 2024 22:17:34 -0400 Subject: [PATCH] [CUSOLVER] Fix the dispatch for syevd! end heevd! --- lib/cusolver/dense.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/cusolver/dense.jl b/lib/cusolver/dense.jl index 9f218d4546..9f572cf27d 100644 --- a/lib/cusolver/dense.jl +++ b/lib/cusolver/dense.jl @@ -908,18 +908,19 @@ for elty in (:Float32, :Float64, :ComplexF32, :ComplexF64) LinearAlgebra.LAPACK.orgqr!(A::StridedCuMatrix{$elty}, tau::CuVector{$elty}) = CUSOLVER.orgqr!(A, tau) LinearAlgebra.LAPACK.gebrd!(A::StridedCuMatrix{$elty}) = CUSOLVER.gebrd!(A) LinearAlgebra.LAPACK.gesvd!(jobu::Char, jobvt::Char, A::StridedCuMatrix{$elty}) = CUSOLVER.gesvd!(jobu, jobvt, A) - LinearAlgebra.LAPACK.syev!(jobz::Char, uplo::Char, A::StridedCuMatrix{$elty}) = CUSOLVER.syevd!(jobz, uplo, A) end end for elty in (:Float32, :Float64) @eval begin + LinearAlgebra.LAPACK.syevd!(jobz::Char, uplo::Char, A::StridedCuMatrix{$elty}) = CUSOLVER.syevd!(jobz, uplo, A) LinearAlgebra.LAPACK.sygvd!(itype::Int, jobz::Char, uplo::Char, A::StridedCuMatrix{$elty}, B::StridedCuMatrix{$elty}) = CUSOLVER.sygvd!(itype, jobz, uplo, A, B) end end for elty in (:ComplexF32, :ComplexF64) @eval begin + LinearAlgebra.LAPACK.syevd!(jobz::Char, uplo::Char, A::StridedCuMatrix{$elty}) = CUSOLVER.heevd!(jobz, uplo, A) LinearAlgebra.LAPACK.sygvd!(itype::Int, jobz::Char, uplo::Char, A::StridedCuMatrix{$elty}, B::StridedCuMatrix{$elty}) = CUSOLVER.hegvd!(itype, jobz, uplo, A, B) end end