mirror/tensorflow
1
0
Fork 0
mirror of https://github.com/tensorflow/tensorflow.git synced 2024-11-21 12:51:46 +00:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Move tsl/platform/{cloud,default,windows} to xla/tsl/platform

Browse Source 
PiperOrigin-RevId: 698575496
This commit is contained in:
David Dunleavy 2024-11-20 17:21:33 -08:00 committed by TensorFlower Gardener
parent c5b2880cdd
commit e435325dab
203 changed files with 863 additions and 856 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/BUILD
View File

@ -1489,7 +1489,7 @@ tf_cc_shared_library(
# To avoid duplication, check that the C++ or python library does not depend on
# the stream executor cuda plugins. Targets that want to use cuda APIs should
# instead depend on the dummy plugins in @local_tsl//tsl/platform/default/build_config
# instead depend on the dummy plugins in //tensorflow/compiler/xla/tsl/platform/default/build_config
# and use header only targets.
# TODO(ddunleavy): This seems completely broken. :tensorflow_cc depends on
# cuda_platform from tf_additional_binary_deps and this doesn't break.

8
tensorflow/compiler/mlir/tools/kernel_gen/BUILD
View File

@ -2,14 +2,14 @@ load(
"@local_config_rocm//rocm:build_defs.bzl",
"if_rocm_is_configured",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load(
"@local_xla//xla/stream_executor:build_defs.bzl",
"if_gpu_is_configured",
)
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load(
"//tensorflow:tensorflow.bzl",
"check_deps",

2
tensorflow/compiler/mlir/tools/kernel_gen/transforms/BUILD
View File

@ -4,7 +4,7 @@ load(
"if_rocm_is_configured",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")

2
tensorflow/compiler/tf2tensorrt/BUILD
View File

@ -5,7 +5,7 @@
load("@local_config_tensorrt//:build_defs.bzl", "if_tensorrt")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
)
load("//tensorflow:strict.default.bzl", "py_strict_library")

30
tensorflow/compiler/tf2xla/BUILD
View File

@ -5,13 +5,13 @@ load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured")
# copybara:uncomment_end
load("@local_tsl//tsl/platform:build_config_root.bzl", "if_static")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("@local_xla//xla:xla.bzl", "xla_py_proto_library")
load("@local_xla//xla/service/cpu:build_defs.bzl", "runtime_copts")
load("@local_xla//xla/tsl/mkl:build_defs.bzl", "mkl_deps")
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//tensorflow:strict.default.bzl", "py_strict_library")
load("//tensorflow:tensorflow.bzl", "if_google", "if_libtpu", "tf_cc_binary", "tf_cc_test", "tf_copts", "tf_cuda_cc_test", "tf_gen_op_wrapper_py", "tf_openmp_copts")
load("//tensorflow:tensorflow.default.bzl", "cuda_py_strict_test", "filegroup", "get_compatible_with_portable")
@ -351,17 +351,17 @@ cc_library(
# "@local_tsl//tsl/platform:thread_annotations",
# "@local_tsl//tsl/platform:tstring",
# "@local_tsl//tsl/platform:types",
# "@local_tsl//tsl/platform/default:cord",
# "@local_tsl//tsl/platform/default:env_time",
# "@local_tsl//tsl/platform/default:logging",
# "@local_tsl//tsl/platform/default:mutex",
# "@local_tsl//tsl/platform/default:types",
# "@local_tsl//tsl/platform/google:cord",
# "@local_tsl//tsl/platform/google:env_time",
# "@local_tsl//tsl/platform/google:logging",
# "@local_tsl//tsl/platform/google:mutex",
# "@local_tsl//tsl/platform/google:types",
# "@local_tsl//tsl/platform/windows:env_time",
# "@local_xla//xla/tsl/platform/default:cord",
# "@local_xla//xla/tsl/platform/default:env_time",
# "@local_xla//xla/tsl/platform/default:logging",
# "@local_xla//xla/tsl/platform/default:mutex",
# "@local_xla//xla/tsl/platform/default:types",
# "@local_xla//xla/tsl/platform/google:cord",
# "@local_xla//xla/tsl/platform/google:env_time",
# "@local_xla//xla/tsl/platform/google:logging",
# "@local_xla//xla/tsl/platform/google:mutex",
# "@local_xla//xla/tsl/platform/google:types",
# "@local_xla//xla/tsl/platform/windows:env_time",
# "//tensorflow/core/platform:bfloat16",
# "//tensorflow/core/platform:blocking_counter",
# "//tensorflow/core/platform:byte_order",

11
tensorflow/core/BUILD
View File

@ -64,8 +64,6 @@
# Placeholder: load py_proto_library
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
load("@local_config_tensorrt//:build_defs.bzl", "if_tensorrt")
# load("//tools/build_defs/go:go_library.bzl", "go_library")
load(
"@local_xla//xla/tsl/mkl:build_defs.bzl",
"if_mkl",
@ -123,6 +121,7 @@ load(
"//tensorflow/core/platform:rules_cc.bzl",
"cc_library",
)
# load("//tools/build_defs/go:go_library.bzl", "go_library")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
@ -490,10 +489,10 @@ cc_library(
"//tensorflow/core/framework:type_traits.h",
"//tensorflow/core/platform:framework_lite_hdrs",
"@local_tsl//tsl/platform:framework_lite_hdrs",
"@local_tsl//tsl/platform/default:integral_types.h",
"@local_tsl//tsl/platform/default:logging.h",
"@local_xla//xla/tsl/framework:numeric_types.h",
"@local_xla//xla/tsl/framework:type_traits.h",
"@local_xla//xla/tsl/platform/default:integral_types.h",
"@local_xla//xla/tsl/platform/default:logging.h",
],
visibility = ["//visibility:public"],
deps =
@ -1521,8 +1520,8 @@ cc_library(
hdrs = [
"//tensorflow/core/platform:tflite_portable_logging_hdrs",
"@local_tsl//tsl/platform:tflite_portable_logging_hdrs",
"@local_tsl//tsl/platform/default:integral_types.h",
"@local_tsl//tsl/platform/default:logging.h",
"@local_xla//xla/tsl/platform/default:integral_types.h",
"@local_xla//xla/tsl/platform/default:logging.h",
],
compatible_with = get_compatible_with_portable(),
copts = tf_copts(),

4
tensorflow/core/common_runtime/eager/context.cc
View File

@ -168,8 +168,8 @@ EagerContext::EagerContext(
&func_lib_def_, opts.config.graph_options().optimizer_options(),
thread_pool_.get(), cluster_flr);
// Starts exporting metrics through a platform-specific monitoring API (if
// provided). For builds using "tensorflow/tsl/platform/default", this is
// currently a no-op.
// provided). For builds using "tensorflow/compiler/xla/tsl/platform/default",
// this is currently a no-op.
eager_context_created->GetCell()->Set(true);
InitPrioritizedDeviceTypeList();
runner_ = [this](std::function<void()> closure) {

8
tensorflow/core/common_runtime/session.cc
View File

@ -63,8 +63,8 @@ absl::Status Session::PRun(
Session* NewSession(const SessionOptions& options) {
// Starts exporting metrics through a platform-specific monitoring API (if
// provided). For builds using "tensorflow/tsl/platform/default", this is
// currently a no-op.
// provided). For builds using "tensorflow/compiler/xla/tsl/platform/default",
// this is currently a no-op.
SetSessionCreatedMetric();
Session* out_session;
absl::Status s = NewSession(options, &out_session);
@ -84,8 +84,8 @@ absl::Status NewSession(const SessionOptions& options, Session** out_session) {
return s;
}
// Starts exporting metrics through a platform-specific monitoring API (if
// provided). For builds using "tensorflow/tsl/platform/default", this is
// currently a no-op.
// provided). For builds using "tensorflow/compiler/xla/tsl/platform/default",
// this is currently a no-op.
SetSessionCreatedMetric();
s = factory->NewSession(options, out_session);
if (!s.ok()) {

8
tensorflow/core/distributed_runtime/worker_session.cc
View File

@ -142,8 +142,8 @@ WorkerSession::WorkerSession(
borrowed_device_mgr_(nullptr),
remote_device_mgr_(std::move(remote_device_mgr)) {
// Starts exporting metrics through a platform-specific monitoring API (if
// provided). For builds using "tensorflow/tsl/platform/default", this is
// currently a no-op.
// provided). For builds using "tensorflow/compiler/xla/tsl/platform/default",
// this is currently a no-op.
worker_session_created->GetCell()->Set(true);
}
@ -191,8 +191,8 @@ WorkerSession::WorkerSession(
borrowed_device_mgr_(borrowed_device_mgr),
remote_device_mgr_(std::move(remote_device_mgr)) {
// Starts exporting metrics through a platform-specific monitoring API (if
// provided). For builds using "tensorflow/tsl/platform/default", this is
// currently a no-op.
// provided). For builds using "tensorflow/compiler/xla/tsl/platform/default",
// this is currently a no-op.
worker_session_created->GetCell()->Set(true);
}

2
tensorflow/core/kernels/rnn/BUILD
View File

@ -6,7 +6,7 @@ load(
"if_rocm_is_configured",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load(

2
tensorflow/core/kernels/sparse_matmul_op.h
View File

@ -21,7 +21,7 @@ limitations under the License.
#include "tensorflow/core/platform/types.h"
#if defined(PLATFORM_WINDOWS)
#include "tsl/platform/windows/intrinsics_port.h"
#include "xla/tsl/platform/windows/intrinsics_port.h"
#endif
namespace Eigen {

4
tensorflow/core/lib/gif/BUILD
View File

@ -51,8 +51,8 @@ cc_library(
"//tensorflow/core/lib/gtl:legacy_android_gif_internal_headers",
"//tensorflow/core/platform:gif_internal_hdrs",
"@local_tsl//tsl/platform:gif_internal_hdrs",
"@local_tsl//tsl/platform/default:integral_types.h",
"@local_tsl//tsl/platform/default:logging.h",
"@local_xla//xla/tsl/platform/default:integral_types.h",
"@local_xla//xla/tsl/platform/default:logging.h",
],
copts = tf_copts(),
features = ["-layering_check"],

4
tensorflow/core/lib/jpeg/BUILD
View File

@ -58,8 +58,8 @@ cc_library(
"//tensorflow/core/lib/core:legacy_lib_core_stringpiece_header",
"//tensorflow/core/platform:jpeg_internal_hdrs",
"@local_tsl//tsl/platform:jpeg_internal_hdrs",
"@local_tsl//tsl/platform/default:integral_types.h",
"@local_tsl//tsl/platform/default:logging.h",
"@local_xla//xla/tsl/platform/default:integral_types.h",
"@local_xla//xla/tsl/platform/default:logging.h",
],
copts = tf_copts(),
linkopts = if_android(["-ldl"]),

8
tensorflow/core/platform/BUILD
View File

@ -14,7 +14,7 @@ load(
"if_rocm_is_configured",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//tensorflow:strict.default.bzl", "py_strict_test")
@ -454,7 +454,7 @@ filegroup(
"types.h",
":ctstring",
] + if_windows([
"@local_tsl//tsl/platform/windows:xla_cpu_runtime_srcs",
"@local_xla//xla/tsl/platform/windows:xla_cpu_runtime_srcs",
]),
)
@ -1400,13 +1400,13 @@ filegroup(
"subprocess.h",
"thread_annotations.h",
":base_hdrs",
# copybara:uncomment(GoogleFileSystem) "@local_xla//xla/tsl/platform/google:env.h",
"//tensorflow/core/platform/profile_utils:android_armv7a_cpu_utils_helper.h",
"//tensorflow/core/platform/profile_utils:clock_cycle_profiler.h",
"//tensorflow/core/platform/profile_utils:cpu_utils.h",
"//tensorflow/core/platform/profile_utils:i_cpu_utils_helper.h",
"@local_tsl//tsl/platform:base_hdrs",
"@local_tsl//tsl/platform:lib_hdrs",
# copybara:uncomment(GoogleFileSystem) "@local_tsl//tsl/platform/google:env.h",
],
visibility = ["//tensorflow/core:__pkg__"],
)
@ -1669,7 +1669,7 @@ filegroup(
"//tensorflow/core/platform/profile_utils:cpu_utils.h",
"//tensorflow/core/platform/profile_utils:i_cpu_utils_helper.h",
"@local_tsl//tsl/platform:mobile_srcs_only_runtime",
"@local_tsl//tsl/platform/default:mobile_srcs_only_runtime",
"@local_xla//xla/tsl/platform/default:mobile_srcs_only_runtime",
] + if_not_fuchsia([
"subprocess.h",
]) + tf_google_mobile_srcs_only_runtime(),

2
tensorflow/core/platform/build_config_root.bzl
View File

@ -1,7 +1,7 @@
"""Provides a redirection point for platform specific implementations of starlark utilities."""
load(
"@local_tsl//tsl/platform/default:build_config_root.bzl",
"@local_xla//xla/tsl/platform/default:build_config_root.bzl",
_if_llvm_aarch32_available = "if_llvm_aarch32_available",
_if_llvm_aarch64_available = "if_llvm_aarch64_available",
_if_llvm_arm_available = "if_llvm_arm_available",

32
tensorflow/core/platform/cloud/BUILD
View File

@ -31,7 +31,7 @@ cc_library(
copts = tsl_copts(),
deps = [
"//tensorflow/core:lib",
"@local_tsl//tsl/platform/cloud:expiring_lru_cache",
"@local_xla//xla/tsl/platform/cloud:expiring_lru_cache",
],
)
@ -42,7 +42,7 @@ cc_library(
deps = [
"//tensorflow/core:lib",
"//tensorflow/core/platform:stringpiece",
"@local_tsl//tsl/platform/cloud:file_block_cache",
"@local_xla//xla/tsl/platform/cloud:file_block_cache",
],
)
@ -55,7 +55,7 @@ cc_library(
":file_block_cache",
"//tensorflow/core:lib",
"//tensorflow/core/platform:stringpiece",
"@local_tsl//tsl/platform/cloud:ram_file_block_cache",
"@local_xla//xla/tsl/platform/cloud:ram_file_block_cache",
],
)
@ -66,7 +66,7 @@ cc_library(
deps = [
":http_request",
"//tensorflow/core:lib",
"@local_tsl//tsl/platform/cloud:gcs_dns_cache",
"@local_xla//xla/tsl/platform/cloud:gcs_dns_cache",
],
)
@ -76,7 +76,7 @@ cc_library(
copts = tsl_copts(),
deps = [
"//tensorflow/core:lib",
"@local_tsl//tsl/platform/cloud:gcs_throttle",
"@local_xla//xla/tsl/platform/cloud:gcs_throttle",
],
)
@ -109,7 +109,7 @@ cc_library(
"//tensorflow/core/platform:stringprintf",
"//tensorflow/core/profiler/lib:traceme",
"@jsoncpp_git//:jsoncpp",
"@local_tsl//tsl/platform/cloud:gcs_file_system",
"@local_xla//xla/tsl/platform/cloud:gcs_file_system",
],
alwayslink = 1,
)
@ -146,7 +146,7 @@ cc_library(
"//tensorflow/core/platform:stringprintf",
"//tensorflow/core/profiler/lib:traceme",
"@jsoncpp_git//:jsoncpp",
"@local_tsl//tsl/platform/cloud:gcs_file_system_disabled",
"@local_xla//xla/tsl/platform/cloud:gcs_file_system_disabled",
],
alwayslink = 1,
)
@ -163,7 +163,7 @@ cc_library(
"//tensorflow/core:framework_headers_lib",
"//tensorflow/core:lib_internal",
"//tensorflow/core/platform:stringpiece",
"@local_tsl//tsl/platform/cloud:http_request",
"@local_xla//xla/tsl/platform/cloud:http_request",
],
)
@ -180,7 +180,7 @@ cc_library(
"//tensorflow/core/platform:stringpiece",
"//tensorflow/core/util:env_var",
"@curl",
"@local_tsl//tsl/platform/cloud:curl_http_request",
"@local_xla//xla/tsl/platform/cloud:curl_http_request",
],
)
@ -200,7 +200,7 @@ cc_library(
"//tensorflow/core/platform:status",
"//tensorflow/core/platform:stringpiece",
"@curl",
"@local_tsl//tsl/platform/cloud:http_request_fake",
"@local_xla//xla/tsl/platform/cloud:http_request_fake",
],
)
@ -227,7 +227,7 @@ cc_library(
"//tensorflow/core/platform:status",
"@com_google_absl//absl/strings",
"@jsoncpp_git//:jsoncpp",
"@local_tsl//tsl/platform/cloud:google_auth_provider",
"@local_xla//xla/tsl/platform/cloud:google_auth_provider",
],
)
@ -243,7 +243,7 @@ cc_library(
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
"//tensorflow/core/platform:retrying_utils",
"@local_tsl//tsl/platform/cloud:compute_engine_metadata_client",
"@local_xla//xla/tsl/platform/cloud:compute_engine_metadata_client",
],
)
@ -261,7 +261,7 @@ cc_library(
"//tensorflow/core/platform:errors",
"//tensorflow/core/platform:status",
"//tensorflow/core/platform:str_util",
"@local_tsl//tsl/platform/cloud:compute_engine_zone_provider",
"@local_xla//xla/tsl/platform/cloud:compute_engine_zone_provider",
],
)
@ -273,7 +273,7 @@ cc_library(
deps = [
"//tensorflow/core:lib",
"//tensorflow/core:lib_internal",
"@local_tsl//tsl/platform/cloud:now_seconds_env",
"@local_xla//xla/tsl/platform/cloud:now_seconds_env",
],
)
@ -293,7 +293,7 @@ cc_library(
"//tensorflow/core/platform:status",
"@boringssl//:crypto",
"@jsoncpp_git//:jsoncpp",
"@local_tsl//tsl/platform/cloud:oauth_client",
"@local_xla//xla/tsl/platform/cloud:oauth_client",
],
)
@ -306,6 +306,6 @@ cc_library(
deps = [
"//tensorflow/core:framework_headers_lib",
"//tensorflow/core:lib_internal",
"@local_tsl//tsl/platform/cloud:time_util",
"@local_xla//xla/tsl/platform/cloud:time_util",
],
)

2
tensorflow/core/platform/cloud/auth_provider.h
View File

@ -18,9 +18,9 @@ limitations under the License.
#include <string>
#include "xla/tsl/platform/cloud/auth_provider.h"
#include "tensorflow/core/platform/errors.h"
#include "tensorflow/core/platform/status.h"
#include "tsl/platform/cloud/auth_provider.h"
namespace tensorflow {
// NOLINTBEGIN(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/compute_engine_metadata_client.h
View File

@ -16,10 +16,10 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_PLATFORM_CLOUD_COMPUTE_ENGINE_METADATA_CLIENT_H_
#define TENSORFLOW_CORE_PLATFORM_CLOUD_COMPUTE_ENGINE_METADATA_CLIENT_H_
#include "xla/tsl/platform/cloud/compute_engine_metadata_client.h"
#include "tensorflow/core/platform/cloud/http_request.h"
#include "tensorflow/core/platform/retrying_utils.h"
#include "tensorflow/core/platform/status.h"
#include "tsl/platform/cloud/compute_engine_metadata_client.h"
namespace tensorflow {
using tsl::ComputeEngineMetadataClient; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/compute_engine_zone_provider.h
View File

@ -16,9 +16,9 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_PLATFORM_CLOUD_COMPUTE_ENGINE_ZONE_PROVIDER_H_
#define TENSORFLOW_CORE_PLATFORM_CLOUD_COMPUTE_ENGINE_ZONE_PROVIDER_H_
#include "xla/tsl/platform/cloud/compute_engine_zone_provider.h"
#include "tensorflow/core/platform/cloud/compute_engine_metadata_client.h"
#include "tensorflow/core/platform/cloud/zone_provider.h"
#include "tsl/platform/cloud/compute_engine_zone_provider.h"
namespace tensorflow {
using tsl::ComputeEngineZoneProvider; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/curl_http_request.h
View File

@ -21,6 +21,7 @@ limitations under the License.
#include <vector>
#include <curl/curl.h>
#include "xla/tsl/platform/cloud/curl_http_request.h"
#include "tensorflow/core/platform/cloud/http_request.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/errors.h"
@ -29,7 +30,6 @@ limitations under the License.
#include "tensorflow/core/platform/status.h"
#include "tensorflow/core/platform/stringpiece.h"
#include "tensorflow/core/platform/types.h"
#include "tsl/platform/cloud/curl_http_request.h"
namespace tensorflow {
// NOLINTBEGIN(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/expiring_lru_cache.h
View File

@ -21,11 +21,11 @@ limitations under the License.
#include <memory>
#include <string>
#include "xla/tsl/platform/cloud/expiring_lru_cache.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/mutex.h"
#include "tensorflow/core/platform/thread_annotations.h"
#include "tensorflow/core/platform/types.h"
#include "tsl/platform/cloud/expiring_lru_cache.h"
namespace tensorflow {
using tsl::ExpiringLRUCache; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/file_block_cache.h
View File

@ -23,6 +23,7 @@ limitations under the License.
#include <string>
#include <vector>
#include "xla/tsl/platform/cloud/file_block_cache.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/mutex.h"
#include "tensorflow/core/platform/notification.h"
@ -30,7 +31,6 @@ limitations under the License.
#include "tensorflow/core/platform/stringpiece.h"
#include "tensorflow/core/platform/thread_annotations.h"
#include "tensorflow/core/platform/types.h"
#include "tsl/platform/cloud/file_block_cache.h"
namespace tensorflow {
// NOLINTBEGIN(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/gcs_file_system.h
View File

@ -21,6 +21,7 @@ limitations under the License.
#include <utility>
#include <vector>
#include "xla/tsl/platform/cloud/gcs_file_system.h"
#include "tensorflow/core/platform/cloud/auth_provider.h"
#include "tensorflow/core/platform/cloud/compute_engine_metadata_client.h"
#include "tensorflow/core/platform/cloud/compute_engine_zone_provider.h"
@ -32,7 +33,6 @@ limitations under the License.
#include "tensorflow/core/platform/file_system.h"
#include "tensorflow/core/platform/retrying_file_system.h"
#include "tensorflow/core/platform/status.h"
#include "tsl/platform/cloud/gcs_file_system.h"
namespace tensorflow {
// NOLINTBEGIN(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/gcs_throttle.h
View File

@ -16,8 +16,8 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_PLATFORM_CLOUD_GCS_THROTTLE_H_
#define TENSORFLOW_CORE_PLATFORM_CLOUD_GCS_THROTTLE_H_
#include "xla/tsl/platform/cloud/gcs_throttle.h"
#include "tensorflow/core/platform/env.h"
#include "tsl/platform/cloud/gcs_throttle.h"
namespace tensorflow {
// NOLINTBEGIN(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/google_auth_provider.h
View File

@ -18,12 +18,12 @@ limitations under the License.
#include <memory>
#include "xla/tsl/platform/cloud/google_auth_provider.h"
#include "tensorflow/core/platform/cloud/auth_provider.h"
#include "tensorflow/core/platform/cloud/compute_engine_metadata_client.h"
#include "tensorflow/core/platform/cloud/oauth_client.h"
#include "tensorflow/core/platform/mutex.h"
#include "tensorflow/core/platform/thread_annotations.h"
#include "tsl/platform/cloud/google_auth_provider.h"
namespace tensorflow {
using tsl::GoogleAuthProvider; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/http_request.h
View File

@ -20,6 +20,7 @@ limitations under the License.
#include <unordered_map>
#include <vector>
#include "xla/tsl/platform/cloud/http_request.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/errors.h"
#include "tensorflow/core/platform/macros.h"
@ -27,7 +28,6 @@ limitations under the License.
#include "tensorflow/core/platform/status.h"
#include "tensorflow/core/platform/stringpiece.h"
#include "tensorflow/core/platform/types.h"
#include "tsl/platform/cloud/http_request.h"
namespace tensorflow {
using tsl::HttpRequest; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/http_request_fake.h
View File

@ -22,6 +22,7 @@ limitations under the License.
#include <vector>
#include <curl/curl.h>
#include "xla/tsl/platform/cloud/http_request_fake.h"
#include "tensorflow/core/lib/core/status_test_util.h"
#include "tensorflow/core/platform/cloud/curl_http_request.h"
#include "tensorflow/core/platform/errors.h"
@ -31,7 +32,6 @@ limitations under the License.
#include "tensorflow/core/platform/stringpiece.h"
#include "tensorflow/core/platform/test.h"
#include "tensorflow/core/platform/types.h"
#include "tsl/platform/cloud/http_request_fake.h"
namespace tensorflow {
// NOLINTBEGIN(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/now_seconds_env.h
View File

@ -16,10 +16,10 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_PLATFORM_CLOUD_NOW_SECONDS_ENV_H_
#define TENSORFLOW_CORE_PLATFORM_CLOUD_NOW_SECONDS_ENV_H_
#include "xla/tsl/platform/cloud/now_seconds_env.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/mutex.h"
#include "tensorflow/core/platform/types.h"
#include "tsl/platform/cloud/now_seconds_env.h"
namespace tensorflow {
using tsl::NowSecondsEnv; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/oauth_client.h
View File

@ -19,10 +19,10 @@ limitations under the License.
#include <memory>
#include "json/json.h"
#include "xla/tsl/platform/cloud/oauth_client.h"
#include "tensorflow/core/platform/cloud/http_request.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/status.h"
#include "tsl/platform/cloud/oauth_client.h"
namespace tensorflow {
using tsl::OAuthClient; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/ram_file_block_cache.h
View File

@ -23,6 +23,7 @@ limitations under the License.
#include <string>
#include <vector>
#include "xla/tsl/platform/cloud/ram_file_block_cache.h"
#include "tensorflow/core/platform/cloud/file_block_cache.h"
#include "tensorflow/core/platform/env.h"
#include "tensorflow/core/platform/mutex.h"
@ -31,7 +32,6 @@ limitations under the License.
#include "tensorflow/core/platform/stringpiece.h"
#include "tensorflow/core/platform/thread_annotations.h"
#include "tensorflow/core/platform/types.h"
#include "tsl/platform/cloud/ram_file_block_cache.h"
namespace tensorflow {
using tsl::RamFileBlockCache; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/time_util.h
View File

@ -16,8 +16,8 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_PLATFORM_CLOUD_TIME_UTIL_H_
#define TENSORFLOW_CORE_PLATFORM_CLOUD_TIME_UTIL_H_
#include "xla/tsl/platform/cloud/time_util.h"
#include "tensorflow/core/platform/status.h"
#include "tsl/platform/cloud/time_util.h"
namespace tensorflow {
using tsl::ParseRfc3339Time; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/cloud/zone_provider.h
View File

@ -18,9 +18,9 @@ limitations under the License.
#include <string>
#include "xla/tsl/platform/cloud/zone_provider.h"
#include "tensorflow/core/platform/errors.h"
#include "tensorflow/core/platform/status.h"
#include "tsl/platform/cloud/zone_provider.h"
namespace tensorflow {
using tsl::ZoneProvider; // NOLINT(misc-unused-using-decls)

2
tensorflow/core/platform/rules_cc.bzl
View File

@ -1,7 +1,7 @@
"""Provides an indirection layer to bazel cc_rules"""
load(
"@local_tsl//tsl/platform/default:rules_cc.bzl",
"@local_xla//xla/tsl/platform/default:rules_cc.bzl",
_cc_binary = "cc_binary",
_cc_import = "cc_import",
_cc_library = "cc_library",

4
tensorflow/core/platform/tracing.h
View File

@ -45,9 +45,9 @@ using tsl::tracing::SetEventCollector;
} // namespace tensorflow
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/tracing_impl.h"
#include "xla/tsl/platform/google/tracing_impl.h"
#else
#include "tsl/platform/default/tracing_impl.h"
#include "xla/tsl/platform/default/tracing_impl.h"
#endif
#endif // TENSORFLOW_CORE_PLATFORM_TRACING_H_

2
tensorflow/core/profiler/backends/gpu/BUILD
View File

@ -1,5 +1,5 @@
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//tensorflow:tensorflow.default.bzl", "tf_cuda_cc_test")

2
tensorflow/lite/acceleration/configuration/BUILD
View File

@ -18,7 +18,7 @@
load("@flatbuffers//:build_defs.bzl", "DEFAULT_FLATC_ARGS", "flatbuffer_android_library", "flatbuffer_cc_library", "flatbuffer_java_library", "flatc_path")
# copybara:comment_begin(oss-only)
load("@local_tsl//tsl/platform/default:build_config.bzl", "tf_proto_library_py")
load("@local_xla//xla/tsl/platform/default:build_config.bzl", "tf_proto_library_py")
# copybara:comment_end
load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")

2
tensorflow/lite/experimental/acceleration/configuration/BUILD
View File

@ -18,7 +18,7 @@
load("@flatbuffers//:build_defs.bzl", "DEFAULT_FLATC_ARGS", "flatbuffer_android_library", "flatbuffer_cc_library", "flatbuffer_java_library")
# copybara:comment_begin(oss-only)
load("@local_tsl//tsl/platform/default:build_config.bzl", "tf_proto_library_py")
load("@local_xla//xla/tsl/platform/default:build_config.bzl", "tf_proto_library_py")
# copybara:comment_end
load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")

6
tensorflow/lite/kernels/CMakeLists.txt
View File

@ -95,9 +95,9 @@ set(TEST_FRAMEWORK_SRC
${TFLITE_SOURCE_DIR}/tools/optimize/quantization_utils.cc
${TFLITE_SOURCE_DIR}/tools/tool_params.cc
${TFLITE_SOURCE_DIR}/tools/versioning/op_version.cc
${TSL_SOURCE_DIR}/tsl/platform/default/env_time.cc
${TSL_SOURCE_DIR}/tsl/platform/default/logging.cc
${TSL_SOURCE_DIR}/tsl/platform/default/mutex.cc
${XLA_SOURCE_DIR}/xla/tsl/platform/default/env_time.cc
${XLA_SOURCE_DIR}/xla/tsl/platform/default/logging.cc
${XLA_SOURCE_DIR}/xla/tsl/platform/default/mutex.cc
internal/test_util.cc
acceleration_test_util.cc
acceleration_test_util_internal.cc

2
tensorflow/python/BUILD
View File

@ -3,7 +3,7 @@
# ":platform" - Low-level and platform-specific Python code.
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//tensorflow:pytype.default.bzl", "pytype_strict_library")

2
third_party/gpus/cuda_configure.bzl vendored
View File

@ -710,7 +710,7 @@ filegroup(name="cudnn-include")
)
# Create dummy files for the CUDA toolkit since they are still required by
# tensorflow/tsl/platform/default/build_config:cuda.
# tensorflow/compiler/xla/tsl/platform/default/build_config:cuda.
repository_ctx.file("cuda/cuda/include/cuda.h")
repository_ctx.file("cuda/cuda/include/cublas.h")
repository_ctx.file("cuda/cuda/include/cudnn.h")

2
third_party/nccl/system.BUILD.tpl vendored
View File

@ -1,6 +1,6 @@
load("@bazel_skylib//rules:write_file.bzl", "write_file")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags"
)

2
third_party/pprof.BUILD vendored
View File

@ -1,4 +1,4 @@
load("@local_tsl//tsl/platform/default:build_config.bzl", "py_proto_library")
load("@local_xla//xla/tsl/platform/default:build_config.bzl", "py_proto_library")
package(
default_visibility = ["//visibility:public"],

2
third_party/riegeli_fix.patch vendored
View File

@ -7,7 +7,7 @@ index cde6d546..7fe18a93 100644
+++ b/python/riegeli/records/BUILD
@@ -1,5 +1,5 @@
-load("@com_google_protobuf//:protobuf.bzl", "py_proto_library")
+load("@local_tsl//tsl/platform/default:build_config.bzl", "py_proto_library")
+load("@local_xla//xla/tsl/platform/default:build_config.bzl", "py_proto_library")
load("//python/riegeli:py_extension.bzl", "py_extension")
package(

2
third_party/xla/third_party/tsl/third_party/gpus/cuda_configure.bzl vendored
View File

@ -710,7 +710,7 @@ filegroup(name="cudnn-include")
)
# Create dummy files for the CUDA toolkit since they are still required by
# tensorflow/tsl/platform/default/build_config:cuda.
# tensorflow/compiler/xla/tsl/platform/default/build_config:cuda.
repository_ctx.file("cuda/cuda/include/cuda.h")
repository_ctx.file("cuda/cuda/include/cublas.h")
repository_ctx.file("cuda/cuda/include/cudnn.h")

2
third_party/xla/third_party/tsl/third_party/nccl/system.BUILD.tpl vendored
View File

@ -1,6 +1,6 @@
load("@bazel_skylib//rules:write_file.bzl", "write_file")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags"
)

3
third_party/xla/third_party/tsl/tsl/platform/BUILD vendored
View File

@ -55,6 +55,7 @@ exports_files(
visibility = internal_visibility([
"//tensorflow/core/platform:__subpackages__",
"//tsl:__subpackages__",
"@local_xla//xla/tsl:__subpackages__",
]),
)
@ -619,7 +620,7 @@ filegroup(
"//conditions:default": [
"file_system_helper.cc",
"tracing.cc",
"//tsl/platform/default:mobile_srcs_no_runtime",
"@local_xla//xla/tsl/platform/default:mobile_srcs_no_runtime",
],
}),
compatible_with = get_compatible_with_portable(),

2
third_party/xla/third_party/tsl/tsl/platform/build_config.bzl vendored
View File

@ -1,7 +1,7 @@
"""Provides a redirection point for platform specific implementations of starlark utilities."""
load(
"//tsl/platform/default:build_config.bzl",
"@local_xla//xla/tsl/platform/default:build_config.bzl",
_pyx_library = "pyx_library",
_tf_additional_all_protos = "tf_additional_all_protos",
_tf_additional_core_deps = "tf_additional_core_deps",

2
third_party/xla/third_party/tsl/tsl/platform/build_config_root.bzl vendored
View File

@ -1,7 +1,7 @@
"""Provides a redirection point for platform specific implementations of starlark utilities."""
load(
"//tsl/platform/default:build_config_root.bzl",
"@local_xla//xla/tsl/platform/default:build_config_root.bzl",
_if_llvm_aarch32_available = "if_llvm_aarch32_available",
_if_llvm_aarch64_available = "if_llvm_aarch64_available",
_if_llvm_arm_available = "if_llvm_arm_available",

4
third_party/xla/third_party/tsl/tsl/platform/casts.h vendored
View File

@ -19,11 +19,11 @@ limitations under the License.
#include "tsl/platform/platform.h"
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/casts.h" // IWYU pragma: export
#include "xla/tsl/platform/google/casts.h" // IWYU pragma: export
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_POSIX_IOS) || \
defined(PLATFORM_GOOGLE_IOS) || defined(PLATFORM_WINDOWS)
#include "tsl/platform/default/casts.h" // IWYU pragma: export
#include "xla/tsl/platform/default/casts.h" // IWYU pragma: export
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform
#endif

4
third_party/xla/third_party/tsl/tsl/platform/context.h vendored
View File

@ -39,9 +39,9 @@ class WithContext;
} // namespace tsl
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/context.h" // IWYU pragma: export
#include "xla/tsl/platform/google/context.h" // IWYU pragma: export
#else
#include "tsl/platform/default/context.h" // IWYU pragma: export
#include "xla/tsl/platform/default/context.h" // IWYU pragma: export
#endif
#endif // TENSORFLOW_TSL_PLATFORM_CONTEXT_H_

4
third_party/xla/third_party/tsl/tsl/platform/crash_analysis.h vendored
View File

@ -20,9 +20,9 @@ limitations under the License.
// Include appropriate platform-dependent implementations
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/crash_analysis.h" // IWYU pragma: export
#include "xla/tsl/platform/google/crash_analysis.h" // IWYU pragma: export
#else
#include "tsl/platform/default/crash_analysis.h" // IWYU pragma: export
#include "xla/tsl/platform/default/crash_analysis.h" // IWYU pragma: export
#endif
#endif // TENSORFLOW_TSL_PLATFORM_CRASH_ANALYSIS_H_

4
third_party/xla/third_party/tsl/tsl/platform/criticality.h vendored
View File

@ -42,9 +42,9 @@ enum class Criticality {
} // namespace tsl
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/criticality.h" // IWYU pragma: export
#include "xla/tsl/platform/google/criticality.h" // IWYU pragma: export
#else
#include "tsl/platform/default/criticality.h" // IWYU pragma: export
#include "xla/tsl/platform/default/criticality.h" // IWYU pragma: export
#endif
#endif // TENSORFLOW_TSL_PLATFORM_CRITICALITY_H_

4
third_party/xla/third_party/tsl/tsl/platform/dso_loader.h vendored
View File

@ -20,10 +20,10 @@ limitations under the License.
// Include appropriate platform-dependent implementations
#if defined(PLATFORM_GOOGLE) || defined(PLATFORM_CHROMIUMOS)
#include "tsl/platform/google/dso_loader.h"
#include "xla/tsl/platform/google/dso_loader.h"
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_WINDOWS)
#include "tsl/platform/default/dso_loader.h"
#include "xla/tsl/platform/default/dso_loader.h"
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform
#endif

2
third_party/xla/third_party/tsl/tsl/platform/env.cc vendored
View File

@ -39,7 +39,7 @@ limitations under the License.
#include <windows.h>
#undef DeleteFile
#undef CopyFile
#include "tsl/platform/windows/wide_char.h"
#include "xla/tsl/platform/windows/wide_char.h"
#define PATH_MAX MAX_PATH
#else
#include <fcntl.h>

4
third_party/xla/third_party/tsl/tsl/platform/logging.h vendored
View File

@ -21,9 +21,9 @@ limitations under the License.
#if defined(PLATFORM_GOOGLE) || defined(PLATFORM_GOOGLE_ANDROID) || \
defined(PLATFORM_GOOGLE_IOS) || defined(GOOGLE_LOGGING) || \
defined(__EMSCRIPTEN__) || defined(PLATFORM_CHROMIUMOS)
#include "tsl/platform/google/logging.h" // IWYU pragma: export
#include "xla/tsl/platform/google/logging.h" // IWYU pragma: export
#else
#include "tsl/platform/default/logging.h" // IWYU pragma: export
#include "xla/tsl/platform/default/logging.h" // IWYU pragma: export
#endif
#endif // TENSORFLOW_TSL_PLATFORM_LOGGING_H_

8
third_party/xla/third_party/tsl/tsl/platform/mutex.h vendored
View File

@ -27,11 +27,11 @@ limitations under the License.
// Include appropriate platform-dependent implementation details of mutex etc.
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/mutex_data.h"
#include "xla/tsl/platform/google/mutex_data.h"
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_POSIX_IOS) || \
defined(PLATFORM_GOOGLE_IOS) || defined(PLATFORM_WINDOWS)
#include "tsl/platform/default/mutex_data.h"
#include "xla/tsl/platform/default/mutex_data.h"
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform
#endif
@ -322,11 +322,11 @@ inline Condition::Condition(const bool* flag)
// Include appropriate platform-dependent implementation details of mutex etc.
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/mutex.h"
#include "xla/tsl/platform/google/mutex.h"
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_POSIX_IOS) || \
defined(PLATFORM_GOOGLE_IOS) || defined(PLATFORM_WINDOWS)
#include "tsl/platform/default/mutex.h"
#include "xla/tsl/platform/default/mutex.h"
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform
#endif

2
third_party/xla/third_party/tsl/tsl/platform/profile_utils/BUILD vendored
View File

@ -14,7 +14,7 @@ package(
"@local_xla//xla/stream_executor:__subpackages__",
"//tensorflow/core/platform:__subpackages__",
"//tsl:__pkg__",
"//tsl/platform/default:__pkg__",
"@local_xla//xla/tsl/platform/default:__pkg__",
]),
licenses = ["notice"],
)

2
third_party/xla/third_party/tsl/tsl/platform/rules_cc.bzl vendored
View File

@ -1,7 +1,7 @@
"""Provides an indirection layer to bazel cc_rules"""
load(
"//tsl/platform/default:rules_cc.bzl",
"@local_xla//xla/tsl/platform/default:rules_cc.bzl",
_cc_binary = "cc_binary",
_cc_import = "cc_import",
_cc_library = "cc_library",

6
third_party/xla/third_party/tsl/tsl/platform/stacktrace.h vendored
View File

@ -20,13 +20,13 @@ limitations under the License.
// Include appropriate platform-dependent implementation.
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/stacktrace.h" // IWYU pragma: export
#include "xla/tsl/platform/google/stacktrace.h" // IWYU pragma: export
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_POSIX_IOS) || \
defined(PLATFORM_GOOGLE_IOS)
#include "tsl/platform/default/stacktrace.h" // IWYU pragma: export
#include "xla/tsl/platform/default/stacktrace.h" // IWYU pragma: export
#elif defined(PLATFORM_WINDOWS)
#include "tsl/platform/windows/stacktrace.h" // IWYU pragma: export
#include "xla/tsl/platform/windows/stacktrace.h" // IWYU pragma: export
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform
#endif

4
third_party/xla/third_party/tsl/tsl/platform/status.h vendored
View File

@ -41,9 +41,9 @@ limitations under the License.
// Include appropriate platform-dependent parts of status.
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/status.h" // IWYU pragma: export
#include "xla/tsl/platform/google/status.h" // IWYU pragma: export
#else
#include "tsl/platform/default/status.h" // IWYU pragma: export
#include "xla/tsl/platform/default/status.h" // IWYU pragma: export
#endif
// TODO: b/323943471 - This macro should eventually be provided by Abseil.

4
third_party/xla/third_party/tsl/tsl/platform/statusor.h vendored
View File

@ -78,9 +78,9 @@ limitations under the License.
// Include appropriate platform-dependent `TF_ASSIGN_OR_RETURN`.
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/statusor.h" // IWYU pragma: export
#include "xla/tsl/platform/google/statusor.h" // IWYU pragma: export
#else
#include "tsl/platform/default/statusor.h" // IWYU pragma: export
#include "xla/tsl/platform/default/statusor.h" // IWYU pragma: export
#endif
// TODO: b/323943471 - This macro should eventually be provided by Abseil.

6
third_party/xla/third_party/tsl/tsl/platform/subprocess.h vendored
View File

@ -59,13 +59,13 @@ std::unique_ptr<SubProcess> CreateSubProcess(const std::vector<string>& argv);
#include "tsl/platform/platform.h"
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/subprocess.h"
#include "xla/tsl/platform/google/subprocess.h"
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_POSIX_IOS) || \
defined(PLATFORM_GOOGLE_IOS)
#include "tsl/platform/default/subprocess.h" // IWYU pragma: export
#include "xla/tsl/platform/default/subprocess.h" // IWYU pragma: export
#elif defined(PLATFORM_WINDOWS)
#include "tsl/platform/windows/subprocess.h" // IWYU pragma: export
#include "xla/tsl/platform/windows/subprocess.h" // IWYU pragma: export
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform
#endif

4
third_party/xla/third_party/tsl/tsl/platform/tracing.h vendored
View File

@ -142,9 +142,9 @@ const char* GetLogDir();
} // namespace tsl
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/tracing_impl.h"
#include "xla/tsl/platform/google/tracing_impl.h"
#else
#include "tsl/platform/default/tracing_impl.h"
#include "xla/tsl/platform/default/tracing_impl.h"
#endif
#endif // TENSORFLOW_TSL_PLATFORM_TRACING_H_

4
third_party/xla/third_party/tsl/tsl/platform/types.h vendored
View File

@ -25,11 +25,11 @@ limitations under the License.
// Include appropriate platform-dependent implementations
#if defined(PLATFORM_GOOGLE) || defined(GOOGLE_INTEGRAL_TYPES)
#include "tsl/platform/google/integral_types.h" // IWYU pragma: export
#include "xla/tsl/platform/google/integral_types.h" // IWYU pragma: export
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_POSIX_IOS) || \
defined(PLATFORM_GOOGLE_IOS) || defined(PLATFORM_WINDOWS)
#include "tsl/platform/default/integral_types.h" // IWYU pragma: export
#include "xla/tsl/platform/default/integral_types.h" // IWYU pragma: export
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform
#endif

4
third_party/xla/third_party/tsl/tsl/platform/unbounded_work_queue.h vendored
View File

@ -22,11 +22,11 @@ limitations under the License.
// whose size automatically increases with demand.
#if defined(PLATFORM_GOOGLE)
#include "tsl/platform/google/unbounded_work_queue.h" // IWYU pragma: export
#include "xla/tsl/platform/google/unbounded_work_queue.h" // IWYU pragma: export
#elif defined(PLATFORM_POSIX) || defined(PLATFORM_POSIX_ANDROID) || \
defined(PLATFORM_GOOGLE_ANDROID) || defined(PLATFORM_POSIX_IOS) || \
defined(PLATFORM_GOOGLE_IOS) || defined(PLATFORM_WINDOWS)
#include "tsl/platform/default/unbounded_work_queue.h" // IWYU pragma: export
#include "xla/tsl/platform/default/unbounded_work_queue.h" // IWYU pragma: export
#else
#error Define the appropriate PLATFORM_<foo> macro for this platform
#endif

8
third_party/xla/third_party/tsl/tsl/profiler/lib/BUILD vendored
View File

@ -1,6 +1,10 @@
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load("@local_xla//xla/tsl:tsl.bzl", "if_not_android", "if_oss", "internal_visibility", "nvtx_headers")
load("@local_xla//xla/tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
load(
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load(
"@local_xla//xla/tsl/profiler/builds:build_config.bzl",
"tf_profiler_copts",
@ -8,10 +12,6 @@ load(
)
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
load("//tsl/platform:build_config_root.bzl", "if_static")
load(
"//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
# copybara:uncomment package(default_applicable_licenses = ["//tensorflow:license"])

2
third_party/xla/xla/BUILD vendored
View File

@ -1363,8 +1363,8 @@ bzl_library(
srcs = ["lit.bzl"],
deps = [
"//xla/tsl:tsl_bzl",
"//xla/tsl/platform/default:cuda_build_defs_bzl",
"@bazel_skylib//lib:paths",
"@local_tsl//tsl/platform/default:cuda_build_defs_bzl",
],
)

8
third_party/xla/xla/backends/profiler/gpu/BUILD vendored
View File

@ -4,10 +4,6 @@ load(
"@local_tsl//tsl/platform:build_config.bzl",
"tf_additional_device_tracer_srcs",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla/tests:build_defs.bzl", "xla_test")
load(
"//xla/tsl:tsl.bzl",
@ -17,6 +13,10 @@ load(
"tsl_copts",
"tsl_gpu_library",
)
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla/tsl/profiler/builds:build_config.bzl", "tf_profiler_copts")
package(

8
third_party/xla/xla/hlo/pass/BUILD vendored
View File

@ -2,12 +2,12 @@
# XLA’s HLO Intermediate Representation implementation.
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/hlo/testlib/BUILD vendored
View File

@ -6,13 +6,13 @@ load(
"if_rocm_is_configured",
)
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:package_groups.bzl", "xla_tests_package_groups")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "filegroup")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/hlo/transforms/BUILD vendored
View File

@ -2,12 +2,12 @@
# Implementation of XLA’s HLO transformations.
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.bzl", "tsl_copts")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/lit.bzl vendored
View File

@ -1,8 +1,8 @@
"""Helper rules for writing LIT tests."""
load("@bazel_skylib//lib:paths.bzl", "paths")
load("@local_tsl//tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
load("//xla/tsl:tsl.bzl", "if_cuda_tools", "if_google", "if_oss")
load("//xla/tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
def enforce_glob(files, **kwargs):
"""A utility to enforce that a list matches a glob expression.

8
third_party/xla/xla/pjrt/c/BUILD vendored
View File

@ -3,10 +3,6 @@ load(
"if_rocm_is_configured",
)
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load(
"//xla:xla.bzl",
"xla_cc_binary",
@ -19,6 +15,10 @@ load(
"if_macos",
"internal_visibility",
)
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/python/BUILD vendored
View File

@ -1315,12 +1315,12 @@ tsl_pybind_extension(
"//xla/service/cpu:collectives_interface",
"//xla/tsl/concurrency:ref_count",
"//xla/tsl/distributed_runtime/preemption:preemption_sync_manager",
"//xla/tsl/platform/cloud:gcs_file_system",
"//xla/tsl/python/lib/core:numpy",
"@local_tsl//tsl/platform",
"@local_tsl//tsl/platform:logging",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:statusor",
"@local_tsl//tsl/platform/cloud:gcs_file_system",
] + select({
# gloo tcp transport only builds on linux
"//xla/tsl:macos": [

8
third_party/xla/xla/service/BUILD vendored
View File

@ -14,10 +14,6 @@ load(
"tf_pyclif_proto_library",
)
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:strict.default.bzl", "py_strict_library", "py_strict_test")
load(
"//xla:xla.bzl",
@ -32,6 +28,10 @@ load("//xla/stream_executor:build_defs.bzl", "if_gpu_is_configured")
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "if_google", "if_libtpu", "internal_visibility", "tsl_copts")
load("//xla/tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable", "internal_hlo_deps")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/service/gpu/BUILD vendored
View File

@ -17,10 +17,6 @@ load(
"@local_tsl//tsl/platform:build_config_root.bzl",
"if_static",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_test", "xla_internal")
load(
"//xla/service/gpu:build_defs.bzl",
@ -42,6 +38,10 @@ load(
"tsl_gpu_library",
)
load("//xla/tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/service/gpu/autotuning/BUILD vendored
View File

@ -5,13 +5,13 @@ load(
"@local_tsl//tsl/platform:build_config.bzl",
"tf_proto_library",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "if_oss")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/service/gpu/build_defs.bzl vendored
View File

@ -3,7 +3,7 @@
load("@local_config_cuda//cuda:build_defs.bzl", "cuda_library")
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured", "rocm_copts", "rocm_library")
load("@local_tsl//tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
load("//xla/tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
load("@bazel_skylib//lib:paths.bzl", "paths")
load("//xla/tests:build_defs.bzl", "prepare_gpu_backend_data")

2
third_party/xla/xla/service/gpu/fusions/BUILD vendored
View File

@ -1,8 +1,8 @@
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured")
load("@local_tsl//tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "if_oss")
load("//xla/tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/service/gpu/fusions/triton/BUILD vendored
View File

@ -1,7 +1,6 @@
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured")
load("@local_tsl//tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
load("//xla:xla.bzl", "xla_cc_test")
load(
"//xla/stream_executor:build_defs.bzl",
@ -9,6 +8,7 @@ load(
)
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "if_google", "if_oss")
load("//xla/tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/service/gpu/kernels/BUILD vendored
View File

@ -1,14 +1,14 @@
load("@local_config_cuda//cuda:build_defs.bzl", "cuda_library")
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_binary")
load("//xla/service/gpu:build_defs.bzl", "gpu_kernel_library")
load("//xla/stream_executor:build_defs.bzl", "if_gpu_is_configured")
load("//xla/tests:build_defs.bzl", "DEFAULT_DISABLED_BACKENDS", "xla_test")
load("//xla/tsl:tsl.bzl", "if_windows")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/service/gpu/llvm_gpu_backend/BUILD vendored
View File

@ -6,16 +6,16 @@ load(
"@local_config_sycl//sycl:build_defs.bzl",
"if_sycl_is_configured",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_test")
load(
"//xla/tsl:tsl.bzl",
"if_google",
"internal_visibility",
)
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/service/gpu/model/BUILD vendored
View File

@ -1,13 +1,13 @@
load("@bazel_skylib//rules:build_test.bzl", "build_test")
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
load("@local_tsl//tsl/platform:build_config.bzl", "tf_proto_library")
load("@local_tsl//tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
load("//xla:xla.bzl", "xla_cc_test")
# Libraries for performance modeling of HLO.
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "if_google", "if_oss", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
load("//xla/tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/service/gpu/runtime/BUILD vendored
View File

@ -1,11 +1,11 @@
load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured")
load("@local_tsl//tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/service/gpu:build_defs.bzl", "get_cub_sort_kernel_types")
load("//xla/stream_executor:build_defs.bzl", "if_gpu_is_configured")
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "if_google", "if_nccl", "if_oss", "internal_visibility", "nvtx_headers")
load("//xla/tsl:tsl.default.bzl", "get_compatible_with_portable")
load("//xla/tsl/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/service/gpu/tests/BUILD vendored
View File

@ -9,10 +9,6 @@ load(
"@local_tsl//tsl/platform:build_config_root.bzl",
"tf_cuda_tests_tags",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:lit.bzl", "enforce_glob", "lit_test_suite")
load(
"//xla:xla.bzl",
@ -22,6 +18,10 @@ load("//xla/stream_executor:build_defs.bzl", "if_gpu_is_configured")
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "if_google", "if_oss")
load("//xla/tsl:tsl.default.bzl", "filegroup")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/service/gpu/transforms/BUILD vendored
View File

@ -1,7 +1,3 @@
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_test")
load(
"//xla/stream_executor:build_defs.bzl",
@ -9,6 +5,10 @@ load(
)
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "if_google", "if_oss")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/service/llvm_ir/BUILD vendored
View File

@ -2,13 +2,13 @@
# Libraries for helping construct LLVM IR for XLA backends.
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_test")
load("//xla/tsl:tsl.bzl", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "filegroup")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

2
third_party/xla/xla/stream_executor/BUILD vendored
View File

@ -29,10 +29,10 @@ bzl_library(
name = "build_defs_bzl",
srcs = ["build_defs.bzl"],
deps = [
"//xla/tsl/platform/default:cuda_build_defs_bzl",
"@local_config_cuda//cuda:build_defs_bzl",
"@local_tsl//third_party/py/rules_pywrap:pywrap_bzl",
"@local_tsl//tsl/platform:rules_cc_bzl",
"@local_tsl//tsl/platform/default:cuda_build_defs_bzl",
] + stream_executor_build_defs_bzl_deps(),
)

8
third_party/xla/xla/stream_executor/cuda/BUILD vendored
View File

@ -8,10 +8,6 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_newer_than",
)
load(
"//xla:xla.bzl",
"xla_cc_test",
@ -32,6 +28,10 @@ load(
"internal_visibility",
"tsl_copts",
)
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_newer_than",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/stream_executor/gpu/BUILD vendored
View File

@ -22,10 +22,6 @@ load(
"@local_tsl//tsl/platform:rules_cc.bzl",
"cc_library",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:xla.bzl", "xla_cc_test")
load(
"//xla/service/gpu:build_defs.bzl",
@ -46,6 +42,10 @@ load(
"tsl_copts",
"tsl_gpu_library",
)
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/tests/BUILD vendored
View File

@ -7,15 +7,15 @@ load(
"if_rocm_is_configured",
)
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:package_groups.bzl", "xla_tests_package_groups")
load("//xla:xla.bzl", "tests_build_defs_bzl_deps", "xla_cc_binary", "xla_cc_test")
load("//xla/tests:build_defs.bzl", "generate_backend_suites", "generate_backend_test_macros", "xla_test", "xla_test_library")
load("//xla/tsl:tsl.bzl", "if_google", "if_oss", "internal_visibility")
load("//xla/tsl:tsl.default.bzl", "filegroup")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

8
third_party/xla/xla/tools/BUILD vendored
View File

@ -8,10 +8,6 @@ load(
"tf_proto_library",
)
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:lit.bzl", "enforce_glob", "lit_test_suite")
load(
"//xla:xla.bzl",
@ -29,6 +25,10 @@ load(
"tsl_gpu_library",
)
load("//xla/tsl:tsl.default.bzl", "filegroup")
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

12
third_party/xla/xla/tools/hlo_opt/BUILD vendored
View File

@ -6,12 +6,6 @@ load(
"@local_tsl//tsl/platform:build_config_root.bzl",
"tf_cuda_tests_tags",
)
# hlo-opt tool.
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
load("//xla:lit.bzl", "enforce_glob", "lit_test_suite")
load(
"//xla/stream_executor:build_defs.bzl",
@ -19,6 +13,12 @@ load(
)
load("//xla/tsl:tsl.default.bzl", "filegroup")
# hlo-opt tool.
load(
"//xla/tsl/platform/default:cuda_build_defs.bzl",
"if_cuda_is_configured",
)
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = ["//xla:internal"],

2
third_party/xla/xla/tsl/cuda/BUILD.bazel vendored
View File

@ -6,7 +6,7 @@ load(
"cc_library",
)
load(
"@local_tsl//tsl/platform/default:cuda_build_defs.bzl",
"@local_xla//xla/tsl/platform/default:cuda_build_defs.bzl",
"cuda_rpath_flags",
"if_cuda_is_configured",
)

2
third_party/xla/xla/tsl/lib/gtl/BUILD vendored
View File

@ -20,7 +20,7 @@ package(
# tensorflow/core/framework uses map_util, and flatmap
"//tensorflow/core/framework:__pkg__",
"//xla/tsl/framework:__pkg__",
"@local_tsl//tsl/platform/cloud:__pkg__",
"//xla/tsl/platform/cloud:__pkg__",
# tensorflow/core/util uses inlined_vector
"//tensorflow/core/util:__pkg__",
# tensorflow/core/tfrt/utils uses inlined_vector

334
third_party/xla/third_party/tsl/tsl/platform/cloud/BUILD → third_party/xla/xla/tsl/platform/cloud/BUILD vendored
View File

@ -1,14 +1,14 @@
# Description:
# Cloud file system implementation.
load("@local_tsl//tsl/platform:build_config.bzl", "tsl_cc_test")
load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library")
load(
"@local_xla//xla/tsl:tsl.bzl",
"//xla/tsl:tsl.bzl",
"if_windows",
"internal_visibility",
"tsl_copts",
)
load("//tsl/platform:build_config.bzl", "tsl_cc_test")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
@ -33,10 +33,10 @@ cc_library(
hdrs = ["expiring_lru_cache.h"],
copts = tsl_copts(),
deps = [
"//tsl/platform:env",
"//tsl/platform:mutex",
"//tsl/platform:thread_annotations",
"//tsl/platform:types",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:mutex",
"@local_tsl//tsl/platform:thread_annotations",
"@local_tsl//tsl/platform:types",
],
)
@ -45,13 +45,13 @@ cc_library(
hdrs = ["file_block_cache.h"],
copts = tsl_copts(),
deps = [
"//tsl/platform:env",
"//tsl/platform:mutex",
"//tsl/platform:notification",
"//tsl/platform:status",
"//tsl/platform:stringpiece",
"//tsl/platform:thread_annotations",
"//tsl/platform:types",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:mutex",
"@local_tsl//tsl/platform:notification",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:stringpiece",
"@local_tsl//tsl/platform:thread_annotations",
"@local_tsl//tsl/platform:types",
],
)
@ -63,14 +63,14 @@ cc_library(
visibility = ["//visibility:public"],
deps = [
":file_block_cache",
"//tsl/platform:env",
"//tsl/platform:mutex",
"//tsl/platform:notification",
"//tsl/platform:status",
"//tsl/platform:stringpiece",
"//tsl/platform:thread_annotations",
"//tsl/platform:types",
"@com_google_absl//absl/cleanup",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:mutex",
"@local_tsl//tsl/platform:notification",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:stringpiece",
"@local_tsl//tsl/platform:thread_annotations",
"@local_tsl//tsl/platform:types",
],
)
@ -81,12 +81,12 @@ cc_library(
copts = tsl_copts(),
deps = [
":http_request",
"//tsl/platform:env",
"//tsl/platform:errors",
"//tsl/platform:retrying_utils",
"//tsl/platform:status",
"@com_google_absl//absl/status",
"@com_google_absl//absl/strings",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:retrying_utils",
"@local_tsl//tsl/platform:status",
],
)
@ -96,7 +96,7 @@ cc_library(
hdrs = ["gcs_throttle.h"],
copts = tsl_copts(),
deps = [
"//tsl/platform:env",
"@local_tsl//tsl/platform:env",
],
)
@ -118,25 +118,25 @@ cc_library(
":http_request",
":ram_file_block_cache",
":time_util",
"//tsl/platform:env",
"//tsl/platform:errors",
"//tsl/platform:file_statistics",
"//tsl/platform:mutex",
"//tsl/platform:numbers",
"//tsl/platform:path",
"//tsl/platform:protobuf",
"//tsl/platform:retrying_file_system",
"//tsl/platform:retrying_utils",
"//tsl/platform:status",
"//tsl/platform:str_util",
"//tsl/platform:strcat",
"//tsl/platform:stringprintf",
"//tsl/platform:thread_annotations",
"//tsl/profiler/lib:traceme",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/status",
"@com_google_absl//absl/strings",
"@jsoncpp_git//:jsoncpp",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:file_statistics",
"@local_tsl//tsl/platform:mutex",
"@local_tsl//tsl/platform:numbers",
"@local_tsl//tsl/platform:path",
"@local_tsl//tsl/platform:protobuf",
"@local_tsl//tsl/platform:retrying_file_system",
"@local_tsl//tsl/platform:retrying_utils",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:str_util",
"@local_tsl//tsl/platform:strcat",
"@local_tsl//tsl/platform:stringprintf",
"@local_tsl//tsl/platform:thread_annotations",
"@local_tsl//tsl/profiler/lib:traceme",
],
alwayslink = 1,
)
@ -162,25 +162,25 @@ cc_library(
":http_request",
":ram_file_block_cache",
":time_util",
"//tsl/platform:env",
"//tsl/platform:errors",
"//tsl/platform:file_statistics",
"//tsl/platform:mutex",
"//tsl/platform:numbers",
"//tsl/platform:path",
"//tsl/platform:protobuf",
"//tsl/platform:retrying_file_system",
"//tsl/platform:retrying_utils",
"//tsl/platform:status",
"//tsl/platform:str_util",
"//tsl/platform:strcat",
"//tsl/platform:stringprintf",
"//tsl/platform:thread_annotations",
"//tsl/profiler/lib:traceme",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/status",
"@com_google_absl//absl/strings",
"@jsoncpp_git//:jsoncpp",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:file_statistics",
"@local_tsl//tsl/platform:mutex",
"@local_tsl//tsl/platform:numbers",
"@local_tsl//tsl/platform:path",
"@local_tsl//tsl/platform:protobuf",
"@local_tsl//tsl/platform:retrying_file_system",
"@local_tsl//tsl/platform:retrying_utils",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:str_util",
"@local_tsl//tsl/platform:strcat",
"@local_tsl//tsl/platform:stringprintf",
"@local_tsl//tsl/platform:thread_annotations",
"@local_tsl//tsl/profiler/lib:traceme",
],
alwayslink = 1,
)
@ -190,13 +190,13 @@ cc_library(
hdrs = ["http_request.h"],
copts = tsl_copts(),
deps = [
"//tsl/platform:env",
"//tsl/platform:errors",
"//tsl/platform:macros",
"//tsl/platform:protobuf",
"//tsl/platform:status",
"//tsl/platform:stringpiece",
"//tsl/platform:types",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:macros",
"@local_tsl//tsl/platform:protobuf",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:stringpiece",
"@local_tsl//tsl/platform:types",
],
)
@ -207,18 +207,18 @@ cc_library(
copts = tsl_copts(),
deps = [
":http_request",
"//tsl/platform:env",
"//tsl/platform:errors",
"//tsl/platform:macros",
"//tsl/platform:protobuf",
"//tsl/platform:scanner",
"//tsl/platform:status",
"//tsl/platform:str_util",
"//tsl/platform:stringpiece",
"//tsl/platform:types",
"//xla/tsl/lib/gtl:map_util",
"//xla/tsl/util:env_var",
"@curl",
"@local_xla//xla/tsl/lib/gtl:map_util",
"@local_xla//xla/tsl/util:env_var",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:macros",
"@local_tsl//tsl/platform:protobuf",
"@local_tsl//tsl/platform:scanner",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:str_util",
"@local_tsl//tsl/platform:stringpiece",
"@local_tsl//tsl/platform:types",
],
)
@ -231,15 +231,15 @@ cc_library(
copts = tsl_copts(),
deps = [
":curl_http_request",
"//tsl/platform:errors",
"//tsl/platform:macros",
"//tsl/platform:protobuf",
"//tsl/platform:status",
"//tsl/platform:stringpiece",
"//tsl/platform:test",
"//tsl/platform:types",
"//xla/tsl/lib/core:status_test_util",
"@curl",
"@local_xla//xla/tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:macros",
"@local_tsl//tsl/platform:protobuf",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:stringpiece",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:types",
],
)
@ -254,16 +254,16 @@ cc_library(
deps = [
":compute_engine_metadata_client",
":oauth_client",
"//tsl/platform:base64",
"//tsl/platform:env",
"//tsl/platform:errors",
"//tsl/platform:mutex",
"//tsl/platform:path",
"//tsl/platform:retrying_utils",
"//tsl/platform:status",
"//tsl/platform:thread_annotations",
"@com_google_absl//absl/strings",
"@jsoncpp_git//:jsoncpp",
"@local_tsl//tsl/platform:base64",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:mutex",
"@local_tsl//tsl/platform:path",
"@local_tsl//tsl/platform:retrying_utils",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:thread_annotations",
],
)
@ -279,10 +279,10 @@ cc_library(
deps = [
":curl_http_request",
":http_request",
"//tsl/platform:env",
"//tsl/platform:retrying_utils",
"//tsl/platform:status",
"@com_google_absl//absl/strings",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:retrying_utils",
"@local_tsl//tsl/platform:status",
],
)
@ -298,9 +298,9 @@ cc_library(
copts = tsl_copts(),
deps = [
":compute_engine_metadata_client",
"//tsl/platform:errors",
"//tsl/platform:status",
"//tsl/platform:str_util",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:str_util",
],
)
@ -310,9 +310,9 @@ cc_library(
hdrs = ["now_seconds_env.h"],
copts = tsl_copts(),
deps = [
"//tsl/platform:env",
"//tsl/platform:mutex",
"//tsl/platform:types",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:mutex",
"@local_tsl//tsl/platform:types",
],
)
@ -328,12 +328,12 @@ cc_library(
deps = [
":curl_http_request",
":http_request",
"//tsl/platform:base64",
"//tsl/platform:env",
"//tsl/platform:errors",
"//tsl/platform:status",
"@boringssl//:crypto",
"@jsoncpp_git//:jsoncpp",
"@local_tsl//tsl/platform:base64",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:status",
],
)
@ -347,8 +347,8 @@ cc_library(
],
copts = tsl_copts(),
deps = [
"//tsl/platform:errors",
"//tsl/platform:status",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:status",
],
)
@ -359,10 +359,10 @@ tsl_cc_test(
deps = [
":expiring_lru_cache",
":now_seconds_env",
"//tsl/platform:env_impl",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_xla//xla/tsl/lib/core:status_test_util",
"//xla/tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -373,13 +373,13 @@ tsl_cc_test(
deps = [
":now_seconds_env",
":ram_file_block_cache",
"//tsl/platform:blocking_counter",
"//tsl/platform:env",
"//tsl/platform:env_impl",
"//tsl/platform:notification",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_xla//xla/tsl/lib/core:status_test_util",
"//xla/tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform:blocking_counter",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:notification",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -390,15 +390,15 @@ tsl_cc_test(
deps = [
":gcs_file_system",
":http_request_fake",
"//tsl/platform:env_impl",
"//tsl/platform:errors",
"//tsl/platform:str_util",
"//tsl/platform:strcat",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_xla//xla/tsl/lib/core:status_test_util",
"@local_xla//xla/tsl/profiler/backends/cpu:traceme_recorder_impl",
"@local_xla//xla/tsl/profiler/utils:time_utils_impl",
"//xla/tsl/lib/core:status_test_util",
"//xla/tsl/profiler/backends/cpu:traceme_recorder_impl",
"//xla/tsl/profiler/utils:time_utils_impl",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:str_util",
"@local_tsl//tsl/platform:strcat",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -409,10 +409,10 @@ tsl_cc_test(
linkopts = if_windows(["-DEFAULTLIB:ws2_32.lib"]),
deps = [
":gcs_dns_cache",
"//tsl/platform:env_impl",
"//tsl/platform:str_util",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:str_util",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -423,11 +423,11 @@ tsl_cc_test(
linkopts = if_windows(["-DEFAULTLIB:ws2_32.lib"]),
deps = [
":gcs_throttle",
"//tsl/platform:env_impl",
"//tsl/platform:str_util",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_xla//xla/tsl/lib/core:status_test_util",
"//xla/tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:str_util",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -437,15 +437,15 @@ tsl_cc_test(
srcs = ["curl_http_request_test.cc"],
deps = [
":curl_http_request",
"//tsl/platform",
"//tsl/platform:env_impl",
"//tsl/platform:path",
"//tsl/platform:platform_port",
"//tsl/platform:test",
"//tsl/platform:test_main",
"//xla/tsl/lib/core:status_test_util",
"@com_google_absl//absl/status",
"@com_google_absl//absl/strings",
"@local_xla//xla/tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:path",
"@local_tsl//tsl/platform:platform_port",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -454,22 +454,22 @@ tsl_cc_test(
size = "small",
srcs = ["oauth_client_test.cc"],
data = [
"//tsl/platform/cloud/testdata:service_account_credentials",
"//tsl/platform/cloud/testdata:service_account_public_key",
"//xla/tsl/platform/cloud/testdata:service_account_credentials",
"//xla/tsl/platform/cloud/testdata:service_account_public_key",
],
tags = ["no_oss"], # TODO(b/327036247): revisit after this moves to XLA
deps = [
":http_request_fake",
":oauth_client",
"//tsl/platform:base64",
"//tsl/platform:env",
"//tsl/platform:env_impl",
"//tsl/platform:path",
"//tsl/platform:scanner",
"//tsl/platform:test",
"//tsl/platform:test_main",
"//xla/tsl/lib/core:status_test_util",
"@boringssl//:crypto",
"@local_xla//xla/tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform:base64",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:path",
"@local_tsl//tsl/platform:scanner",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -478,19 +478,19 @@ tsl_cc_test(
size = "small",
srcs = ["google_auth_provider_test.cc"],
data = [
"//tsl/platform/cloud/testdata:application_default_credentials",
"//tsl/platform/cloud/testdata:service_account_credentials",
"//xla/tsl/platform/cloud/testdata:application_default_credentials",
"//xla/tsl/platform/cloud/testdata:service_account_credentials",
],
tags = ["no_oss"], # TODO(b/327036247): revisit after this moves to XLA
deps = [
":google_auth_provider",
":http_request_fake",
":oauth_client",
"//tsl/platform:env_impl",
"//tsl/platform:path",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_xla//xla/tsl/lib/core:status_test_util",
"//xla/tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:path",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -501,10 +501,10 @@ tsl_cc_test(
deps = [
":compute_engine_metadata_client",
":http_request_fake",
"//tsl/platform:env",
"//tsl/platform:env_impl",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -515,9 +515,9 @@ tsl_cc_test(
deps = [
":compute_engine_zone_provider",
":http_request_fake",
"//tsl/platform:env_impl",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_tsl//tsl/platform:env_impl",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)
@ -527,8 +527,8 @@ tsl_cc_test(
srcs = ["time_util_test.cc"],
deps = [
":time_util",
"//tsl/platform:test",
"//tsl/platform:test_main",
"@local_xla//xla/tsl/lib/core:status_test_util",
"//xla/tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform:test",
"@local_tsl//tsl/platform:test_main",
],
)

6
third_party/xla/third_party/tsl/tsl/platform/cloud/auth_provider.h → third_party/xla/xla/tsl/platform/cloud/auth_provider.h vendored
View File

@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#ifndef TENSORFLOW_TSL_PLATFORM_CLOUD_AUTH_PROVIDER_H_
#define TENSORFLOW_TSL_PLATFORM_CLOUD_AUTH_PROVIDER_H_
#ifndef XLA_TSL_PLATFORM_CLOUD_AUTH_PROVIDER_H_
#define XLA_TSL_PLATFORM_CLOUD_AUTH_PROVIDER_H_
#include <string>
@ -52,4 +52,4 @@ class EmptyAuthProvider : public AuthProvider {
} // namespace tsl
#endif // TENSORFLOW_TSL_PLATFORM_CLOUD_AUTH_PROVIDER_H_
#endif // XLA_TSL_PLATFORM_CLOUD_AUTH_PROVIDER_H_

4
third_party/xla/third_party/tsl/tsl/platform/cloud/compute_engine_metadata_client.cc → third_party/xla/xla/tsl/platform/cloud/compute_engine_metadata_client.cc vendored
View File

@ -13,13 +13,13 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include "tsl/platform/cloud/compute_engine_metadata_client.h"
#include "xla/tsl/platform/cloud/compute_engine_metadata_client.h"
#include <cstdlib>
#include <utility>
#include "absl/strings/str_cat.h"
#include "tsl/platform/cloud/curl_http_request.h"
#include "xla/tsl/platform/cloud/curl_http_request.h"
namespace tsl {

8
third_party/xla/third_party/tsl/tsl/platform/cloud/compute_engine_metadata_client.h → third_party/xla/xla/tsl/platform/cloud/compute_engine_metadata_client.h vendored
View File

@ -13,10 +13,10 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#ifndef TENSORFLOW_TSL_PLATFORM_CLOUD_COMPUTE_ENGINE_METADATA_CLIENT_H_
#define TENSORFLOW_TSL_PLATFORM_CLOUD_COMPUTE_ENGINE_METADATA_CLIENT_H_
#ifndef XLA_TSL_PLATFORM_CLOUD_COMPUTE_ENGINE_METADATA_CLIENT_H_
#define XLA_TSL_PLATFORM_CLOUD_COMPUTE_ENGINE_METADATA_CLIENT_H_
#include "tsl/platform/cloud/http_request.h"
#include "xla/tsl/platform/cloud/http_request.h"
#include "tsl/platform/retrying_utils.h"
#include "tsl/platform/status.h"
@ -64,4 +64,4 @@ class ComputeEngineMetadataClient {
} // namespace tsl
#endif // TENSORFLOW_TSL_PLATFORM_CLOUD_COMPUTE_ENGINE_METADATA_CLIENT_H_
#endif // XLA_TSL_PLATFORM_CLOUD_COMPUTE_ENGINE_METADATA_CLIENT_H_

4
third_party/xla/third_party/tsl/tsl/platform/cloud/compute_engine_metadata_client_test.cc → third_party/xla/xla/tsl/platform/cloud/compute_engine_metadata_client_test.cc vendored
View File

@ -13,9 +13,9 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include "tsl/platform/cloud/compute_engine_metadata_client.h"
#include "xla/tsl/platform/cloud/compute_engine_metadata_client.h"
#include "tsl/platform/cloud/http_request_fake.h"
#include "xla/tsl/platform/cloud/http_request_fake.h"
#include "tsl/platform/env.h"
#include "tsl/platform/test.h"

Some files were not shown because too many files have changed in this diff Show More