From 0873eb155fe5e3498edda70dc6a9f363eeaacda1 Mon Sep 17 00:00:00 2001 From: Tim Gymnich Date: Mon, 6 May 2024 15:21:35 +0200 Subject: [PATCH] Make LinearAlgebra.generic_trimatmul more specific (#2359) --- lib/cublas/linalg.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/cublas/linalg.jl b/lib/cublas/linalg.jl index 763b32aafa..a5a6cd25b5 100644 --- a/lib/cublas/linalg.jl +++ b/lib/cublas/linalg.jl @@ -221,10 +221,10 @@ end if VERSION >= v"1.10-" # multiplication -LinearAlgebra.generic_trimatmul!(c::StridedCuVector{T}, uploc, isunitc, tfun::Function, A::StridedCuMatrix{T}, b::AbstractVector{T}) where {T<:CublasFloat} = +LinearAlgebra.generic_trimatmul!(c::StridedCuVector{T}, uploc, isunitc, tfun::Function, A::StridedCuMatrix{T}, b::StridedCuVector{T}) where {T<:CublasFloat} = trmv!(uploc, tfun === identity ? 'N' : tfun === transpose ? 'T' : 'C', isunitc, A, c === b ? c : copyto!(c, b)) # division -LinearAlgebra.generic_trimatdiv!(C::StridedCuVector{T}, uploc, isunitc, tfun::Function, A::StridedCuMatrix{T}, B::AbstractVector{T}) where {T<:CublasFloat} = +LinearAlgebra.generic_trimatdiv!(C::StridedCuVector{T}, uploc, isunitc, tfun::Function, A::StridedCuMatrix{T}, B::StridedCuVector{T}) where {T<:CublasFloat} = trsv!(uploc, tfun === identity ? 'N' : tfun === transpose ? 'T' : 'C', isunitc, A, C === B ? C : copyto!(C, B)) else ## direct multiplication/division