[XLA:GPU] Add while-loop-simplifier before while loop double buffering.

It's not needed for anything in particular, but it's a nice property to work on the simplified IR in subsequent passes. PiperOrigin-RevId: 678613784
openxla · Sep 25, 2024 · ccdb368 · ccdb368
1 parent 41c58a5
commit ccdb368
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xla/service/gpu/gpu_compiler.cc b/xla/service/gpu/gpu_compiler.cc
@@ -1102,6 +1102,7 @@ absl::Status RunPostFusionPasses(
     unroll_strategy = DoubleBufferLoopUnrolling::UnrollStrategy::kAuto;
   }
   if (unroll_strategy != std::nullopt) {
+    pipeline.AddPass<WhileLoopSimplifier>();
     pipeline.AddPass<DoubleBufferLoopUnrolling>(*unroll_strategy);
     pipeline.AddPass<TupleSimplifier>();
     pipeline.AddPass<HloDCE>();