mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Deprecate using build_cuda_plugin_from_source flag and rely on jaxlib_build config.

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

PiperOrigin-RevId: 660926791
This commit is contained in:
Jieying Luo 2024-08-08 11:57:18 -07:00 committed by jax authors
parent e6303244bf
commit 751b5742fd
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
jaxlib/jax.bzl
View File

@ -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,