Skip to content

Commit 99babff

Browse files
Address reviwer comments
Signed-off-by: Nirvedh <nirvedh@gmail.com>
1 parent 8dc6169 commit 99babff

File tree

1 file changed

+2
-3
lines changed
  • compiler/src/iree/compiler/Codegen/LLVMGPU

1 file changed

+2
-3
lines changed

compiler/src/iree/compiler/Codegen/LLVMGPU/Passes.cpp

+2-3
Original file line numberDiff line numberDiff line change
@@ -484,9 +484,6 @@ void addGPUTileAndFusePassPipeline(OpPassManager &funcPassManager,
484484
options.redundantHoisting = false;
485485
funcPassManager.addPass(createOptimizeVectorTransferPass());
486486
}
487-
// Step 10. Hoist statically bounded allocations. Need to pad dynamic
488-
// allocations at this stage to make sure they can be hoisted.
489-
funcPassManager.addPass(createPadDynamicAllocPass());
490487
funcPassManager.addPass(createHoistStaticallyBoundAllocationsPass());
491488
funcPassManager.addPass(createCanonicalizerPass());
492489
funcPassManager.addPass(createCSEPass());
@@ -1095,6 +1092,8 @@ static void addLowerToLLVMGPUPasses(OpPassManager &modulePassManager,
10951092
// Pad allocations with dynamic dimension after linalg lowering but before
10961093
// lowering SCF and affine ops.
10971094
.addPass(createPadDynamicAllocPass)
1095+
// Hoist any newly static allocations from PadDynamicAlloc.
1096+
.addPass(createHoistStaticallyBoundAllocationsPass)
10981097

10991098
.addPass(createLowerAffinePass)
11001099
.addPass(createCanonicalizerPass)

0 commit comments

Comments
 (0)