fix oop jac for sparsemat

Author: Langwen Huang

src/differentiation/compute_jacobian_ad.jl

@@ -111,8 +111,12 @@ function forwarddiff_color_jacobian(f,x::AbstractArray{<:Number},jac_cache::Forw
                 len_rows = length(pick_inds)
                 unused_rows = setdiff(1:nrows,rows_index_c)
                 perm_rows = sortperm(vcat(rows_index_c,unused_rows))
-                cols_index_c = vcat(cols_index_c,zeros(Int,nrows-len_rows))[perm_rows]
-                Ji = [j==cols_index_c[i] ? dx[i] : false for i in 1:nrows, j in 1:ncols]
+                if J isa SparseMatrixCSC
+                    Ji = sparse(rows_index_c, cols_index_c, dx[rows_index_c],nrows,ncols)
+                else
+                    cols_index_c = vcat(cols_index_c,zeros(Int,nrows-len_rows))[perm_rows]
+                    Ji = [j==cols_index_c[i] ? dx[i] : false for i in 1:nrows, j in 1:ncols]
+                end
                 J = J + Ji
                 color_i += 1
                 (color_i > maxcolor) && return J