Skip to content

Commit 11ae318

Browse files
committed
overload mul!(y::DualMtx, m::FloatMtx, x::DualMtx)
1 parent add12d3 commit 11ae318

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/dual.jl

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -832,9 +832,10 @@ for MT in (StridedMatrix{<:LinearAlgebra.BlasFloat},
832832

833833
@eval Base.:*(m::$MT, x::StridedVector{<:Dual}) = mul!(similar(x, (size(m, 1),)), m, x)
834834

835-
@eval Base.:*(m::$MT, x::StridedMatrix{<:Dual}) =
836-
_map_dual_components!((y, x) -> mul!(y, m, x), (y, x, _) -> mul!(y, m, x),
837-
similar(x, (size(m, 1), size(x, 2))), x)
835+
@eval LinearAlgebra.mul!(y::StridedMatrix{T}, m::$MT, x::StridedMatrix{T}) where T <: Dual =
836+
_map_dual_components!((y, x) -> mul!(y, m, x), (y, x, _) -> mul!(y, m, x), y, x)
837+
838+
@eval Base.:*(m::$MT, x::StridedMatrix{<:Dual}) = mul!(similar(x, (size(m, 1), size(x, 2))), m, x)
838839
end
839840

840841
###################

0 commit comments

Comments
 (0)