[benchmarks] Disallow XRT for XLA:CUDA. (#8006)

pytorch · Sep 13, 2024 · afbd7bd · afbd7bd
1 parent abb07e4
commit afbd7bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/benchmarks/benchmark_experiment.py b/benchmarks/benchmark_experiment.py
@@ -114,7 +114,8 @@ def _is_available(self,
       if cfg_xla is None:
         return False
     elif cfg_accelerator in ("cpu", "cuda"):
-      pass
+      if cfg_xla == "XRT":
+        return False
     else:
       raise NotImplementedError