[Codegen] Add RematerializeParallelOps pass back to codegen pipeline (#14744)
Removing this pass breaks codegen (CPU/CUDA/SPIRV) for dequantization by
adding a big alloc.
Fixes #14713
Fixes #14741
Fixes #14740
diff --git a/compiler/src/iree/compiler/Codegen/Common/Passes.cpp b/compiler/src/iree/compiler/Codegen/Common/Passes.cpp
index 6254af0..77a463d 100644
--- a/compiler/src/iree/compiler/Codegen/Common/Passes.cpp
+++ b/compiler/src/iree/compiler/Codegen/Common/Passes.cpp
@@ -17,6 +17,7 @@
passManager.addPass(createBufferizeCopyOnlyDispatchesPass());
passManager.addNestedPass<func::FuncOp>(
IREE::LinalgExt::createDecomposeSoftmaxPass());
+ passManager.addNestedPass<func::FuncOp>(createRematerializeParallelOpsPass());
}
//===---------------------------------------------------------------------===//