@@ -144,14 +144,7 @@ load("@local_config_android//:android.bzl", "android_workspace")
144144android_workspace ()
145145
146146load (
147- "@local_xla//third_party/py:python_wheel.bzl" ,
148- "python_wheel_version_suffix_repository" ,
149- )
150-
151- python_wheel_version_suffix_repository (name = "tf_wheel_version_suffix" )
152-
153- load (
154- "@local_xla//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl" ,
147+ "@local_tsl//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl" ,
155148 "cuda_json_init_repository" ,
156149)
157150
@@ -163,7 +156,7 @@ load(
163156 "CUDNN_REDISTRIBUTIONS" ,
164157)
165158load (
166- "@local_xla //third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl" ,
159+ "@local_tsl //third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl" ,
167160 "cuda_redist_init_repositories" ,
168161 "cudnn_redist_init_repository" ,
169162)
@@ -177,21 +170,21 @@ cudnn_redist_init_repository(
177170)
178171
179172load (
180- "@local_xla //third_party/gpus/cuda/hermetic:cuda_configure.bzl" ,
173+ "@local_tsl //third_party/gpus/cuda/hermetic:cuda_configure.bzl" ,
181174 "cuda_configure" ,
182175)
183176
184177cuda_configure (name = "local_config_cuda" )
185178
186179load (
187- "@local_xla //third_party/nccl/hermetic:nccl_redist_init_repository.bzl" ,
180+ "@local_tsl //third_party/nccl/hermetic:nccl_redist_init_repository.bzl" ,
188181 "nccl_redist_init_repository" ,
189182)
190183
191184nccl_redist_init_repository ()
192185
193186load (
194- "@local_xla //third_party/nccl/hermetic:nccl_configure.bzl" ,
187+ "@local_tsl //third_party/nccl/hermetic:nccl_configure.bzl" ,
195188 "nccl_configure" ,
196189)
197190
0 commit comments