From 751b5742fdb5ff3e36212e8e64ef668de41363ff Mon Sep 17 00:00:00 2001
From: Jieying Luo <jieying@google.com>
Date: Thu, 8 Aug 2024 11:57:18 -0700
Subject: [PATCH] Deprecate using build_cuda_plugin_from_source flag and rely
 on jaxlib_build config.

If jaxlib needs to be built from source, cuda plugin will be built from source as well.

PiperOrigin-RevId: 660926791
---
 jaxlib/jax.bzl | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/jaxlib/jax.bzl b/jaxlib/jax.bzl
index d1decfd3a..7df848aad 100644
--- a/jaxlib/jax.bzl
+++ b/jaxlib/jax.bzl
@@ -267,10 +267,10 @@ def jax_test(
             deps = [
                 "//jax",
                 "//jax:test_util",
-            ] + deps + if_building_jaxlib(["//jaxlib/cuda:gpu_only_test_deps"]) + select({
-                "//jax:enable_build_cuda_plugin_from_source": ["//jax_plugins:gpu_plugin_only_test_deps"],
-                "//conditions:default": [],
-            }),
+            ] + deps + if_building_jaxlib([
+                "//jaxlib/cuda:gpu_only_test_deps",
+                "//jax_plugins:gpu_plugin_only_test_deps",
+            ]),
             data = data,
             shard_count = test_shards,
             tags = test_tags,