Skip to content

Commit ed4337e

Browse files
committed
overload mul!(y::DualMtx, m::FloatMtx, x::DualMtx)
1 parent f8a617d commit ed4337e

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
@@ -863,9 +863,10 @@ for MT in (StridedMatrix{<:LinearAlgebra.BlasFloat},
863863

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

866-
@eval Base.:*(m::$MT, x::StridedMatrix{<:Dual}) =
867-
_map_dual_components!((y, x) -> mul!(y, m, x), (y, x, _) -> mul!(y, m, x),
868-
similar(x, (size(m, 1), size(x, 2))), x)
866+
@eval LinearAlgebra.mul!(y::StridedMatrix{T}, m::$MT, x::StridedMatrix{T}) where T <: Dual =
867+
_map_dual_components!((y, x) -> mul!(y, m, x), (y, x, _) -> mul!(y, m, x), y, x)
868+
869+
@eval Base.:*(m::$MT, x::StridedMatrix{<:Dual}) = mul!(similar(x, (size(m, 1), size(x, 2))), m, x)
869870
end
870871

871872
###################

0 commit comments

Comments
 (0)