Skip to content

Commit

Permalink
Deprecate using build_cuda_plugin_from_source flag and rely on jaxlib…
Browse files Browse the repository at this point in the history
…_build config.

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

PiperOrigin-RevId: 660926791
  • Loading branch information
Jieying Luo authored and jax authors committed Aug 8, 2024
1 parent e630324 commit 751b574
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions jaxlib/jax.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 751b574

Please sign in to comment.