diff --git a/compiler/src/iree/compiler/GlobalOptimization/Passes.cpp b/compiler/src/iree/compiler/GlobalOptimization/Passes.cpp index 6897a981b876..34b632e36a3f 100644 --- a/compiler/src/iree/compiler/GlobalOptimization/Passes.cpp +++ b/compiler/src/iree/compiler/GlobalOptimization/Passes.cpp @@ -36,6 +36,8 @@ void buildGlobalOptimizationPassPipeline( .addPass(IREE::Flow::createTopLevelSCFToCFGPass); mainPassManager.addPass(IREE::Flow::createExpandTensorShapesPass()); + mainPassManager.addPass(IREE::Flow::createTensorPadToTensorInsertSlicePass( + /*skipSingleLinalgOpUses=*/false)); FunctionLikeNest(mainPassManager) // Preprocess the input to a form more amenable for fusion // - Convert all elementwise ops to Linalg