From 39a67e247183d6562a933fc74ceff668d9984184 Mon Sep 17 00:00:00 2001 From: Hanhan Wang Date: Thu, 24 Aug 2023 16:57:12 -0700 Subject: [PATCH] test for CUDA --- compiler/src/iree/compiler/GlobalOptimization/Passes.cpp | 2 ++ 1 file changed, 2 insertions(+) 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