[ROCm] Skip GemmFusionIsNoOpWhenGemmFusionAutotunerFallsBackToCublas …

…test in gpu_compiler_test because not using autotuner on ROCM yet.
ROCm · Jun 12, 2024 · 569bd86 · 569bd86
1 parent 9bdf8cb
commit 569bd86
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/xla/service/gpu/gpu_compiler_test.cc b/xla/service/gpu/gpu_compiler_test.cc
@@ -64,6 +64,10 @@ using ::testing::Not;
 using ::testing::TempDir;
 
 class GpuCompilerTest : public HloTestBase {
+  const auto& device_desc() {
+    return backend().default_stream_executor()->GetDeviceDescription();
+  }
+
  public:
   absl::Status Schedule(HloModule* module) {
     auto compiler = backend().compiler();
@@ -73,6 +77,10 @@ class GpuCompilerTest : public HloTestBase {
     return tensorflow::down_cast<GpuCompiler*>(compiler)
         ->RunPostSchedulingPipelines(module, 4 * 1024 * 1024, gpu_device_info);
   }
+
+  const se::GpuComputeCapability& GpuComputeComp() {
+    return device_desc().gpu_compute_capability();
+  }
 };
 
 TEST_F(GpuCompilerTest, CompiledProgramsCount) {
@@ -335,6 +343,10 @@ ENTRY main {
 
 TEST_F(GpuCompilerTest,
        GemmFusionIsNoOpWhenGemmFusionAutotunerFallsBackToCublas) {
+  if (std::holds_alternative<se::RocmComputeCapability>(GpuComputeComp())) {
+    GTEST_SKIP() << "Not using autotuner on ROCM yet.";
+  }
+
   const absl::string_view hlo_string = R"(
 HloModule test