diff --git a/crates/burn-jit/src/kernel/index/scatter.rs b/crates/burn-jit/src/kernel/index/scatter.rs index a17ea3836..1aaaf5e7c 100644 --- a/crates/burn-jit/src/kernel/index/scatter.rs +++ b/crates/burn-jit/src/kernel/index/scatter.rs @@ -77,7 +77,7 @@ pub(crate) fn scatter tensor, + true => tensor.copy(), false => tensor.copy(), };