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

Update all uses of @tsl//third_party to @xla//third_party

Browse Source 
PiperOrigin-RevId: 733495240
This commit is contained in:
David Dunleavy 2025-03-04 15:54:40 -08:00 committed by jax authors
parent c145102ef4
commit 1a19d5594a
3 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
BUILD.bazel
View File

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
load("@tsl//third_party/py:python_wheel.bzl", "collect_data_files", "transitive_py_deps") load("@xla//third_party/py:python_wheel.bzl", "collect_data_files", "transitive_py_deps")
load( load(
"//jaxlib:jax.bzl", "//jaxlib:jax.bzl",
"jax_wheel", "jax_wheel",

12
WORKSPACE
View File

@ -70,7 +70,7 @@ jax_python_wheel_repository(
) )
load( load(
"@tsl//third_party/py:python_wheel.bzl", "@xla//third_party/py:python_wheel.bzl",
"python_wheel_version_suffix_repository", "python_wheel_version_suffix_repository",
) )
python_wheel_version_suffix_repository( python_wheel_version_suffix_repository(
@ -78,7 +78,7 @@ python_wheel_version_suffix_repository(
) )
load( load(
"@tsl//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl", "@xla//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl",
"cuda_json_init_repository", "cuda_json_init_repository",
) )
@ -90,7 +90,7 @@ load(
"CUDNN_REDISTRIBUTIONS", "CUDNN_REDISTRIBUTIONS",
) )
load( load(
"@tsl//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl", "@xla//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl",
"cuda_redist_init_repositories", "cuda_redist_init_repositories",
"cudnn_redist_init_repository", "cudnn_redist_init_repository",
) )
@ -104,21 +104,21 @@ cudnn_redist_init_repository(
) )
load( load(
"@tsl//third_party/gpus/cuda/hermetic:cuda_configure.bzl", "@xla//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
"cuda_configure", "cuda_configure",
) )
cuda_configure(name = "local_config_cuda") cuda_configure(name = "local_config_cuda")
load( load(
"@tsl//third_party/nccl/hermetic:nccl_redist_init_repository.bzl", "@xla//third_party/nccl/hermetic:nccl_redist_init_repository.bzl",
"nccl_redist_init_repository", "nccl_redist_init_repository",
) )
nccl_redist_init_repository() nccl_redist_init_repository()
load( load(
"@tsl//third_party/nccl/hermetic:nccl_configure.bzl", "@xla//third_party/nccl/hermetic:nccl_configure.bzl",
"nccl_configure", "nccl_configure",
) )

2
jaxlib/tools/BUILD.bazel
View File

@ -19,7 +19,7 @@ load("@bazel_skylib//rules:common_settings.bzl", "string_flag")
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda") load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm") load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm")
load( load(
"@tsl//third_party/py:py_manylinux_compliance_test.bzl", "@xla//third_party/py:py_manylinux_compliance_test.bzl",
"verify_manylinux_compliance_test", "verify_manylinux_compliance_test",
) )
load( load(