Skip to content

Commit 38ed23c

Browse files
ethanluoyctensorflower-gardener
ethanluoyc
authored and
tensorflower-gardener
committed
Use @xla instead of @local_xla in vendored copy of XLA
PiperOrigin-RevId: 859657038
1 parent 53773c2 commit 38ed23c

520 files changed

Lines changed: 6689 additions & 6689 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎.bazelrc‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@
7171

7272
# These are used to generate the ML wheel version string.
7373
# See the explanation in the file comment of
74-
# @local_xla//third_party/py/python_wheel.bzl.
74+
# @xla//third_party/py/python_wheel.bzl.
7575
# The generated version suffix is used in
7676
# third_party/tensorflow/core/public/release_version.h and
7777
# third_party/tensorflow/tools/pip_package/setup.oss.py.tpl
@@ -856,27 +856,27 @@ common:tf_public_macos_cache_push --config=tf_public_macos_cache --remote_upload
856856
test:linux_cpu_wheel_test_filters --test_tag_filters=-no_oss,-tf_tosa,-oss_excluded,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_oss_py310,-no_oss_py313
857857
test:linux_cpu_wheel_test_filters --build_tag_filters=-no_oss,-tf_tosa,-oss_excluded,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_oss_py310,-no_oss_py313
858858
test:linux_cpu_wheel_test_filters --test_lang_filters=py --test_size_filters=small,medium
859-
test:linux_cpu_wheel_test --@local_xla//third_party/py:wheel_dependency=true --config=linux_cpu_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_cpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/...
859+
test:linux_cpu_wheel_test --@xla//third_party/py:wheel_dependency=true --config=linux_cpu_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_cpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/...
860860
# CUDA WHEEL
861861
test:linux_cuda_wheel_test_filters --test_tag_filters=gpu,requires-gpu,-no_gpu,-no_oss,-tf_tosa,-oss_excluded,-oss_serial,-benchmark-test,-no_cuda11,-no_oss_py310,-no_oss_py313
862862
test:linux_cuda_wheel_test_filters --build_tag_filters=gpu,requires-gpu,-no_gpu,-no_oss,-tf_tosa,-oss_excluded,-oss_serial,-benchmark-test,-no_cuda11,-no_oss_py310,-no_oss_py313
863863
test:linux_cuda_wheel_test_filters --test_lang_filters=py --test_size_filters=small,medium
864-
test:linux_cuda_wheel_test --repo_env=HERMETIC_CUDA_UMD_VERSION=12.8.1 --@local_xla//third_party/py:wheel_dependency=true --config=linux_cuda_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_gpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/...
864+
test:linux_cuda_wheel_test --repo_env=HERMETIC_CUDA_UMD_VERSION=12.8.1 --@xla//third_party/py:wheel_dependency=true --config=linux_cuda_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_gpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/...
865865
# ARM64 WHEEL
866866
test:linux_arm64_wheel_test_filters --test_tag_filters=-no_oss,-tf_tosa,-no_aarch64,-oss_excluded,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_oss_py310,-no_oss_py313
867867
test:linux_arm64_wheel_test_filters --build_tag_filters=-no_oss,-tf_tosa,-no_aarch64,-oss_excluded,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_oss_py310,-no_oss_py313
868868
test:linux_arm64_wheel_test_filters --test_lang_filters=py --test_size_filters=small,medium
869-
test:linux_arm64_wheel_test --@local_xla//third_party/py:wheel_dependency=true --config=linux_arm64_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_cpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/... -//tensorflow/go/... -//tensorflow/java/... -//tensorflow/core/grappler/optimizers:auto_mixed_precision_test_cpu -//tensorflow/core/grappler/optimizers:remapper_test_cpu -//tensorflow/core/kernels/image:resize_bicubic_op_test
869+
test:linux_arm64_wheel_test --@xla//third_party/py:wheel_dependency=true --config=linux_arm64_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_cpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/... -//tensorflow/go/... -//tensorflow/java/... -//tensorflow/core/grappler/optimizers:auto_mixed_precision_test_cpu -//tensorflow/core/grappler/optimizers:remapper_test_cpu -//tensorflow/core/kernels/image:resize_bicubic_op_test
870870
# MACOS ARM64 WHEEL
871871
test:macos_arm64_wheel_test_filters --test_tag_filters=-no_oss,-tf_tosa,-oss_excluded,-oss_serial,-no_oss_py310,-no_oss_py313,-nomac,-no_mac,-mac_excluded,-v1only,-gpu,-tpu,-benchmark-test,-no_mac_arm64,-no_aarch64
872872
test:macos_arm64_wheel_test_filters --build_tag_filters=-no_oss,-tf_tosa,-oss_excluded,-oss_serial,-no_oss_py310,-no_oss_py313,-nomac,-no_mac,-mac_excluded,-v1only,-gpu,-tpu,-benchmark-test,-no_mac_arm64,-no_aarch64
873873
test:macos_arm64_wheel_test_filters --test_lang_filters=py --test_size_filters=small,medium
874-
test:macos_arm64_wheel_test --@local_xla//third_party/py:wheel_dependency=true --config=macos_arm64_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_cpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/... -//tensorflow/go/... -//tensorflow/java/... -//tensorflow/compiler/aot/...
874+
test:macos_arm64_wheel_test --@xla//third_party/py:wheel_dependency=true --config=macos_arm64_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_cpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/... -//tensorflow/go/... -//tensorflow/java/... -//tensorflow/compiler/aot/...
875875
# MACOS X86 WHEEL
876876
test:macos_x86_wheel_test_filters --test_tag_filters=-no_oss,-tf_tosa,-oss_excluded,-oss_serial,-no_oss_py310,-no_oss_py313,-nomac,-no_mac,-mac_excluded,-v1only,-gpu,-tpu,-benchmark-test
877877
test:macos_x86_wheel_test_filters --build_tag_filters=-no_oss,-tf_tosa,-oss_excluded,-oss_serial,-no_oss_py310,-no_oss_py313,-nomac,-no_mac,-mac_excluded,-v1only,-gpu,-tpu,-benchmark-test
878878
test:macos_x86_wheel_test_filters --test_lang_filters=py --test_size_filters=small,medium
879-
test:macos_x86_wheel_test --@local_xla//third_party/py:wheel_dependency=true --config=macos_x86_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_cpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/... -//tensorflow/go/... -//tensorflow/java/... -//tensorflow/compiler/aot/...
879+
test:macos_x86_wheel_test --@xla//third_party/py:wheel_dependency=true --config=macos_x86_wheel_test_filters -- //tensorflow/... //tensorflow/tools/pip_package:prebuilt_wheel_import_api_packages_test_cpu -//tensorflow/compiler/tf2tensorrt/... -//tensorflow/core/tpu/... -//tensorflow/lite/... -//tensorflow/tools/toolchains/... -//tensorflow/go/... -//tensorflow/java/... -//tensorflow/compiler/aot/...
880880
# WINDOWS X86 WHEEL
881881
test:windows_x86_cpu_2022_wheel_test_filters --test_tag_filters=-no_windows,-windows_excluded,-no_oss,-oss_excluded,-gpu,-tpu,-benchmark-test,-v1only
882882
test:windows_x86_cpu_2022_wheel_test_filters --build_tag_filters=-no_windows,-windows_excluded,-no_oss,-oss_excluded,-benchmark-test,-v1only

