Internal CI configuration to run tests on H100

Do some required BUILD file adjustments PiperOrigin-RevId: 621777011
openxla · Apr 4, 2024 · 31da0a2 · 31da0a2
1 parent ff468cb
commit 31da0a2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD
@@ -654,7 +654,7 @@ xla_test(
         "requires-gpu-sm80",
     ]},
     backends = [
-        "gpu",
+        "gpu_a100",
         "gpu_h100",
     ],
     tags = [

diff --git a/xla/service/gpu/kernels/BUILD b/xla/service/gpu/kernels/BUILD
@@ -105,6 +105,8 @@ xla_test(
     name = "cutlass_gemm_fusion_test",
     srcs = ["cutlass_gemm_fusion_test.cc"],
     backends = ["gpu"],
+    # TODO(b/332820384): Enable when it passes on H100.
+    disabled_backends = ["gpu_h100"],
     tags = ["no_rocm"],
     deps = [
         ":custom_kernel_fusion_pattern",