From ced268eba763a20e57557fb8d138272a4d7873a5 Mon Sep 17 00:00:00 2001 From: Anton Smirnov Date: Fri, 21 Jun 2024 14:27:00 +0300 Subject: [PATCH] Minor refactor --- .../ChainRulesKernelAbstractionsExt.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ext/ChainRulesKernelAbstractionsExt/ChainRulesKernelAbstractionsExt.jl b/ext/ChainRulesKernelAbstractionsExt/ChainRulesKernelAbstractionsExt.jl index a55837dba..c171d049f 100644 --- 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