‎WORKSPACE‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -60,11 +60,11 @@ rules_shell_dependencies()
6060
rules_shell_toolchains()
6161

6262
# Initialize hermetic Python
63-
load("@local_xla//third_party/py:python_init_rules.bzl", "python_init_rules")
63+
load("@xla//third_party/py:python_init_rules.bzl", "python_init_rules")
6464

6565
python_init_rules()
6666

67-
load("@local_xla//third_party/py:python_init_repositories.bzl", "python_init_repositories")
67+
load("@xla//third_party/py:python_init_repositories.bzl", "python_init_repositories")
6868

6969
python_init_repositories(
7070
default_python_version = "system",
@@ -82,11 +82,11 @@ python_init_repositories(
8282
},
8383
)
8484

85-
load("@local_xla//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")
85+
load("@xla//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")
8686

8787
python_init_toolchains()
8888

89-
load("@local_xla//third_party/py:python_init_pip.bzl", "python_init_pip")
89+
load("@xla//third_party/py:python_init_pip.bzl", "python_init_pip")
9090

9191
python_init_pip()
9292

@@ -108,7 +108,7 @@ load("@//tensorflow:workspace0.bzl", "tf_workspace0")
108108
tf_workspace0()
109109

110110
load(
111-
"@local_xla//third_party/py:python_wheel.bzl",
111+
"@xla//third_party/py:python_wheel.bzl",
112112
"nvidia_wheel_versions_repository",
113113
"python_wheel_version_suffix_repository",
114114
)
@@ -141,7 +141,7 @@ load(
141141
"@rules_ml_toolchain//gpu/cuda:cuda_redist_versions.bzl",
142142
"REDIST_VERSIONS_TO_BUILD_TEMPLATES",
143143
)
144-
load("@local_xla//third_party/cccl:workspace.bzl", "CCCL_DIST_DICT", "CCCL_GITHUB_VERSIONS_TO_BUILD_TEMPLATES")
144+
load("@xla//third_party/cccl:workspace.bzl", "CCCL_DIST_DICT", "CCCL_GITHUB_VERSIONS_TO_BUILD_TEMPLATES")
145145

