Implement type piracy.

Author: RomeoV

src/ThreadedDenseSparseMul.jl

@@ -9,10 +9,12 @@ export fastdensesparsemul_outer!, fastdensesparsemul_outer_threaded!
 
 # Adapted from https://github.com/BacAmorim/ThreadedSparseCSR.jl/tree/main
 include("set_num_threads.jl")
+include("override.jl")
 function __init__()
     set_num_threads(Threads.nthreads())
 end
 
+
 const VecOrView{T} = Union{Vector{T}, SubArray{T, 1, Matrix{T}}}
 const MatOrView{T} = Union{Matrix{T}, SubArray{T, 2, Matrix{T}}}
 

src/override.jl

@@ -0,0 +1,19 @@
+import LinearAlgebra: mul!, Adjoint
+
+function override_mul!(threaded = true)
+    if threaded
+        @eval function  mul!(C::MatOrView{T}, A::MatOrView{T}, B::SparseMatrixCSC{T}, α::Number, β::Number) where T
+            fastdensesparsemul_threaded!(C, A, B, α, β)
+        end
+        @eval function mul!(C::MatOrView{T}, a::VecOrView{T}, b::Adjoint{<:SparseVector{T}}, α::Number, β::Number) where T
+            fastdensesparsemul_outer_threaded!(C, a, b', α, β)
+        end
+    else
+        @eval function  mul!(C::MatOrView{T}, A::MatOrView{T}, B::SparseMatrixCSC{T}, α::Number, β::Number) where T
+            fastdensesparsemul!(C, A, B, α, β)
+        end
+        @eval function mul!(C::MatOrView{T}, a::VecOrView{T}, b::Adjoint{<:SparseVector{T}}, α::Number, β::Number) where T
+            fastdensesparsemul_outer!(C, a, b', α, β)
+        end
+    end
+end