Skip to content

Commit

Permalink
test for CUDA
Browse files Browse the repository at this point in the history
  • Loading branch information
hanhanW committed Aug 25, 2023
1 parent 58113c7 commit 39a67e2
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions compiler/src/iree/compiler/GlobalOptimization/Passes.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 39a67e2

Please sign in to comment.