jungpark-mlir wrote: Also thought about further optimizations but it could be better to be a separate pass. We can add an option to choose the order but this simply looks more sensible to me now, feel free to discuss. https://github.com/llvm/llvm-project/pull/79592