Enable gb200 backend for xla_test

PiperOrigin-RevId: 874840441

Author: Google-ML-Automation

build_tools/lint/tags.py

@@ -93,6 +93,7 @@
     "xla_a100": "Runs on an a100.",
     "xla_h100": "Runs on an h100.",
     "xla_b200": "Runs on a b200.",
+    "xla_gb200": "Runs on a gb200.",
     # Below tags are consumed by `xla_test`.
     "test_migrated_to_hlo_runner_pjrt": (
         "Adds the appropriate `xla/tests:pjrt_$BACKEND_client_registry` to the"

xla/tests/build_defs.bzl

@@ -27,6 +27,7 @@ NVIDIA_GPU_BACKENDS = [
     "a100",
     "h100",
     "b200",
+    "gb200",
 ]
 
 # The generic "gpu" backend includes the actual backends in this list.
@@ -35,6 +36,7 @@ NVIDIA_GPU_DEFAULT_BACKENDS = [
     "a100",
     "h100",
     "b200",
+    "gb200",
 ]
 
 AMD_GPU_DEFAULT_BACKENDS = ["amdgpu_any"]
@@ -80,6 +82,7 @@ def prepare_nvidia_gpu_backend_data(backends, disabled_backends, backend_tags, b
         "a100": (8, 0),
         "h100": (9, 0),
         "b200": (10, 0),
+        "gb200": (10, 0),
     }
     for gpu_backend in NVIDIA_GPU_BACKENDS:
         all_tags = new_backend_tags[gpu_backend]