[XLA] Bump Python version to 3.10 in JAX:GPU presubmit.

JAX has dropped Python 3.9 support. PiperOrigin-RevId: 647034652
tensorflow · Jun 26, 2024 · 2d8b350 · 2d8b350
1 parent b5e4303
commit 2d8b350
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/third_party/xla/build_tools/build.py b/third_party/xla/build_tools/build.py
@@ -286,7 +286,7 @@ def nvidia_gpu_build_with_compute_capability(
     configs=(
         "avx_posix",
         "mkl_open_source_only",
-        "rbe_linux_cuda12.3_nvcc_py3.9",
+        "rbe_linux_cuda12.3_nvcc_py3.10",
         "tensorflow_testing_rbe_linux",
     ),
     target_patterns=("//tests:gpu_tests", "//tests:backend_independent_tests"),