fix in-place sparse matrix code

JuliaSymbolics · Mar 13, 2021 · 5a42da6 · 5a42da6
1 parent 9cac075
commit 5a42da6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/build_function.jl b/src/build_function.jl
@@ -298,6 +298,10 @@ function set_array(s::MultithreadedForm, closed_args, out, outputidxs, rhss, che
     SpawnFetch{MultithreadedForm}(first.(arrays), last.(arrays), @inline noop(args...) = nothing)
 end
 
+function _set_array(out, outputidxs, rhss::AbstractSparseArray, checkbounds, skipzeros)
+    _set_array(LiteralExpr(:($out.nzval)), nothing, rhss.nzval, checkbounds, skipzeros)
+end
+
 function _set_array(out, outputidxs, rhss::AbstractArray, checkbounds, skipzeros)
     if outputidxs === nothing
         outputidxs = collect(eachindex(rhss))