Minor refactor

JuliaDiff · Jun 21, 2024 · c7742bf · c7742bf
1 parent 00cc8c3
commit c7742bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/ChainRulesKernelAbstractionsExt/ChainRulesKernelAbstractionsExt.jl b/ext/ChainRulesKernelAbstractionsExt/ChainRulesKernelAbstractionsExt.jl
@@ -32,8 +32,8 @@ end
 @generated function _scatter!(i, op, dest, src, idims, Is::Vararg{Any, N}) where N
     quote
         is = @inbounds CartesianIndices(idims)[i]
-        Base.Cartesian.@nexprs $N j -> I_j = @inbounds((Is[j])[is[j]])
         dv = src[i]
+        Base.Cartesian.@nexprs $N j -> I_j = @inbounds((Is[j])[is[j]])
         Base.Cartesian.@ncall $N _accum! op dest dv j -> I_j
     end
 end