146146
cuda_redist_init_repositories(
147147
cuda_redistributions = CUDA_REDISTRIBUTIONS | CCCL_DIST_DICT,

‎ci/official/utilities/code_check_full.bats‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@ do_external_licenses_check(){
4646
@bazel_tools//third_party/
4747
@bazel_tools//tools
4848
@local
49+
@xla//xla
50+
@tsl//tsl
4951
@com_google_absl//absl
5052
@pybind11_abseil//pybind11_abseil
5153
@org_tensorflow//
@@ -64,8 +66,8 @@ EOF
6466

6567
# grep patterns for targets which are allowed to be extra licenses
6668
cat > $BATS_TEST_TMPDIR/allowed_to_be_extra <<EOF
67-
@local_xla//third_party/mkl
68-
@local_xla//third_party/mkl_dnn
69+
@xla//third_party/mkl
70+
@xla//third_party/mkl_dnn
6971
@absl_py//
7072
@bazel_tools//src
7173
@bazel_tools//platforms
@@ -77,6 +79,8 @@ EOF
7779
@pybind11_abseil//pybind11_abseil
7880
//external
7981
@local
82+
@xla//xla
83+
@tsl//tsl
8084
@com_github_googlecloudplatform_google_cloud_cpp//
8185
@embedded_jdk//
8286
^//$

‎tensorflow/BUILD‎

Lines changed: 21 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ load(
3333
"tf_cc_shared_library",
3434
)
3535
load(
36-
"@local_xla//xla/tsl/mkl:build_defs.bzl",
36+
"@xla//xla/tsl/mkl:build_defs.bzl",
3737
"if_mkl_ml",
3838
)
3939
load(
@@ -136,8 +136,8 @@ PACKAGE_STATIC_DEPS = [
136136
"@flatbuffers//:__subpackages__",
137137
"@nccl_archive//:__subpackages__",
138138
"@triton//:__subpackages__",
139-
"@local_tsl//:__subpackages__",
140-
"@local_xla//:__subpackages__",
139+
"@tsl//:__subpackages__",
140+
"@xla//:__subpackages__",
141141
] + tsl_async_value_deps()
142142

143143
package(
@@ -1071,7 +1071,7 @@ filegroup(
10711071
name = "intel_binary_blob",
10721072
data = if_mkl_ml(
10731073
[
1074-
"@local_xla//xla/tsl/mkl:intel_binary_blob",
1074+
"@xla//xla/tsl/mkl:intel_binary_blob",
10751075
],
10761076
),
10771077
)
@@ -1095,14 +1095,14 @@ bzl_library(
10951095
# copybara:uncomment "//devtools/build_cleaner/skylark:build_defs_lib",
10961096
"@rules_cc//cc/common",
10971097
"@rules_java//java:rules",
1098-
"@local_xla//xla/tsl:tsl_bzl",
1099-
"@local_xla//xla/tsl:tsl_default_bzl",
1100-
"@local_xla//xla/tsl/mkl:build_defs_bzl",
1098+
"@xla//xla/tsl:tsl_bzl",
1099+
"@xla//xla/tsl:tsl_default_bzl",
1100+
"@xla//xla/tsl/mkl:build_defs_bzl",
11011101
"//tensorflow/core/platform:build_config_root_bzl",
11021102
"//tensorflow/core/platform:rules_cc_bzl",
1103-
"@local_xla//third_party/py/rules_pywrap:pywrap_bzl",
1104-
"@local_xla//third_party/compute_library:build_defs_bzl",
1105-
"@local_xla//third_party/llvm_openmp:openmp_bzl",
1103+
"@xla//third_party/py/rules_pywrap:pywrap_bzl",
1104+
"@xla//third_party/compute_library:build_defs_bzl",
1105+
"@xla//third_party/llvm_openmp:openmp_bzl",
11061106
"@bazel_skylib//lib:new_sets",
11071107
"@bazel_skylib//rules:common_settings",
11081108
"@local_config_cuda//cuda:build_defs_bzl",
@@ -1201,9 +1201,9 @@ tf_cc_shared_library(
12011201
"//tensorflow/core/profiler:profiler_impl",
12021202
"//tensorflow/core/util:determinism", # Must be linked and exported to libtensorflow_framework.so.
12031203
"//tensorflow/lite/kernels/shim:tf_kernel_shim",
1204-
"@local_xla//xla/stream_executor:stream_executor_impl",
1205-
"@local_xla//xla/tsl/framework:bfc_allocator",
1206-
"@local_xla//xla/tsl/framework:metrics",
1204+
"@xla//xla/stream_executor:stream_executor_impl",
1205+
"@xla//xla/tsl/framework:bfc_allocator",
1206+
"@xla//xla/tsl/framework:metrics",
12071207
"//tensorflow/core/data:captured_function",
12081208
] + tf_additional_binary_deps(),
12091209
soversion = VERSION,
@@ -1302,8 +1302,8 @@ tf_cc_shared_library(
13021302
}),
13031303
exports_filter = [
13041304
"//:__subpackages__",
1305-
"@local_tsl//tsl:__subpackages__",
1306-
"@local_xla//xla:__subpackages__",
1305+
"@tsl//tsl:__subpackages__",
1306+
"@xla//xla:__subpackages__",
13071307
],
13081308
framework_so = [],
13091309
linkopts = select({
@@ -1366,7 +1366,7 @@ tf_cc_shared_library(
13661366
"//tensorflow/compiler/mlir/tensorflow:mlir_import_options",
13671367
"//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags",
13681368
"//tensorflow/compiler/mlir/tensorflow:translate_lib",
1369-
"@local_xla//xla/service:computation_placer",
1369+
"@xla//xla/service:computation_placer",
13701370
"//tensorflow/core",
13711371
"//tensorflow/core/common_runtime/eager:context",
13721372
"//tensorflow/core/common_runtime/eager:tensor_handle",
@@ -1475,11 +1475,11 @@ check_deps(
14751475
disallowed_deps = if_static(
14761476
[],
14771477
otherwise = [
1478-
"@local_xla//xla/stream_executor/cuda:all_runtime",
1479-
"@local_xla//xla/stream_executor/cuda:cuda_platform",
1480-
"@local_xla//xla/stream_executor/cuda:cudnn_plugin",
1481-
"@local_xla//xla/stream_executor/cuda:cufft_plugin",
1482-
"@local_xla//xla/stream_executor:cuda_platform",
1478+
"@xla//xla/stream_executor/cuda:all_runtime",
1479+
"@xla//xla/stream_executor/cuda:cuda_platform",
1480+
"@xla//xla/stream_executor/cuda:cudnn_plugin",
1481+
"@xla//xla/stream_executor/cuda:cufft_plugin",
1482+
"@xla//xla/stream_executor:cuda_platform",
14831483
],
14841484
),
14851485
deps = if_cuda([

‎tensorflow/c/BUILD‎

Lines changed: 27 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33

44
load("@bazel_skylib//lib:selects.bzl", "selects")
55
load("@local_config_tensorrt//:build_defs.bzl", "if_tensorrt")
6-
load("@local_xla//xla/tsl/platform:build_config_root.bzl", "if_pywrap")
7-
load("@local_xla//xla/tsl/platform:rules_cc.bzl", "cc_library")
6+
load("@xla//xla/tsl/platform:build_config_root.bzl", "if_pywrap")
7+
load("@xla//xla/tsl/platform:rules_cc.bzl", "cc_library")
88
load(
99
"//tensorflow:tensorflow.bzl",
1010
"check_deps",
@@ -69,7 +69,7 @@ filegroup(
6969
"tf_tensor_helper.h",
7070
"tf_tstring.h",
7171
"//tensorflow/core/platform:ctstring",
72-
"@local_xla//xla/tsl/c:headers",
72+
"@xla//xla/tsl/c:headers",
7373
] + if_tensorrt([
7474
"//tensorflow/compiler/tf2tensorrt:headers",
7575
]),
@@ -93,8 +93,8 @@ filegroup(
9393
) + [
9494
"//tensorflow/cc:srcs_no_runtime",
9595
"//tensorflow/core/distributed_runtime:server_lib.h",
96-
"@local_tsl//tsl/platform:ctstring",
97-
"@local_xla//xla/tsl/c:srcs",
96+
"@tsl//tsl/platform:ctstring",
97+
"@xla//xla/tsl/c:srcs",
9898
],
9999
visibility = ["//visibility:public"],
100100
)
@@ -112,7 +112,7 @@ cc_library(
112112
"tf_status_internal.h",
113113
"tf_tensor_helper.h",
114114
"tf_tensor_internal.h",
115-
"@local_xla//xla/tsl/c:tsl_status_internal_headers",
115+
"@xla//xla/tsl/c:tsl_status_internal_headers",
116116
],
117117
visibility = [
118118
"//tensorflow/core:__pkg__",
@@ -250,7 +250,7 @@ tf_cuda_library(
250250
"//tensorflow/core:protos_all_cc",
251251
"//tensorflow/core/platform:tstring",
252252
"@com_google_absl//absl/status",
253-
"@local_xla//xla/tsl/c:tsl_status",
253+
"@xla//xla/tsl/c:tsl_status",
254254
] + select({
255255
"//tensorflow:with_xla_support": [
256256
"//tensorflow/compiler/jit",
@@ -329,7 +329,7 @@ tf_cuda_library(
329329
hdrs = [
330330
"tf_status.h",
331331
"tf_status_internal.h",
332-
"@local_xla//xla/tsl/c:tsl_status_internal_headers",
332+
"@xla//xla/tsl/c:tsl_status_internal_headers",
333333
],
334334
visibility = [
335335
"//tensorflow/c:__subpackages__",
@@ -342,9 +342,9 @@ tf_cuda_library(
342342
],
343343
deps = [
344344
":c_api_macros_hdrs",
345-
"@local_xla//xla/tsl/c:tsl_status",
346-
"@local_xla//xla/tsl/c:tsl_status_internal",
347-
"@local_xla//xla/tsl/platform:status",
345+
"@xla//xla/tsl/c:tsl_status",
346+
"@xla//xla/tsl/c:tsl_status_internal",
347+
"@xla//xla/tsl/platform:status",
348348
] + select({
349349
"//tensorflow:android": [
350350
"//tensorflow/core:portable_tensorflow_lib_lite", # TODO(annarev): exclude runtime srcs
@@ -359,7 +359,7 @@ filegroup(
359359
name = "tf_status_internal_headers",
360360
srcs = [
361361
"tf_status_internal.h",
362-
"@local_xla//xla/tsl/c:tsl_status_internal_headers",
362+
"@xla//xla/tsl/c:tsl_status_internal_headers",
363363
],
364364
visibility = [
365365
"//tensorflow/python:__subpackages__",
@@ -399,7 +399,7 @@ cc_library(
399399
deps = [
400400
":c_api_macros_hdrs",
401401
":tf_status_internal",
402-
"@local_xla//xla/tsl/c:tsl_status",
402+
"@xla//xla/tsl/c:tsl_status",
403403
] + select({
404404
"//tensorflow:android": [
405405
"//tensorflow/core:portable_tensorflow_lib_lite", # TODO(annarev): exclude runtime srcs
@@ -416,7 +416,7 @@ cc_library(
416416
visibility = ["//visibility:public"],
417417
deps = [
418418
":c_api_macros_hdrs",
419-
"@local_xla//xla/tsl/c:tsl_status_headers",
419+
"@xla//xla/tsl/c:tsl_status_headers",
420420
],
421421
)
422422

@@ -437,7 +437,7 @@ cc_library(
437437
":tf_tensor_hdrs",
438438
"//tensorflow/core/platform:status",
439439
"//tensorflow/core/platform:tstring",
440-
"@local_xla//xla/tsl/c:tsl_status",
440+
"@xla//xla/tsl/c:tsl_status",
441441
],
442442
)
443443

@@ -647,7 +647,7 @@ tf_cuda_library(
647647
"//tensorflow/core/platform:blocking_counter",
648648
"@com_google_absl//absl/strings",
649649
"@com_google_absl//absl/synchronization",
650-
"@local_xla//xla/tsl/c:tsl_status_internal",
650+
"@xla//xla/tsl/c:tsl_status_internal",
651651
],
652652
alwayslink = 1,
653653
)
@@ -676,8 +676,8 @@ tf_cuda_library(
676676
visibility = ["//visibility:public"],
677677
deps = [
678678
":tf_status_headers",
679-
"@local_xla//xla/tsl/c:tsl_status_helper",
680-
"@local_xla//xla/tsl/platform:status",
679+
"@xla//xla/tsl/c:tsl_status_helper",
680+
"@xla//xla/tsl/platform:status",
681681
],
682682
)
683683

@@ -688,10 +688,10 @@ tf_cc_test(
688688
":tf_status",
689689
":tf_status_helper",
690690
"@com_google_googletest//:gtest_main",
691-
"@local_xla//xla/tsl/platform:errors",
692-
"@local_xla//xla/tsl/platform:status",
693-
"@local_xla//xla/tsl/platform:test",
694-
"@local_xla//xla/tsl/platform:test_main",
691+
"@xla//xla/tsl/platform:errors",
692+
"@xla//xla/tsl/platform:status",
693+
"@xla//xla/tsl/platform:test",
694+
"@xla//xla/tsl/platform:test_main",
695695
],
696696
)
697697

@@ -759,7 +759,7 @@ cc_library(
759759
":tf_tensor_hdrs",
760760
"//tensorflow/c/experimental/stream_executor:stream_executor_hdrs",
761761
"//tensorflow/core/common_runtime/next_pluggable_device/c:tf_rendezvous_c_api",
762-
"@local_tsl//tsl/platform",
762+
"@tsl//tsl/platform",
763763
],
764764
)
765765

@@ -787,7 +787,7 @@ tf_cuda_library(
787787
"@com_google_absl//absl/status",
788788
"@com_google_absl//absl/strings",
789789
"@com_google_absl//absl/synchronization",
790-
"@local_xla//xla/tsl/c:tsl_status_internal",
790+
"@xla//xla/tsl/c:tsl_status_internal",
791791
] + select({
792792
"//tensorflow:android": [
793793
"//tensorflow/c/experimental/stream_executor:stream_executor_hdrs",
@@ -800,9 +800,9 @@ tf_cuda_library(
800800
"//tensorflow/core:framework",
801801
"//tensorflow/core:framework_lite",
802802
"//tensorflow/core:protos_all_cc",
803-
"@local_xla//xla/stream_executor:stream",
804-
"@local_xla//xla/tsl/framework:device_id_utils",
805-
"@local_xla//xla/tsl/platform:statusor",
803+
"@xla//xla/stream_executor:stream",
804+
"@xla//xla/tsl/framework:device_id_utils",
805+
"@xla//xla/tsl/platform:statusor",
806806
],
807807
}),
808808
)

0 commit comments

Comments
 (0)