mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-13 02:16: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
# 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(
"//jaxlib:jax.bzl",
"jax_wheel",

12
WORKSPACE
View File

@ -70,7 +70,7 @@ jax_python_wheel_repository(
)
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(
@ -78,7 +78,7 @@ python_wheel_version_suffix_repository(
)
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",
)
@ -90,7 +90,7 @@ load(
"CUDNN_REDISTRIBUTIONS",
)
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",
"cudnn_redist_init_repository",
)
@ -104,21 +104,21 @@ cudnn_redist_init_repository(
)
load(
"@tsl//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
"@xla//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
"cuda_configure",
)
cuda_configure(name = "local_config_cuda")
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()
load(
"@tsl//third_party/nccl/hermetic:nccl_configure.bzl",
"@xla//third_party/nccl/hermetic:nccl_configure.bzl",
"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_rocm//rocm:build_defs.bzl", "if_rocm")
load(
"@tsl//third_party/py:py_manylinux_compliance_test.bzl",
"@xla//third_party/py:py_manylinux_compliance_test.bzl",
"verify_manylinux_compliance_test",
)
load(