Merge pull request #63 from ROCm/ci_clang_31_2

Merge fixes to 31 QA
ROCm · Nov 19, 2024 · 430d8c3 · 430d8c3
1 parent 08d8691
commit 430d8c3
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/xla/service/gpu/ir_emitter_triton_parametrized_test.cc b/xla/service/gpu/ir_emitter_triton_parametrized_test.cc
@@ -59,6 +59,19 @@ class MixedTypeTest : public GpuCodegenTest,
         .cuda_compute_capability();
   }
 
+  se::GpuComputeCapability GetGpuComputeCapability() {
+    return backend()
+        .default_stream_executor()
+        ->GetDeviceDescription()
+        .gpu_compute_capability();
+  }
+
+  void SetUp() override {
+    if (std::holds_alternative<se::RocmComputeCapability>(GetGpuComputeCapability())) {
+      GTEST_SKIP() << "Related fusions are not performed on ROCm without Triton.";
+    }
+  }
+
   DebugOptions GetDebugOptionsForTest() override {
     DebugOptions debug_options = GpuCodegenTest::GetDebugOptionsForTest();
     // We are testing Triton, remove cuBLAS fallback for these tests.