From 3ad60e4c906367cee99740c844a9cf3c9c28b283 Mon Sep 17 00:00:00 2001
From: Greg Olechwierowicz <olechwierowicz@google.com>
Date: Mon, 16 Sep 2024 03:11:22 -0700
Subject: [PATCH] [XLA:GPU] Run post-scheduling verification under the
 `xla_gpu_enable_pgle_accuracy_checker` flag.

PiperOrigin-RevId: 675075419
---
 xla/service/gpu/gpu_compiler.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/xla/service/gpu/gpu_compiler.cc b/xla/service/gpu/gpu_compiler.cc
index 6cb0d4d37ecd1..725556404b47b 100644
--- 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();
 }