fixup 3-arg mul!

JuliaSparse · Aug 23, 2024 · b36cdb4 · b36cdb4
1 parent 6cd2f43
commit b36cdb4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/interface.jl b/src/interface.jl
@@ -32,8 +32,11 @@ function mul!(C::StridedMatrix{T}, A::SimpleOrSpecialOrAdjMat{T, S}, B::StridedM
 end
 
 # 3-arg mul!() calls 5-arg mul!()
-mul!(C::StridedArray{T, N}, A::SimpleOrSpecialOrAdjMat{T, S}, B::StridedArray{T, N}) where {T <: BlasFloat, S <: MKLSparseMat{T}, N} =
+mul!(C::StridedMatrix{T}, A::SimpleOrSpecialOrAdjMat{T, S}, B::StridedMatrix{T}) where {T <: BlasFloat, S <: MKLSparseMat{T}} =
     mul!(C, A, B, one(T), zero(T))
+mul!(y::StridedVector{T}, A::SimpleOrSpecialOrAdjMat{T, S}, x::StridedVector{T}) where {T <: BlasFloat, S <: MKLSparseMat{T}} =
+    mul!(y, A, x, one(T), zero(T))
+
 
 # define 4-arg ldiv!(C, A, B, a) (C := alpha*inv(A)*B) that is not present in standard LinearAlgrebra
 # redefine 3-arg ldiv!(C, A, B) using 4-arg ldiv!(C, A, B, 1)