@@ -263,16 +263,16 @@ function _build_function(target::JuliaTarget, rhss::AbstractArray, args...;
263
263
264
264
if parallel isa DistributedForm
265
265
numworks = Distributed. nworkers ()
266
- reducevars = [Variable ( gensym (:MTReduceVar ))( ) for i in 1 : numworks]
266
+ reducevars = [gensym (:MTReduceVar ) for i in 1 : numworks]
267
267
lens = Int (ceil (rhs_length/ numworks))
268
268
finalsize = rhs_length - (numworks- 1 )* lens
269
- _rhss = vcat (reduce (vcat,[[getindex (reducevars[i],j) for j in 1 : lens] for i in 1 : numworks- 1 ],init= Expr[]),
270
- [getindex (reducevars[end ],j) for j in 1 : finalsize])
269
+ _rhss = vcat (reduce (vcat,[[Variable (reducevars[i],j) for j in 1 : lens] for i in 1 : numworks- 1 ],init= Expr[]),
270
+ [Variable (reducevars[end ],j) for j in 1 : finalsize])
271
271
272
272
elseif parallel isa DaggerForm
273
- computevars = [Variable ( gensym (:MTComputeVar ))( ) for i in axes (rhss,1 )]
274
- reducevar = Variable (gensym (:MTReduceVar ))()
275
- _rhss = [getindex (reducevar,i) for i in axes (rhss,1 )]
273
+ computevars = [gensym (:MTComputeVar ) for i in axes (rhss,1 )]
274
+ reducevar = Variable (gensym (:MTReduceVar ))
275
+ _rhss = [Variable (reducevar,i) for i in axes (rhss,1 )]
276
276
elseif rhss isa SparseMatrixCSC
277
277
_rhss = rhss. nzval
278
278
else
0 commit comments