Skip to content

Commit 6f7d99a

Browse files
committed
Fix ROCBLASFloat
1 parent 062d0a4 commit 6f7d99a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/blas/highlevel.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ function LinearAlgebra.rmul!(A::ROCMatrix{T}, B::Diagonal{T,<:ROCVector{T}}) whe
340340
return dgmm!('R', A, B.diag, A)
341341
end
342342

343-
function LinearAlgebra.mul!(C::Diagonal{T, <:ROCVector}, A::Union{<:ROCMatrix{T}, Adjoint{T, <:ROCMatrix}, Transpose{T, <:ROCMatrix}}, B::Union{<:ROCMatrix{T}, Adjoint{T, <:ROCMatrix}, Transpose{T, <:ROCMatrix}}) where {T<:ROCBLASloat}
343+
function LinearAlgebra.mul!(C::Diagonal{T, <:ROCVector}, A::Union{<:ROCMatrix{T}, Adjoint{T, <:ROCMatrix}, Transpose{T, <:ROCMatrix}}, B::Union{<:ROCMatrix{T}, Adjoint{T, <:ROCMatrix}, Transpose{T, <:ROCMatrix}}) where {T<:ROCBLASFloat}
344344
Cfull = A*B
345345
C.diag .= diag(Cfull)
346346
return C

0 commit comments

Comments
 (0)