[XLA:GPU] Run post-scheduling verification under the `xla_gpu_enable_…

…pgle_accuracy_checker` flag. PiperOrigin-RevId: 675075419
openxla · Sep 16, 2024 · 3ad60e4 · 3ad60e4
1 parent 1765b13
commit 3ad60e4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/xla/service/gpu/gpu_compiler.cc b/xla/service/gpu/gpu_compiler.cc
@@ -2556,6 +2556,10 @@ absl::Status GpuCompiler::RunPostSchedulingPipelines(
     pipeline.AddPass<SanitizeConstantNames>();
   }
 
+  if (module->config().debug_options().xla_gpu_enable_pgle_accuracy_checker()) {
+    AddHloVerifier(&main_pipeline,
+                   HloVerifierOpts{}.VerifyInstructionNameUnchanged());
+  }
   return main_pipeline.Run(module).status();
 }