diff --git a/src/Compiler.jl b/src/Compiler.jl index 96f348125c..e3f80556f8 100644 --- a/src/Compiler.jl +++ b/src/Compiler.jl @@ -1071,6 +1071,7 @@ function optimization_passes(; "transpose_batch_norm_inference", "transpose_batch_norm_grad", "transpose_if", + "transpose_scatter", ], ) if AGGRESSIVE_PROPAGATION[]