diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 01f48cad192..16f6b860308 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -1,6 +1,7 @@ # Description: # C API for TensorFlow, for use by client language bindings. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", @@ -9,6 +10,11 @@ load( "tf_custom_op_library", "tf_kernel_library", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") package( diff --git a/tensorflow/c/eager/BUILD b/tensorflow/c/eager/BUILD index da18424f936..f28faba87c0 100644 --- a/tensorflow/c/eager/BUILD +++ b/tensorflow/c/eager/BUILD @@ -1,12 +1,20 @@ # Experimental extensions to the C API for eager execution of kernels. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", + "if_tpu", "tf_cc_test", "tf_copts", "tf_cuda_cc_test", "tf_cuda_library", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "internal_tfrt_deps") load( "//tensorflow/core/platform:build_config.bzl", "tf_kernel_tests_linkstatic", @@ -87,7 +95,7 @@ tf_cuda_library( "//tensorflow/core/distributed_runtime:server_lib", "//tensorflow/core/distributed_runtime:worker_env", "//tensorflow/core:gpu_runtime", - ], + ] + internal_tfrt_deps(), alwayslink = 1, ) @@ -248,22 +256,24 @@ cc_library( ":c_api_unified_internal", ":gradients_internal", ":tape", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", "//tensorflow/c:c_api", "//tensorflow/c:tf_status_helper", "//tensorflow/c/experimental/ops:array_ops", "//tensorflow/c/experimental/ops:math_ops", "//tensorflow/c/experimental/ops:nn_ops", "//tensorflow/cc/profiler", - "//tensorflow/compiler/mlir/tensorflow/c:mlir_c_api_registration", "//tensorflow/core:framework", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core/lib/llvm_rtti", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:flat_hash_set", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/types:span", - ], + ] + if_tpu( + if_false = ["//tensorflow/compiler/mlir/tensorflow/c:mlir_c_api_registration"], + if_true = [], + ), ) cc_library( @@ -311,6 +321,8 @@ cc_library( ":c_api_unified_internal", ":gradients_internal", ":gradients_util", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", "//tensorflow/c:c_api", "//tensorflow/c:tf_status_helper", "//tensorflow/c/experimental/gradients:math_grad", @@ -319,13 +331,13 @@ cc_library( "//tensorflow/c/experimental/ops:math_ops", "//tensorflow/c/experimental/ops:nn_ops", "//tensorflow/cc/profiler", - "//tensorflow/compiler/mlir/tensorflow/c:mlir_c_api_registration", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core/lib/llvm_rtti", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/types:span", - ], + ] + if_tpu( + if_false = ["//tensorflow/compiler/mlir/tensorflow/c:mlir_c_api_registration"], + if_true = [], + ), ) tf_cuda_cc_test( diff --git a/tensorflow/c/eager/c_api.cc b/tensorflow/c/eager/c_api.cc index fb5ce22ae5f..a91f09ab071 100644 --- a/tensorflow/c/eager/c_api.cc +++ b/tensorflow/c/eager/c_api.cc @@ -39,7 +39,7 @@ limitations under the License. #include "tensorflow/c/eager/tfe_op_internal.h" #include "tensorflow/c/eager/tfe_tensorhandle_internal.h" #include "tensorflow/c/tf_tensor_internal.h" -#ifdef PLATFORM_GOOGLE +#if defined(PLATFORM_GOOGLE) && !defined(LIBTFTPU) #include "tensorflow/core/tfrt/eager/c_api_tfrt.h" #endif #include "tensorflow/core/common_runtime/device.h" @@ -729,7 +729,7 @@ void TFE_DeleteContextOptions(TFE_ContextOptions* options) { delete options; } TFE_Context* TFE_NewContext(const TFE_ContextOptions* opts, TF_Status* status) { if (opts->use_tfrt) { -#ifdef PLATFORM_GOOGLE +#if defined(PLATFORM_GOOGLE) && !defined(LIBTFTPU) return tensorflow::wrap(new tfrt::tf::ContextInterface(opts->async)); #else status->status = tensorflow::errors::Unimplemented("TFRT is not supported"); diff --git a/tensorflow/c/eager/parallel_device/BUILD b/tensorflow/c/eager/parallel_device/BUILD index 3eec95294b3..473ab503834 100644 --- a/tensorflow/c/eager/parallel_device/BUILD +++ b/tensorflow/c/eager/parallel_device/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", diff --git a/tensorflow/c/experimental/filesystem/BUILD b/tensorflow/c/experimental/filesystem/BUILD index 061fdbd893b..c05c7dc3f7e 100644 --- a/tensorflow/c/experimental/filesystem/BUILD +++ b/tensorflow/c/experimental/filesystem/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Experimental filesystem C APIs for TensorFlow. # Will be moved in proper place once all filesystems are converted to the # modular framework. diff --git a/tensorflow/c/experimental/filesystem/plugins/gcs/BUILD b/tensorflow/c/experimental/filesystem/plugins/gcs/BUILD index 54217db1de0..0fc9f260b21 100644 --- a/tensorflow/c/experimental/filesystem/plugins/gcs/BUILD +++ b/tensorflow/c/experimental/filesystem/plugins/gcs/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Experimental gcs filesystem plugin. load("//tensorflow:tensorflow.bzl", "get_win_copts", "tf_cc_shared_object", "tf_cc_test") diff --git a/tensorflow/c/experimental/filesystem/plugins/hadoop/BUILD b/tensorflow/c/experimental/filesystem/plugins/hadoop/BUILD index bb97587d6d1..cd697906121 100644 --- a/tensorflow/c/experimental/filesystem/plugins/hadoop/BUILD +++ b/tensorflow/c/experimental/filesystem/plugins/hadoop/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Experimental hadoop filesystem plugin. load("//tensorflow:tensorflow.bzl", "get_win_copts", "tf_cc_shared_object", "tf_cc_test") @@ -20,6 +22,7 @@ cc_library( name = "hadoop_filesystem_impl", srcs = ["hadoop_filesystem.cc"], hdrs = ["hadoop_filesystem.h"], + compatible_with = [], copts = select({ "//conditions:default": [], "//tensorflow:windows": get_win_copts(), diff --git a/tensorflow/c/experimental/filesystem/plugins/posix/BUILD b/tensorflow/c/experimental/filesystem/plugins/posix/BUILD index 3afe114b5a6..b87dddb96a1 100644 --- a/tensorflow/c/experimental/filesystem/plugins/posix/BUILD +++ b/tensorflow/c/experimental/filesystem/plugins/posix/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Experimental posix filesystem plugin. load("//tensorflow:tensorflow.bzl", "tf_cc_shared_object") diff --git a/tensorflow/c/experimental/filesystem/plugins/windows/BUILD b/tensorflow/c/experimental/filesystem/plugins/windows/BUILD index b845d1e3616..02f7b5ba706 100644 --- a/tensorflow/c/experimental/filesystem/plugins/windows/BUILD +++ b/tensorflow/c/experimental/filesystem/plugins/windows/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Experimental windows filesystem plugin. load("//tensorflow:tensorflow.bzl", "get_win_copts", "tf_cc_shared_object") diff --git a/tensorflow/c/experimental/gradients/BUILD b/tensorflow/c/experimental/gradients/BUILD index 5386c0cf3f7..e8a50e32216 100644 --- a/tensorflow/c/experimental/gradients/BUILD +++ b/tensorflow/c/experimental/gradients/BUILD @@ -1,3 +1,6 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Library of gradient functions. package( licenses = ["notice"], # Apache 2.0 @@ -76,6 +79,7 @@ cc_library( ":array_grad", ":math_grad", ":nn_grad", + "//tensorflow/c/eager:gradients_internal", ], ) diff --git a/tensorflow/c/experimental/ops/BUILD b/tensorflow/c/experimental/ops/BUILD index d2c22e65f80..bdde4e2cf27 100644 --- a/tensorflow/c/experimental/ops/BUILD +++ b/tensorflow/c/experimental/ops/BUILD @@ -1,3 +1,6 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Experimental ops. These will eventually be replaced by machine-generated versions. package( licenses = ["notice"], # Apache 2.0 diff --git a/tensorflow/c/experimental/saved_model/core/BUILD b/tensorflow/c/experimental/saved_model/core/BUILD index 4127fdfe0ee..486846e90c9 100644 --- a/tensorflow/c/experimental/saved_model/core/BUILD +++ b/tensorflow/c/experimental/saved_model/core/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Experimental SavedModel C APIs for TensorFlow. See RFC # https://github.com/tensorflow/community/pull/207 # Targets in this directory are pure C++ "Classes" underlying the C API types diff --git a/tensorflow/c/experimental/saved_model/core/ops/BUILD b/tensorflow/c/experimental/saved_model/core/ops/BUILD index 673ea1a80e2..549980b03e9 100644 --- a/tensorflow/c/experimental/saved_model/core/ops/BUILD +++ b/tensorflow/c/experimental/saved_model/core/ops/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # This package contains written convenience helpers for Eager Operations # used by SavedModel. Once we autogenerate C++ Eager Op wrappers, we can remove these. load( diff --git a/tensorflow/c/experimental/saved_model/core/revived_types/BUILD b/tensorflow/c/experimental/saved_model/core/revived_types/BUILD index 93a485e717f..1205f12b948 100644 --- a/tensorflow/c/experimental/saved_model/core/revived_types/BUILD +++ b/tensorflow/c/experimental/saved_model/core/revived_types/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # This package contains classes corresponding to Revived SavedObjectGraph types # used by SavedModel. See https://cs.opensource.google/tensorflow/tensorflow/+/c575e2ba93c442121d98d3f125d83fed1339924d:tensorflow/core/protobuf/saved_object_graph.proto;l=56-62 package( diff --git a/tensorflow/c/experimental/saved_model/internal/BUILD b/tensorflow/c/experimental/saved_model/internal/BUILD index 5d6d23b403f..091f5c86967 100644 --- a/tensorflow/c/experimental/saved_model/internal/BUILD +++ b/tensorflow/c/experimental/saved_model/internal/BUILD @@ -9,6 +9,8 @@ # Note(bmzhao): The *.cc files in this directory form the direct implementation of the # C API functions exposed in tf/c/experimental/saved_model/public/. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Note(bmzhao): All *type.h files in this directory are the internal definitions of # the opaque C types. These headers should only be visible to internal tensorflow # implementors. diff --git a/tensorflow/c/experimental/saved_model/public/BUILD b/tensorflow/c/experimental/saved_model/public/BUILD index d29585ae1ba..33c3ce91aef 100644 --- a/tensorflow/c/experimental/saved_model/public/BUILD +++ b/tensorflow/c/experimental/saved_model/public/BUILD @@ -8,6 +8,8 @@ # programmatic checks that all "public" headers only include other "public" # headers. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( # This is intentionally public default_visibility = [ diff --git a/tensorflow/c/experimental/stream_executor/BUILD b/tensorflow/c/experimental/stream_executor/BUILD index 7daa311d461..95bb12e8e50 100644 --- a/tensorflow/c/experimental/stream_executor/BUILD +++ b/tensorflow/c/experimental/stream_executor/BUILD @@ -1,6 +1,7 @@ # Description: # StreamExecutor C API. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", diff --git a/tensorflow/c/kernels/BUILD b/tensorflow/c/kernels/BUILD index 6bb2b347a30..d89eda3eb4e 100644 --- a/tensorflow/c/kernels/BUILD +++ b/tensorflow/c/kernels/BUILD @@ -1,9 +1,13 @@ -load( - "//tensorflow:tensorflow.bzl", - "tf_cc_test", - "tf_gen_op_libs", - "tf_kernel_library", -) +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_gen_op_libs") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") +load("//tensorflow:tensorflow.bzl", "tf_cc_test") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") package( default_visibility = ["//visibility:public"], diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index 8602bfafff8..bf5ada89cdd 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -2,16 +2,22 @@ # TensorFlow is a computational framework, primarily for use in machine # learning applications. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "cc_library_with_android_deps", "tf_cc_binary", "tf_cc_test", "tf_copts", - "tf_gen_op_wrappers_cc", "transitive_hdrs", ) +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrappers_cc") + package( default_visibility = ["//visibility:public"], licenses = ["notice"], # Apache 2.0 diff --git a/tensorflow/cc/experimental/base/public/BUILD b/tensorflow/cc/experimental/base/public/BUILD index 045d4e6cd97..0aaf2238e6a 100644 --- a/tensorflow/cc/experimental/base/public/BUILD +++ b/tensorflow/cc/experimental/base/public/BUILD @@ -8,6 +8,8 @@ # 2. Are std:: types # 3. Wrap an opaque C type +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( # This is intentionally public default_visibility = [ diff --git a/tensorflow/cc/experimental/base/tests/BUILD b/tensorflow/cc/experimental/base/tests/BUILD index f449d618f72..f7f6e77c98f 100644 --- a/tensorflow/cc/experimental/base/tests/BUILD +++ b/tensorflow/cc/experimental/base/tests/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + # Tests for the C++ header-only base types. load("//tensorflow:tensorflow.bzl", "tf_cc_test") diff --git a/tensorflow/cc/profiler/BUILD b/tensorflow/cc/profiler/BUILD index 057ce7cb993..43240506f8c 100644 --- a/tensorflow/cc/profiler/BUILD +++ b/tensorflow/cc/profiler/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") package( diff --git a/tensorflow/cc/saved_model/BUILD b/tensorflow/cc/saved_model/BUILD index d25b580dace..4efe56ff822 100644 --- a/tensorflow/cc/saved_model/BUILD +++ b/tensorflow/cc/saved_model/BUILD @@ -1,6 +1,8 @@ # Description: # TensorFlow SavedModel. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "if_android", diff --git a/tensorflow/cc/saved_model/experimental/public/BUILD b/tensorflow/cc/saved_model/experimental/public/BUILD index 9640848ebf5..a0f8204c937 100644 --- a/tensorflow/cc/saved_model/experimental/public/BUILD +++ b/tensorflow/cc/saved_model/experimental/public/BUILD @@ -1,6 +1,8 @@ # Experimental C++ SavedModel Header Only APIs. See RFC # https://github.com/tensorflow/community/pull/207 +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( # This is intentionally public default_visibility = [ diff --git a/tensorflow/cc/tools/BUILD b/tensorflow/cc/tools/BUILD index a192c4bdb18..73d84483199 100644 --- a/tensorflow/cc/tools/BUILD +++ b/tensorflow/cc/tools/BUILD @@ -1,6 +1,7 @@ # Description: # TensorFlow cc tools. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", diff --git a/tensorflow/compiler/aot/BUILD b/tensorflow/compiler/aot/BUILD index ff255dd9cc1..72bb8ee62ba 100644 --- a/tensorflow/compiler/aot/BUILD +++ b/tensorflow/compiler/aot/BUILD @@ -1,4 +1,5 @@ load("//tensorflow/compiler/aot:tfcompile.bzl", "tf_library") +load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test") load("//tensorflow/core/platform:build_config.bzl", "if_llvm_aarch64_available", "if_llvm_system_z_available") @@ -303,6 +304,7 @@ cc_library( name = "aot_only_var_handle_op", srcs = ["aot_only_var_handle_op.cc"], hdrs = ["aot_only_var_handle_op.h"], + compatible_with = get_compatible_with_cloud(), visibility = [ "//tensorflow/compiler/tf2xla:__pkg__", ], diff --git a/tensorflow/compiler/jit/BUILD b/tensorflow/compiler/jit/BUILD index 82375577610..ae1e4ab7067 100644 --- a/tensorflow/compiler/jit/BUILD +++ b/tensorflow/compiler/jit/BUILD @@ -1,5 +1,16 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "cc_header_only_library", "if_mlir", "tf_cc_test") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "if_tpu", "tf_copts") load("//tensorflow/stream_executor:build_defs.bzl", "if_cuda_or_rocm") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library", "tf_jit_compilation_passes_extra_deps") load("//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library") load("//tensorflow/core/platform:build_config_root.bzl", "tf_cuda_tests_tags") @@ -66,8 +77,10 @@ cc_library( "//tensorflow/compiler/jit/kernels:xla_ops", "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops", "//tensorflow/compiler/tf2xla/kernels:xla_ops", - "//tensorflow/compiler/xla/service:cpu_plugin", - ], + ] + if_tpu( + if_false = ["//tensorflow/compiler/xla/service:cpu_plugin"], + if_true = [], + ), alwayslink = 1, ) @@ -94,15 +107,19 @@ cc_library( ":jit_compilation_passes", ":xla_device", ":xla_kernel_creator", # buildcleaner: keep + "@com_google_absl//absl/memory", "//tensorflow/compiler/jit/kernels:xla_ops", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla/kernels:xla_ops", - "//tensorflow/compiler/xla/service:cpu_plugin", # buildcleaner: keep "//tensorflow/core:core_cpu_internal", "//tensorflow/core:lib", - "@com_google_absl//absl/memory", - ], + ] + if_tpu( + if_false = [ + "//tensorflow/compiler/xla/service:cpu_plugin", # buildcleaner: keep + ], + if_true = [], + ), alwayslink = 1, ) @@ -115,17 +132,21 @@ cc_library( ":jit_compilation_passes", ":xla_device", ":xla_kernel_creator", # buildcleaner: keep + "@com_google_absl//absl/memory", + "@com_google_absl//absl/strings", "//tensorflow/compiler/jit/kernels:xla_ops", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_op_registry", "//tensorflow/compiler/tf2xla/kernels:xla_ops", - "//tensorflow/compiler/xla/service:gpu_plugin", # buildcleaner: keep "//tensorflow/core:core_cpu_internal", "//tensorflow/core:lib", "//tensorflow/core/common_runtime/gpu:gpu_init", - "@com_google_absl//absl/memory", - "@com_google_absl//absl/strings", - ], + ] + if_tpu( + if_false = [ + "//tensorflow/compiler/xla/service:gpu_plugin", # buildcleaner: keep + ], + if_true = [], + ), alwayslink = 1, ) @@ -329,12 +350,17 @@ cc_library( name = "xla_compilation_cache", srcs = ["xla_compilation_cache.cc"], hdrs = ["xla_compilation_cache.h"], + copts = tf_copts(), deps = [ ":flags", ":xla_activity_listener", ":xla_activity_proto_cc", - "//tensorflow/compiler/mlir:array_container_utils", - "//tensorflow/compiler/mlir/tensorflow:compile_mlir_util_no_tf_dialect_passes", + "@com_google_absl//absl/base", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:optional", + "@com_google_absl//absl/types:span", "//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:xla_compiler", "//tensorflow/compiler/tf2xla:xla_context", @@ -349,13 +375,13 @@ cc_library( "//tensorflow/core:lib_internal", "//tensorflow/core:protos_all_cc", "//tensorflow/core/platform:logging", - "@com_google_absl//absl/base", - "@com_google_absl//absl/container:flat_hash_map", - "@com_google_absl//absl/container:inlined_vector", - "@com_google_absl//absl/strings", - "@com_google_absl//absl/types:optional", - "@com_google_absl//absl/types:span", - ], + ] + if_tpu( + if_false = [ + "//tensorflow/compiler/mlir:array_container_utils", + "//tensorflow/compiler/mlir/tensorflow:compile_mlir_util_no_tf_dialect_passes", + ], + if_true = [], + ), ) tf_cc_test( diff --git a/tensorflow/compiler/jit/graphcycles/BUILD b/tensorflow/compiler/jit/graphcycles/BUILD index 61d0c0de35f..23d994c27c5 100644 --- a/tensorflow/compiler/jit/graphcycles/BUILD +++ b/tensorflow/compiler/jit/graphcycles/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/compiler/jit/kernels/BUILD b/tensorflow/compiler/jit/kernels/BUILD index eb9ad8a2e85..be9e1d31f3b 100644 --- a/tensorflow/compiler/jit/kernels/BUILD +++ b/tensorflow/compiler/jit/kernels/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( default_visibility = [ "//tensorflow/compiler/tf2xla:internal", diff --git a/tensorflow/compiler/jit/ops/BUILD b/tensorflow/compiler/jit/ops/BUILD index afc96a8e68c..6ca8fd0e34a 100644 --- a/tensorflow/compiler/jit/ops/BUILD +++ b/tensorflow/compiler/jit/ops/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py") package( diff --git a/tensorflow/compiler/jit/tests/BUILD b/tensorflow/compiler/jit/tests/BUILD index 412dfefb9b7..88ce43902fd 100644 --- a/tensorflow/compiler/jit/tests/BUILD +++ b/tensorflow/compiler/jit/tests/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") licenses(["notice"]) # Apache 2.0 diff --git a/tensorflow/compiler/jit/xla_compilation_cache.cc b/tensorflow/compiler/jit/xla_compilation_cache.cc index fb184d62e27..3ab3c19e439 100644 --- a/tensorflow/compiler/jit/xla_compilation_cache.cc +++ b/tensorflow/compiler/jit/xla_compilation_cache.cc @@ -23,8 +23,6 @@ limitations under the License. #include "tensorflow/compiler/jit/flags.h" #include "tensorflow/compiler/jit/xla_activity.pb.h" #include "tensorflow/compiler/jit/xla_activity_listener.h" -#include "tensorflow/compiler/mlir/tensorflow/utils/compile_mlir_util.h" -#include "tensorflow/compiler/mlir/utils/array_container_utils.h" #include "tensorflow/compiler/tf2xla/shape_util.h" #include "tensorflow/compiler/tf2xla/type_util.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" @@ -49,6 +47,11 @@ limitations under the License. #include "tensorflow/core/public/version.h" #include "tensorflow/core/util/dump_graph.h" +#if !defined(LIBTFTPU) +#include "tensorflow/compiler/mlir/tensorflow/utils/compile_mlir_util.h" +#include "tensorflow/compiler/mlir/utils/array_container_utils.h" +#endif + namespace tensorflow { constexpr int64 XlaCompilationCache::kDefaultCompilationThreshold; @@ -286,6 +289,13 @@ Status XlaCompilationCache::CompileSingleOp( }); const ConfigProto* config = ctx->function_library()->config_proto(); bool use_mlir = config && config->experimental().enable_mlir_bridge(); +#ifdef LIBTFTPU + if (use_mlir && has_tensor_list_arg) { + LOG(WARNING) << "MLIR is not supported in this environment."; + } + return compiler->CompileGraph(compile_options, node_def.name(), + std::move(graph), args, result); +#else // TODO(b/155596779): Support TensorList args. if (!use_mlir || !has_tensor_list_arg) { return compiler->CompileGraph(compile_options, node_def.name(), @@ -297,6 +307,7 @@ Status XlaCompilationCache::CompileSingleOp( *graph, mlir::SpanToArrayRef(args), options.device_type.type_string(), compile_options.use_tuple_arg, *options.flib_def, debug_info, options.shape_representation_fn, result); +#endif }; return CompileImpl(options, name, args, compile_op, /*compile_threshold=*/absl::nullopt, diff --git a/tensorflow/compiler/mlir/tensorflow/c/BUILD b/tensorflow/compiler/mlir/tensorflow/c/BUILD index 5c6f39699bf..1ed916f64d4 100644 --- a/tensorflow/compiler/mlir/tensorflow/c/BUILD +++ b/tensorflow/compiler/mlir/tensorflow/c/BUILD @@ -19,6 +19,7 @@ tf_cuda_library( srcs = [ "c_api_unified_experimental_mlir.cc", ], + compatible_with = [], copts = tf_copts(), deps = [ "//tensorflow/c:c_api", diff --git a/tensorflow/compiler/plugin/BUILD b/tensorflow/compiler/plugin/BUILD index c2ba5cb3ecd..dc1c2391e94 100644 --- a/tensorflow/compiler/plugin/BUILD +++ b/tensorflow/compiler/plugin/BUILD @@ -13,6 +13,8 @@ # limitations under the License. # ============================================================================== +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + """Configuration file for an XLA plugin. please don't check in changes to this file. to prevent changes appearing diff --git a/tensorflow/compiler/tf2tensorrt/BUILD b/tensorflow/compiler/tf2tensorrt/BUILD index 44fb5513886..695e7f08d29 100644 --- a/tensorflow/compiler/tf2tensorrt/BUILD +++ b/tensorflow/compiler/tf2tensorrt/BUILD @@ -234,6 +234,7 @@ tf_cuda_library( name = "trt_engine_utils", srcs = ["utils/trt_engine_utils.cc"], hdrs = ["utils/trt_engine_utils.h"], + compatible_with = [], deps = [ ":trt_logging", ":utils", @@ -248,6 +249,7 @@ tf_cuda_library( name = "trt_logging", srcs = ["utils/trt_logger.cc"], hdrs = ["utils/trt_logger.h"], + compatible_with = [], visibility = ["//visibility:public"], deps = [ ":common_utils", @@ -289,6 +291,7 @@ tf_cuda_library( "utils/trt_lru_cache.h", "utils/trt_shape_optimization_profiles.h", ], + compatible_with = [], deps = [ ":trt_allocator", ":trt_logging", @@ -306,6 +309,7 @@ tf_cuda_library( name = "trt_allocator", srcs = ["utils/trt_allocator.cc"], hdrs = ["utils/trt_allocator.h"], + compatible_with = [], deps = [ "//tensorflow/core:framework_headers_lib", "//tensorflow/core:framework_lite", @@ -365,6 +369,7 @@ tf_cuda_library( hdrs = [ "convert/logger_registry.h", ], + compatible_with = [], copts = tf_copts(), deps = [ "@com_google_absl//absl/strings", @@ -385,6 +390,7 @@ tf_cuda_library( "convert/convert_nodes.h", "convert/trt_optimization_pass.h", ], + compatible_with = [], deps = [ ":common_utils", ":logger_registry", @@ -543,6 +549,7 @@ tf_cuda_library( name = "trt_plugins", srcs = ["plugin/trt_plugin.cc"], hdrs = ["plugin/trt_plugin.h"], + compatible_with = [], deps = [ "//tensorflow/core:framework_lite", "//tensorflow/core:lib_proto_parsing", diff --git a/tensorflow/compiler/tf2xla/BUILD b/tensorflow/compiler/tf2xla/BUILD index 1a91f54afc9..a1deaa25ac5 100644 --- a/tensorflow/compiler/tf2xla/BUILD +++ b/tensorflow/compiler/tf2xla/BUILD @@ -1,4 +1,5 @@ -load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test", "tf_cuda_cc_test", "tf_openmp_copts") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") +load("//tensorflow:tensorflow.bzl", "if_tpu", "tf_cc_binary", "tf_cc_test", "tf_copts", "tf_cuda_cc_test", "tf_openmp_copts") load( "//tensorflow/core/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured", @@ -10,6 +11,11 @@ load( "tf_proto_library_cc", ) load("//tensorflow/compiler/xla:xla.bzl", "xla_py_proto_library") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_portable_proto_library") load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured") load("//tensorflow/compiler/xla/service/cpu:build_defs.bzl", "runtime_copts") @@ -165,6 +171,7 @@ cc_library( name = "mlir_tf2xla", srcs = ["mlir_tf2xla.cc"], hdrs = ["tf2xla.h"], + compatible_with = [], visibility = ["//visibility:public"], deps = [ ":tf2xla_proto_cc", @@ -258,6 +265,7 @@ cc_library( "@com_google_absl//absl/synchronization", "//third_party/eigen3", "//tensorflow/core/framework:numeric_types", + "//tensorflow/core/platform:bfloat16", ] + tf_additional_tensor_coding_deps(), alwayslink = 1, ) @@ -303,14 +311,18 @@ cc_library( "//tensorflow/compiler/xla/client:client_library", "//tensorflow/compiler/xla/client:local_client", "//tensorflow/compiler/xla/client:xla_computation", - "//tensorflow/compiler/xla/service:cpu_plugin", "//tensorflow/compiler/xla/service:platform_util", - "//tensorflow/compiler/xla/service/cpu:buffer_info_util", - "//tensorflow/compiler/xla/service/cpu:cpu_executable", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/stream_executor:platform", - ], + ] + if_tpu( + if_false = [ + "//tensorflow/compiler/xla/service:cpu_plugin", + "//tensorflow/compiler/xla/service/cpu:buffer_info_util", + "//tensorflow/compiler/xla/service/cpu:cpu_executable", + ], + if_true = [], + ), ) cc_library( @@ -334,6 +346,7 @@ cc_library( "xla_op_kernel.h", "xla_op_registry.h", ], + copts = tf_copts(), visibility = [":friends"], deps = [ ":common", @@ -349,11 +362,13 @@ cc_library( ":xla_helpers", ":xla_op_registry", ":xla_resource", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/memory", + "@com_google_absl//absl/types:span", + "@com_google_absl//absl/types:variant", "//tensorflow/compiler/jit:common", "//tensorflow/compiler/jit:flags", "//tensorflow/compiler/jit:shape_inference", - "//tensorflow/compiler/mlir:array_container_utils", - "//tensorflow/compiler/mlir/tensorflow:compile_mlir_util_no_tf_dialect_passes", "//tensorflow/compiler/xla:protobuf_util", "//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:status_macros", @@ -371,11 +386,13 @@ cc_library( "//tensorflow/core:lib_internal", "//tensorflow/core:ops", "//tensorflow/core:protos_all_cc", - "@com_google_absl//absl/algorithm:container", - "@com_google_absl//absl/memory", - "@com_google_absl//absl/types:span", - "@com_google_absl//absl/types:variant", - ], + ] + if_tpu( + if_false = [ + "//tensorflow/compiler/mlir:array_container_utils", + "//tensorflow/compiler/mlir/tensorflow:compile_mlir_util_no_tf_dialect_passes", + ], + if_true = [], + ), alwayslink = 1, ) @@ -864,6 +881,7 @@ cc_library( name = "mlir_bridge_pass", srcs = ["mlir_bridge_pass.cc"], hdrs = ["mlir_bridge_pass.h"], + compatible_with = [], deps = [ "//tensorflow/compiler/jit:flags", "//tensorflow/compiler/mlir:mlir_graph_optimization_pass", @@ -877,14 +895,20 @@ cc_library( cc_library( name = "mlir_bridge_pass_registration", - srcs = [ - "mlir_bridge_pass_registration.cc", - ], - deps = [ - ":mlir_bridge_pass", - "//tensorflow/compiler/mlir:mlir_graph_optimization_pass_registration", - "//tensorflow/core:core_cpu", - ], + srcs = if_tpu( + if_false = [ + "mlir_bridge_pass_registration.cc", + ], + if_true = [], + ), + deps = if_tpu( + if_false = [ + ":mlir_bridge_pass", + "//tensorflow/compiler/mlir:mlir_graph_optimization_pass_registration", + "//tensorflow/core:core_cpu", + ], + if_true = [], + ), alwayslink = 1, ) diff --git a/tensorflow/compiler/tf2xla/cc/BUILD b/tensorflow/compiler/tf2xla/cc/BUILD index 973aafe1ad8..45a099baabe 100644 --- a/tensorflow/compiler/tf2xla/cc/BUILD +++ b/tensorflow/compiler/tf2xla/cc/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_cc") package( diff --git a/tensorflow/compiler/tf2xla/kernels/BUILD b/tensorflow/compiler/tf2xla/kernels/BUILD index 0edd918a92d..7e1878682f2 100644 --- a/tensorflow/compiler/tf2xla/kernels/BUILD +++ b/tensorflow/compiler/tf2xla/kernels/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_kernel_library") package( diff --git a/tensorflow/compiler/tf2xla/lib/BUILD b/tensorflow/compiler/tf2xla/lib/BUILD index 531679d3905..703f6c2eb72 100644 --- a/tensorflow/compiler/tf2xla/lib/BUILD +++ b/tensorflow/compiler/tf2xla/lib/BUILD @@ -1,5 +1,8 @@ # Utilities for building XLA computations. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( default_visibility = ["//tensorflow/compiler/tf2xla:friends"], licenses = ["notice"], # Apache 2.0 diff --git a/tensorflow/compiler/tf2xla/ops/BUILD b/tensorflow/compiler/tf2xla/ops/BUILD index b116a09dd02..50ff1ea5d16 100644 --- a/tensorflow/compiler/tf2xla/ops/BUILD +++ b/tensorflow/compiler/tf2xla/ops/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_custom_op_library", diff --git a/tensorflow/compiler/tf2xla/xla_compiler.cc b/tensorflow/compiler/tf2xla/xla_compiler.cc index b22dc05eaa1..e67e183aef9 100644 --- a/tensorflow/compiler/tf2xla/xla_compiler.cc +++ b/tensorflow/compiler/tf2xla/xla_compiler.cc @@ -23,8 +23,6 @@ limitations under the License. #include "tensorflow/compiler/jit/defs.h" #include "tensorflow/compiler/jit/flags.h" #include "tensorflow/compiler/jit/shape_inference.h" -#include "tensorflow/compiler/mlir/tensorflow/utils/compile_mlir_util.h" -#include "tensorflow/compiler/mlir/utils/array_container_utils.h" #include "tensorflow/compiler/tf2xla/graph_compiler.h" #include "tensorflow/compiler/tf2xla/rearrange_function_argument.h" #include "tensorflow/compiler/tf2xla/shape_util.h" @@ -58,6 +56,11 @@ limitations under the License. #include "tensorflow/core/protobuf/graph_debug_info.pb.h" #include "tensorflow/core/util/dump_graph.h" +#ifndef LIBTFTPU +#include "tensorflow/compiler/mlir/tensorflow/utils/compile_mlir_util.h" +#include "tensorflow/compiler/mlir/utils/array_container_utils.h" +#endif + namespace tensorflow { namespace { @@ -730,6 +733,13 @@ Status XlaCompiler::CompileFunction( } VLOG(1) << "===================================================="; +#ifdef LIBTFTPU + if (GetMlirCommonFlags()->tf_mlir_enable_mlir_bridge) { + VLOG(1) << "MLIR is not supported in this environment."; + } + TF_RETURN_IF_ERROR( + CompileGraph(options, function_id, std::move(graph), args, result)); +#else if (GetMlirCommonFlags()->tf_mlir_enable_mlir_bridge) { VLOG(1) << "Using MLIR bridge"; GraphDebugInfo debug_info; @@ -742,6 +752,7 @@ Status XlaCompiler::CompileFunction( TF_RETURN_IF_ERROR( CompileGraph(options, function_id, std::move(graph), args, result)); } +#endif VLOG(1) << "===================================================="; cache_[{function_id, arg_vector}] = *result; diff --git a/tensorflow/compiler/xla/BUILD b/tensorflow/compiler/xla/BUILD index 598112e00df..d4ab6f6bb1d 100644 --- a/tensorflow/compiler/xla/BUILD +++ b/tensorflow/compiler/xla/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "cc_header_only_library", "tf_cc_test") load( "//tensorflow/core/platform:build_config.bzl", diff --git a/tensorflow/compiler/xla/client/BUILD b/tensorflow/compiler/xla/client/BUILD index a51970bb168..4172de1f6e0 100644 --- a/tensorflow/compiler/xla/client/BUILD +++ b/tensorflow/compiler/xla/client/BUILD @@ -1,6 +1,8 @@ # Description: # XLA client libraries. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/compiler/xla/client/lib/BUILD b/tensorflow/compiler/xla/client/lib/BUILD index 6cd77bf9f19..92d222f32b2 100644 --- a/tensorflow/compiler/xla/client/lib/BUILD +++ b/tensorflow/compiler/xla/client/lib/BUILD @@ -1,5 +1,7 @@ # Common computation builders for XLA. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/compiler/xla/tests:build_defs.bzl", "generate_backend_suites", "xla_test") package( diff --git a/tensorflow/compiler/xla/pjrt/BUILD b/tensorflow/compiler/xla/pjrt/BUILD index 54240587282..50fd6a12d66 100644 --- a/tensorflow/compiler/xla/pjrt/BUILD +++ b/tensorflow/compiler/xla/pjrt/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda") diff --git a/tensorflow/compiler/xla/pjrt/distributed/BUILD b/tensorflow/compiler/xla/pjrt/distributed/BUILD index 175b4268dda..b80f9d85e2c 100644 --- a/tensorflow/compiler/xla/pjrt/distributed/BUILD +++ b/tensorflow/compiler/xla/pjrt/distributed/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/core/platform:build_config.bzl", "tf_proto_library_cc") load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") load("//tensorflow:tensorflow.bzl", "tf_cc_test") diff --git a/tensorflow/compiler/xla/python/BUILD b/tensorflow/compiler/xla/python/BUILD index 6ad1d789d48..fb9c705fdf3 100644 --- a/tensorflow/compiler/xla/python/BUILD +++ b/tensorflow/compiler/xla/python/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/core/platform:build_config.bzl", "pyx_library") load("//tensorflow/compiler/xla:xla.bzl", "xla_py_test_deps") load("//tensorflow:tensorflow.bzl", "tf_cc_test") diff --git a/tensorflow/compiler/xla/python/tpu_driver/BUILD b/tensorflow/compiler/xla/python/tpu_driver/BUILD index 70aeb3f2a86..5e560deee03 100644 --- a/tensorflow/compiler/xla/python/tpu_driver/BUILD +++ b/tensorflow/compiler/xla/python/tpu_driver/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/core/platform:build_config.bzl", "tf_proto_library_cc") load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") load( @@ -77,6 +78,7 @@ cc_library( cc_library( name = "direct_tpu_driver", srcs = ["direct_tpu_driver.cc"], + compatible_with = [], deps = [ ":tpu_driver", "@com_google_absl//absl/strings:str_format", diff --git a/tensorflow/compiler/xla/python/tpu_driver/client/BUILD b/tensorflow/compiler/xla/python/tpu_driver/client/BUILD index 30a220ece45..2592c439e7b 100644 --- a/tensorflow/compiler/xla/python/tpu_driver/client/BUILD +++ b/tensorflow/compiler/xla/python/tpu_driver/client/BUILD @@ -1,3 +1,9 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "pybind_extension") package( @@ -11,6 +17,7 @@ cc_library( hdrs = [ "tpu_client.h", ], + compatible_with = [], deps = [ "//tensorflow/compiler/xla:literal", "//tensorflow/compiler/xla:shape_util", diff --git a/tensorflow/compiler/xla/rpc/BUILD b/tensorflow/compiler/xla/rpc/BUILD index 6e345b06e43..8886dbd5ade 100644 --- a/tensorflow/compiler/xla/rpc/BUILD +++ b/tensorflow/compiler/xla/rpc/BUILD @@ -1,6 +1,8 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") load( "//tensorflow:tensorflow.bzl", + "if_tpu", "tf_cc_binary", "tf_cc_test", ) @@ -50,13 +52,15 @@ cc_library( srcs = ["grpc_service_main.cc"], deps = [ ":grpc_service", - "//tensorflow/compiler/xla/service:cpu_plugin", + "@com_google_absl//absl/strings:str_format", "//tensorflow/compiler/xla/service:platform_util", "//tensorflow/core:framework_internal", "//tensorflow/core:lib", - "@com_google_absl//absl/strings:str_format", tf_grpc_cc_dependency(), - ], + ] + if_tpu( + if_false = ["//tensorflow/compiler/xla/service:cpu_plugin"], + if_true = [], + ), ) tf_cc_binary( diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index 6874d00445c..13233262c12 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -1,6 +1,7 @@ # Description: # XLA service implementation. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/compiler/xla/tests:build_defs.bzl", "xla_test") load( "//tensorflow/core/platform:build_config.bzl", @@ -8,6 +9,12 @@ load( "tf_proto_library_py", ) load("//tensorflow:tensorflow.bzl", "tf_cc_test") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "internal_hlo_deps") load( "//tensorflow/core/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured", @@ -1018,6 +1025,7 @@ cc_library( cc_library( name = "cpu_plugin", + compatible_with = [], deps = [ ":service", "//tensorflow/compiler/xla/service/cpu:cpu_compiler", @@ -1037,6 +1045,7 @@ config_setting( # GPU plugin should be used or not. cc_library( name = "gpu_plugin", + compatible_with = [], deps = select( { ":with_mlir_gpu_support": [ @@ -1051,6 +1060,7 @@ cc_library( cc_library( name = "gpu_plugin_no_mlir", + compatible_with = [], deps = [ ":service", "//tensorflow/compiler/xla/service/gpu:gpu_compiler", @@ -1067,6 +1077,7 @@ cc_library( cc_library( name = "gpu_plugin_mlir", + compatible_with = [], deps = [ ":service", "//tensorflow/compiler/xla/service/gpu:gpu_transfer_manager", @@ -1147,6 +1158,10 @@ cc_library( ":maybe_owning_device_memory", ":shaped_buffer", ":stream_pool", + "@com_google_absl//absl/memory", + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/types:span", + "@com_google_absl//absl/types:variant", "//tensorflow/compiler/xla:debug_options_flags", "//tensorflow/compiler/xla:executable_run_options", "//tensorflow/compiler/xla:shape_tree", @@ -1161,11 +1176,7 @@ cc_library( "//tensorflow/stream_executor", "//tensorflow/stream_executor:device_description", "//tensorflow/stream_executor:device_memory_allocator", - "@com_google_absl//absl/memory", - "@com_google_absl//absl/strings:str_format", - "@com_google_absl//absl/types:span", - "@com_google_absl//absl/types:variant", - ], + ] + internal_hlo_deps(), ) cc_library( @@ -1194,6 +1205,7 @@ cc_library( name = "llvm_compiler", srcs = ["llvm_compiler.cc"], hdrs = ["llvm_compiler.h"], + compatible_with = [], deps = [ ":compiler", "//tensorflow/core:lib_internal", @@ -4042,6 +4054,7 @@ cc_library( name = "elemental_ir_emitter", srcs = ["elemental_ir_emitter.cc"], hdrs = ["elemental_ir_emitter.h"], + compatible_with = [], deps = [ ":hlo", ":hlo_casting_utils", diff --git a/tensorflow/compiler/xla/service/cpu/BUILD b/tensorflow/compiler/xla/service/cpu/BUILD index 4e25d667d03..d45f1c1aeb9 100644 --- a/tensorflow/compiler/xla/service/cpu/BUILD +++ b/tensorflow/compiler/xla/service/cpu/BUILD @@ -6,6 +6,12 @@ load( "//third_party/mkl:build_defs.bzl", "mkl_deps", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test", "tf_openmp_copts") load(":build_defs.bzl", "runtime_copts") load("//tensorflow/core/platform:build_config.bzl", "if_llvm_system_z_available") @@ -742,6 +748,7 @@ cc_library( name = "runtime_single_threaded_matmul", srcs = ["runtime_single_threaded_matmul.cc"], hdrs = ["runtime_single_threaded_matmul.h"], + compatible_with = get_compatible_with_cloud(), copts = runtime_copts(), visibility = ["//visibility:public"], deps = [ diff --git a/tensorflow/compiler/xla/service/gpu/BUILD b/tensorflow/compiler/xla/service/gpu/BUILD index c861ceffc05..aa316404715 100644 --- a/tensorflow/compiler/xla/service/gpu/BUILD +++ b/tensorflow/compiler/xla/service/gpu/BUILD @@ -26,6 +26,11 @@ load( "//tensorflow/core/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "if_nccl") load("//third_party/mlir:tblgen.bzl", "gentbl") @@ -61,6 +66,7 @@ cc_library( name = "gpu_executable_run_options", srcs = ["gpu_executable_run_options.cc"], hdrs = ["gpu_executable_run_options.h"], + compatible_with = get_compatible_with_cloud(), visibility = ["//visibility:public"], deps = [ "//tensorflow/compiler/xla:statusor", @@ -434,6 +440,7 @@ tf_cuda_library( ["dummy_all_reduce_thunk.cc"], ), hdrs = ["nccl_all_reduce_thunk.h"], + compatible_with = [], deps = [ ":buffer_allocations", ":hlo_execution_profiler", @@ -470,6 +477,7 @@ cc_library( hdrs = [ "gpu_debug_info_manager.h", ], + compatible_with = get_compatible_with_cloud(), deps = [ "//tensorflow/compiler/xla/service:buffer_assignment", "//tensorflow/compiler/xla/service:hlo", @@ -1374,6 +1382,7 @@ cc_library( cc_library( name = "xfeed_queue", hdrs = ["xfeed_queue.h"], + compatible_with = get_compatible_with_cloud(), deps = [ "//tensorflow/core:lib", "@com_google_absl//absl/base:core_headers", @@ -1384,6 +1393,7 @@ cc_library( name = "infeed_manager", srcs = ["infeed_manager.cc"], hdrs = ["infeed_manager.h"], + compatible_with = get_compatible_with_cloud(), deps = [ ":xfeed_queue", "//tensorflow/compiler/xla:shape_tree", @@ -1398,6 +1408,7 @@ cc_library( name = "outfeed_manager", srcs = ["outfeed_manager.cc"], hdrs = ["outfeed_manager.h"], + compatible_with = get_compatible_with_cloud(), deps = [ ":xfeed_queue", "//tensorflow/compiler/xla:literal", diff --git a/tensorflow/compiler/xla/service/interpreter/BUILD b/tensorflow/compiler/xla/service/interpreter/BUILD index 3444d4cae42..46767888916 100644 --- a/tensorflow/compiler/xla/service/interpreter/BUILD +++ b/tensorflow/compiler/xla/service/interpreter/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow/core/platform:build_config_root.bzl", "if_static", diff --git a/tensorflow/compiler/xla/service/spmd/BUILD b/tensorflow/compiler/xla/service/spmd/BUILD index 5fd7b7850cf..9ebaaa8242f 100644 --- a/tensorflow/compiler/xla/service/spmd/BUILD +++ b/tensorflow/compiler/xla/service/spmd/BUILD @@ -1,5 +1,6 @@ # Description: SPMD partitioning pass. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/compiler/xla/tests/BUILD b/tensorflow/compiler/xla/tests/BUILD index 361d2065a00..dbbb11f007f 100644 --- a/tensorflow/compiler/xla/tests/BUILD +++ b/tensorflow/compiler/xla/tests/BUILD @@ -6,7 +6,14 @@ load( "//tensorflow/core/platform:build_config_root.bzl", "tf_cuda_tests_tags", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "genrule") load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") package( default_visibility = [":friends"], @@ -177,6 +184,7 @@ cc_library( tf_cc_binary( name = "local_client_aot_test_helper", srcs = ["local_client_aot_test_helper.cc"], + compatible_with = [], deps = [ "//tensorflow/compiler/xla:types", "//tensorflow/compiler/xla:util", @@ -239,6 +247,7 @@ cc_library( testonly = True, srcs = ["llvm_irgen_test_base.cc"], hdrs = ["llvm_irgen_test_base.h"], + compatible_with = [], deps = [ ":codegen_test_base", ":filecheck", @@ -266,6 +275,7 @@ cc_library( testonly = True, srcs = ["filecheck.cc"], hdrs = ["filecheck.h"], + compatible_with = [], data = [ "@llvm-project//llvm:FileCheck", ], diff --git a/tensorflow/compiler/xla/tools/BUILD b/tensorflow/compiler/xla/tools/BUILD index fc1ca7d3105..bdbf7044fe2 100644 --- a/tensorflow/compiler/xla/tools/BUILD +++ b/tensorflow/compiler/xla/tools/BUILD @@ -1,5 +1,7 @@ # Tools and utilities that aid in XLA development and usage. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "if_cuda_or_rocm", diff --git a/tensorflow/compiler/xrt/BUILD b/tensorflow/compiler/xrt/BUILD index 172a970d207..b1fdeef421c 100644 --- a/tensorflow/compiler/xrt/BUILD +++ b/tensorflow/compiler/xrt/BUILD @@ -1,11 +1,12 @@ # Description: Operations defined for XRT +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_custom_op_py_library", - "tf_gen_op_libs", "tf_gen_op_wrapper_py", ) +load("//tensorflow:tensorflow.bzl", "tf_gen_op_libs") load( "//tensorflow/core/platform:build_config.bzl", "tf_proto_library_cc", diff --git a/tensorflow/compiler/xrt/cc/BUILD b/tensorflow/compiler/xrt/cc/BUILD index 99ab50c8a8d..c8932150cb5 100644 --- a/tensorflow/compiler/xrt/cc/BUILD +++ b/tensorflow/compiler/xrt/cc/BUILD @@ -1,7 +1,4 @@ -load( - "//tensorflow:tensorflow.bzl", - "tf_gen_op_wrappers_cc", -) +load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrappers_cc") package( default_visibility = ["//visibility:public"], diff --git a/tensorflow/compiler/xrt/kernels/BUILD b/tensorflow/compiler/xrt/kernels/BUILD index 494ba29e981..68c24f88703 100644 --- a/tensorflow/compiler/xrt/kernels/BUILD +++ b/tensorflow/compiler/xrt/kernels/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( default_visibility = [ "//learning/brain:__subpackages__", diff --git a/tensorflow/compiler/xrt/tests/BUILD b/tensorflow/compiler/xrt/tests/BUILD index 2f1faf1cdf1..724cfe38d54 100644 --- a/tensorflow/compiler/xrt/tests/BUILD +++ b/tensorflow/compiler/xrt/tests/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cuda_cc_test") load( "//tensorflow/core/platform:build_config_root.bzl", diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 6da5c43ce82..4c2a6e9ca05 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -65,13 +65,13 @@ load( "//tensorflow:tensorflow.bzl", - "cc_header_only_library", "if_android", "if_chromiumos", "if_cuda_or_rocm", "if_ios", "if_mobile", "if_not_windows", + "if_tpu", "tf_android_core_proto_headers", "tf_cc_test", "tf_cc_test_mkl", @@ -80,13 +80,18 @@ load( "tf_cuda_library", "tf_defines_nortti_if_lite_protos", "tf_features_nomodules_if_mobile", - "tf_gen_op_libs", "tf_genrule_cmd_append_to_srcs", "tf_opts_nortti_if_lite_protos", "tf_portable_full_lite_protos", "transitive_hdrs", ) +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "cc_header_only_library") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_gen_op_libs") + # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "if_nccl") @@ -102,6 +107,12 @@ load("//tensorflow:tensorflow.bzl", "tf_cc_tests_gpu") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_monitoring_framework_deps") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "genrule") + # For platform specific build config load( "//tensorflow/core/platform:build_config.bzl", @@ -882,7 +893,6 @@ cc_library( "//tensorflow/c/kernels:histogram_summary_op_lib", "//tensorflow/c/kernels:merge_summary_op_lib", "//tensorflow/c/kernels:summary_op_lib", - "//tensorflow/compiler/mlir/tensorflow:mlir_passthrough_op", ] + if_chromiumos( [], # Non-tpu platforms don't need tpu dependency. It would be best to guard @@ -907,7 +917,10 @@ cc_library( ]) + if_tensorrt([ "//tensorflow/compiler/tf2tensorrt:trt_engine_resource_ops_op_lib", "//tensorflow/compiler/tf2tensorrt:trt_op_libs", - ]), + ]) + if_tpu( + if_false = ["//tensorflow/compiler/mlir/tensorflow:mlir_passthrough_op"], + if_true = [], + ), alwayslink = 1, ) diff --git a/tensorflow/core/api_def/BUILD b/tensorflow/core/api_def/BUILD index e72f74e26e4..f9e2adaec6b 100644 --- a/tensorflow/core/api_def/BUILD +++ b/tensorflow/core/api_def/BUILD @@ -6,6 +6,8 @@ # :python_api_def # :java_api_def +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_binary", diff --git a/tensorflow/core/api_def/java_api/BUILD b/tensorflow/core/api_def/java_api/BUILD index 6aa8bb64961..682bfed7deb 100644 --- a/tensorflow/core/api_def/java_api/BUILD +++ b/tensorflow/core/api_def/java_api/BUILD @@ -1,6 +1,8 @@ # Description: # Provides Java ApiDef access and ApiDef validation for TensorFlow. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( default_visibility = ["//visibility:private"], licenses = ["notice"], # Apache 2.0 diff --git a/tensorflow/core/common_runtime/BUILD b/tensorflow/core/common_runtime/BUILD index 788f7dda5e1..f0378ad7538 100644 --- a/tensorflow/core/common_runtime/BUILD +++ b/tensorflow/core/common_runtime/BUILD @@ -1,6 +1,5 @@ load( "//tensorflow:tensorflow.bzl", - "cc_header_only_library", "if_tpu", "tf_cc_test", "tf_cc_test_mkl", @@ -10,12 +9,18 @@ load( "tf_openmp_copts", ) +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "cc_header_only_library") + # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cc_test_gpu") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + # For platform specific build config load( "//tensorflow/core/platform:build_config.bzl", diff --git a/tensorflow/core/common_runtime/eager/BUILD b/tensorflow/core/common_runtime/eager/BUILD index 38f5f85f528..f02d800430a 100644 --- a/tensorflow/core/common_runtime/eager/BUILD +++ b/tensorflow/core/common_runtime/eager/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", diff --git a/tensorflow/core/common_runtime/gpu/BUILD b/tensorflow/core/common_runtime/gpu/BUILD index 6618842f68f..efbf30d4e65 100644 --- a/tensorflow/core/common_runtime/gpu/BUILD +++ b/tensorflow/core/common_runtime/gpu/BUILD @@ -14,6 +14,9 @@ load("//tensorflow:tensorflow.bzl", "tf_cc_tests_gpu") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + # For platform specific build config load( "//tensorflow/core/platform:build_config.bzl", diff --git a/tensorflow/core/data/BUILD b/tensorflow/core/data/BUILD index 1b6e6790559..e86603828ec 100644 --- a/tensorflow/core/data/BUILD +++ b/tensorflow/core/data/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") load( "//tensorflow/core/platform:build_config.bzl", diff --git a/tensorflow/core/data/service/BUILD b/tensorflow/core/data/service/BUILD index ef4f8fbe4f5..5176421c02d 100644 --- a/tensorflow/core/data/service/BUILD +++ b/tensorflow/core/data/service/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("@com_github_grpc_grpc//bazel:cc_grpc_library.bzl", "cc_grpc_library") load( "//tensorflow/core/platform:build_config.bzl", @@ -5,10 +6,17 @@ load( "tf_proto_library", "tf_protos_profiler_service", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "cc_header_only_library") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") load( "//tensorflow:tensorflow.bzl", - "cc_header_only_library", "tf_cc_test", ) @@ -152,6 +160,7 @@ tf_cc_test( cc_grpc_library( name = "dispatcher_cc_grpc_proto", srcs = [":dispatcher_proto"], + compatible_with = get_compatible_with_cloud(), generate_mocks = True, grpc_only = True, deps = [":dispatcher_proto_cc"], @@ -422,6 +431,7 @@ tf_cc_test( cc_grpc_library( name = "worker_cc_grpc_proto", srcs = [":worker_proto"], + compatible_with = get_compatible_with_cloud(), generate_mocks = True, grpc_only = True, deps = [":worker_proto_cc"], diff --git a/tensorflow/core/debug/BUILD b/tensorflow/core/debug/BUILD index ca3118c51e0..45cfda90ff1 100644 --- a/tensorflow/core/debug/BUILD +++ b/tensorflow/core/debug/BUILD @@ -12,6 +12,7 @@ # a watch state. # ":debug_node_key" - Defines a struct used for tracking tensors. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") load( "//tensorflow:tensorflow.bzl", diff --git a/tensorflow/core/distributed_runtime/BUILD b/tensorflow/core/distributed_runtime/BUILD index fa8a5eb8cfc..2459dfb5949 100644 --- a/tensorflow/core/distributed_runtime/BUILD +++ b/tensorflow/core/distributed_runtime/BUILD @@ -3,9 +3,11 @@ # to be distributed and performed in parallel across multiple # processes. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_copts", "tf_cuda_library") load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") # buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") # buildifier: disable=same-origin-load # For platform specific build config load( diff --git a/tensorflow/core/distributed_runtime/eager/BUILD b/tensorflow/core/distributed_runtime/eager/BUILD index fb9808b80cf..08814653e3e 100644 --- a/tensorflow/core/distributed_runtime/eager/BUILD +++ b/tensorflow/core/distributed_runtime/eager/BUILD @@ -1,3 +1,9 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") load( "//tensorflow:tensorflow.bzl", diff --git a/tensorflow/core/distributed_runtime/rpc/BUILD b/tensorflow/core/distributed_runtime/rpc/BUILD index 8b27a34f1a2..f2c9889b549 100644 --- a/tensorflow/core/distributed_runtime/rpc/BUILD +++ b/tensorflow/core/distributed_runtime/rpc/BUILD @@ -1,6 +1,7 @@ # Description: # RPC communication interfaces and implementations for TensorFlow. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "if_windows", @@ -12,6 +13,7 @@ load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") # buildifier: disable=sa load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_tests") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_grpc_dependency") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") # buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") # buildifier: disable=same-origin-load # For platform specific build config load( diff --git a/tensorflow/core/distributed_runtime/rpc/eager/BUILD b/tensorflow/core/distributed_runtime/rpc/eager/BUILD index ff362c3411f..ac0339fe6cd 100644 --- a/tensorflow/core/distributed_runtime/rpc/eager/BUILD +++ b/tensorflow/core/distributed_runtime/rpc/eager/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") load("//tensorflow:tensorflow.bzl", "tf_cc_test") diff --git a/tensorflow/core/example/BUILD b/tensorflow/core/example/BUILD index d817f782daa..5c5168d295f 100644 --- a/tensorflow/core/example/BUILD +++ b/tensorflow/core/example/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "@io_bazel_rules_closure//closure:defs.bzl", "closure_proto_library", diff --git a/tensorflow/core/framework/BUILD b/tensorflow/core/framework/BUILD index d2c305e5eed..e255eafde16 100644 --- a/tensorflow/core/framework/BUILD +++ b/tensorflow/core/framework/BUILD @@ -9,8 +9,16 @@ load( "tf_cc_tests", "tf_copts", "tf_cuda_library", - "tf_generate_proto_text_sources", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_generate_proto_text_sources") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_selective_registration_deps") load( "//tensorflow/core/platform:build_config_root.bzl", "if_static", @@ -20,9 +28,6 @@ load( "cc_library", ) -# buildifier: disable=same-origin-load -load("//tensorflow:tensorflow.bzl", "tf_selective_registration_deps") - package( default_visibility = [ "//tensorflow/core:__subpackages__", diff --git a/tensorflow/core/graph/BUILD b/tensorflow/core/graph/BUILD index 4834cdc53ed..b9461291bfc 100644 --- a/tensorflow/core/graph/BUILD +++ b/tensorflow/core/graph/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", diff --git a/tensorflow/core/grappler/BUILD b/tensorflow/core/grappler/BUILD index fdd72968a40..8a9b23ff96f 100644 --- a/tensorflow/core/grappler/BUILD +++ b/tensorflow/core/grappler/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cuda_library") package( diff --git a/tensorflow/core/grappler/clusters/BUILD b/tensorflow/core/grappler/clusters/BUILD index 5b8a545031e..8ac60a0916e 100644 --- a/tensorflow/core/grappler/clusters/BUILD +++ b/tensorflow/core/grappler/clusters/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda") load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cuda_library") load( diff --git a/tensorflow/core/grappler/costs/BUILD b/tensorflow/core/grappler/costs/BUILD index 02c69920b84..b205f5c3e56 100644 --- a/tensorflow/core/grappler/costs/BUILD +++ b/tensorflow/core/grappler/costs/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", diff --git a/tensorflow/core/grappler/graph_analyzer/BUILD b/tensorflow/core/grappler/graph_analyzer/BUILD index d252a950557..03ebad9686a 100644 --- a/tensorflow/core/grappler/graph_analyzer/BUILD +++ b/tensorflow/core/grappler/graph_analyzer/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/grappler/inputs/BUILD b/tensorflow/core/grappler/inputs/BUILD index 250665995e6..9faad74f7d8 100644 --- a/tensorflow/core/grappler/inputs/BUILD +++ b/tensorflow/core/grappler/inputs/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/grappler/inputs/testdata/BUILD b/tensorflow/core/grappler/inputs/testdata/BUILD index fa859118c84..42b3bef163d 100644 --- a/tensorflow/core/grappler/inputs/testdata/BUILD +++ b/tensorflow/core/grappler/inputs/testdata/BUILD @@ -1,6 +1,8 @@ # Description: # grappler testdata packages. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/grappler/optimizers/BUILD b/tensorflow/core/grappler/optimizers/BUILD index e81816f0acb..7fc74b0aca5 100644 --- a/tensorflow/core/grappler/optimizers/BUILD +++ b/tensorflow/core/grappler/optimizers/BUILD @@ -1,4 +1,11 @@ -load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cc_test_mkl", "tf_copts", "tf_cuda_cc_test", "tf_kernel_library") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") +load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_cc_test_mkl", "tf_copts", "tf_cuda_cc_test") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") # Platform specific build config load( diff --git a/tensorflow/core/grappler/optimizers/data/BUILD b/tensorflow/core/grappler/optimizers/data/BUILD index 54a21706c37..b9731e2ef0e 100644 --- a/tensorflow/core/grappler/optimizers/data/BUILD +++ b/tensorflow/core/grappler/optimizers/data/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") load("//tensorflow/core/platform:build_config.bzl", "tf_protos_all") diff --git a/tensorflow/core/grappler/optimizers/data/vectorization/BUILD b/tensorflow/core/grappler/optimizers/data/vectorization/BUILD index 57962163562..9ae0a9d5391 100644 --- a/tensorflow/core/grappler/optimizers/data/vectorization/BUILD +++ b/tensorflow/core/grappler/optimizers/data/vectorization/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") load("//tensorflow/core/platform:build_config.bzl", "tf_protos_all") diff --git a/tensorflow/core/grappler/utils/BUILD b/tensorflow/core/grappler/utils/BUILD index 8e04b573770..e87ec9d16ca 100644 --- a/tensorflow/core/grappler/utils/BUILD +++ b/tensorflow/core/grappler/utils/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") load( "//tensorflow/core/platform:build_config.bzl", diff --git a/tensorflow/core/grappler/verifiers/BUILD b/tensorflow/core/grappler/verifiers/BUILD index 939972b0617..c9501d95a54 100644 --- a/tensorflow/core/grappler/verifiers/BUILD +++ b/tensorflow/core/grappler/verifiers/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/kernels/BUILD b/tensorflow/core/kernels/BUILD index ea99c87223c..39a2d4e0bd4 100644 --- a/tensorflow/core/kernels/BUILD +++ b/tensorflow/core/kernels/BUILD @@ -1,7 +1,7 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("@bazel_skylib//rules:build_test.bzl", "build_test") load( "//tensorflow:tensorflow.bzl", - "cc_header_only_library", "if_android", "if_cuda_or_rocm", "if_mobile", @@ -12,11 +12,16 @@ load( "tf_cc_tests", "tf_copts", "tf_cuda_library", - "tf_kernel_library", "tf_opts_nortti_if_lite_protos", ) load("//tensorflow/core/kernels/mlir_generated:build_defs.bzl", "if_mlir_generated_gpu_kernels_enabled") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "cc_header_only_library") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") + # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "if_nccl") @@ -26,6 +31,9 @@ load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_tests") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + # For platform specific build config load( "//tensorflow/core/platform:build_config.bzl", @@ -501,6 +509,7 @@ tf_cuda_library( name = "gpu_utils", srcs = if_cuda_or_rocm(["gpu_utils.cc"]), hdrs = ["gpu_utils.h"], + compatible_with = [], deps = [ ":gpu_util_hdrs", "//tensorflow/core:autotuning_proto_cc", @@ -1476,6 +1485,7 @@ tf_kernel_library( tf_kernel_library( name = "cudnn_rnn_kernels", srcs = ["cudnn_rnn_ops.cc"], + compatible_with = [], visibility = ["//visibility:public"], deps = [ ":bounds_check_lib", @@ -3832,13 +3842,13 @@ tf_kernel_library( "//tensorflow/core:lib_internal", "//tensorflow/core/util:image_resizer_state", "//tensorflow/core/util/proto:proto_utils", - "//tensorflow/stream_executor/gpu:gpu_asm_opts", ] + select({ ":xsmm_convolutions": [ "@libxsmm_archive//:xsmm_avx", ], "//conditions:default": [], }) + if_cuda([ + "//tensorflow/stream_executor/gpu:gpu_asm_opts", "//tensorflow/core/platform/default/build_config:cublas_plugin", "//tensorflow/core/platform/default/build_config:cudnn_plugin", "//tensorflow/stream_executor:tf_allocator_adapter", diff --git a/tensorflow/core/kernels/batching_util/BUILD b/tensorflow/core/kernels/batching_util/BUILD index 0ecc6f7f10c..16de576d2c9 100644 --- a/tensorflow/core/kernels/batching_util/BUILD +++ b/tensorflow/core/kernels/batching_util/BUILD @@ -1,5 +1,6 @@ # Description: Utilities. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/kernels/boosted_trees/BUILD b/tensorflow/core/kernels/boosted_trees/BUILD index 4ca40dc4177..c9cb81694b7 100644 --- a/tensorflow/core/kernels/boosted_trees/BUILD +++ b/tensorflow/core/kernels/boosted_trees/BUILD @@ -1,11 +1,12 @@ # Description: # OpKernels for boosted trees ops. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", - "tf_kernel_library", ) +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") load( "//tensorflow/core/platform:build_config.bzl", "tf_proto_library", diff --git a/tensorflow/core/kernels/boosted_trees/quantiles/BUILD b/tensorflow/core/kernels/boosted_trees/quantiles/BUILD index fb03e284d8d..8b25e00965e 100644 --- a/tensorflow/core/kernels/boosted_trees/quantiles/BUILD +++ b/tensorflow/core/kernels/boosted_trees/quantiles/BUILD @@ -1,6 +1,8 @@ # Description: # This directory contains common quantile utilities used in boosted_trees. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/kernels/data/BUILD b/tensorflow/core/kernels/data/BUILD index c8ebea599c8..e9c5430d0fb 100644 --- a/tensorflow/core/kernels/data/BUILD +++ b/tensorflow/core/kernels/data/BUILD @@ -1,13 +1,19 @@ # Description: # OpKernels for tf.data +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "if_not_mobile", "tf_cc_test", - "tf_kernel_library", ) +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") + package( default_visibility = ["//visibility:public"], licenses = ["notice"], # Apache 2.0 diff --git a/tensorflow/core/kernels/data/experimental/BUILD b/tensorflow/core/kernels/data/experimental/BUILD index cd318b1343b..86c37a4b6d5 100644 --- a/tensorflow/core/kernels/data/experimental/BUILD +++ b/tensorflow/core/kernels/data/experimental/BUILD @@ -1,11 +1,9 @@ # Description: # Contains experimental kernels for datasets and iterators. -load( - "//tensorflow:tensorflow.bzl", - "tf_cc_test", - "tf_kernel_library", -) +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") +load("//tensorflow:tensorflow.bzl", "tf_cc_test") +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") package( default_visibility = ["//tensorflow:internal"], diff --git a/tensorflow/core/kernels/data/experimental/sql/BUILD b/tensorflow/core/kernels/data/experimental/sql/BUILD index 923cebf9f03..937ab34c587 100644 --- a/tensorflow/core/kernels/data/experimental/sql/BUILD +++ b/tensorflow/core/kernels/data/experimental/sql/BUILD @@ -1,6 +1,8 @@ # Description: # SQL library. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( default_visibility = ["//visibility:public"], licenses = ["notice"], # Apache 2.0 diff --git a/tensorflow/core/kernels/fuzzing/BUILD b/tensorflow/core/kernels/fuzzing/BUILD index 4133462cad5..ce71fe77cb2 100644 --- a/tensorflow/core/kernels/fuzzing/BUILD +++ b/tensorflow/core/kernels/fuzzing/BUILD @@ -1,5 +1,12 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + +# buildifier: disable=same-origin-load load("//tensorflow/core/kernels/fuzzing:tf_ops_fuzz_target_lib.bzl", "tf_ops_fuzz_target_lib") + +# buildifier: disable=same-origin-load load("//tensorflow/core/kernels/fuzzing:tf_ops_fuzz_target_lib.bzl", "tf_oss_fuzz_corpus") + +# buildifier: disable=same-origin-load load("//tensorflow/core/kernels/fuzzing:tf_ops_fuzz_target_lib.bzl", "tf_oss_fuzz_dict") package( diff --git a/tensorflow/core/kernels/hexagon/BUILD b/tensorflow/core/kernels/hexagon/BUILD index 9b15b729ffe..6174ced9d4d 100644 --- a/tensorflow/core/kernels/hexagon/BUILD +++ b/tensorflow/core/kernels/hexagon/BUILD @@ -1,11 +1,12 @@ # Description: # quantization-specific OpKernels for hexagon +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", - "tf_kernel_library", ) +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") package( default_visibility = ["//visibility:public"], diff --git a/tensorflow/core/kernels/image/BUILD b/tensorflow/core/kernels/image/BUILD index 1c1004a43bb..82d770a7e9e 100644 --- a/tensorflow/core/kernels/image/BUILD +++ b/tensorflow/core/kernels/image/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("@bazel_skylib//rules:build_test.bzl", "build_test") load( "//tensorflow:tensorflow.bzl", @@ -6,14 +7,21 @@ load( "tf_cc_test", "tf_cc_tests", "tf_copts", - "tf_kernel_library", ) load( "//tensorflow/core/platform:build_config_root.bzl", "tf_cuda_tests_tags", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") + # TODO(rmlarsen): Remove ASAP. package_group( name = "friends", diff --git a/tensorflow/core/kernels/linalg/BUILD b/tensorflow/core/kernels/linalg/BUILD index ab25fad3ec3..dabe8bd0b7d 100644 --- a/tensorflow/core/kernels/linalg/BUILD +++ b/tensorflow/core/kernels/linalg/BUILD @@ -1,15 +1,23 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "if_cuda_or_rocm", - "tf_kernel_library", ) load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda") load( "@local_config_rocm//rocm:build_defs.bzl", "if_rocm", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") + package( default_visibility = [ "//tensorflow:__subpackages__", diff --git a/tensorflow/core/kernels/neon/BUILD b/tensorflow/core/kernels/neon/BUILD index 0a9bd5a7fd5..668ac4b2f64 100644 --- a/tensorflow/core/kernels/neon/BUILD +++ b/tensorflow/core/kernels/neon/BUILD @@ -1,7 +1,4 @@ -load( - "//tensorflow:tensorflow.bzl", - "tf_kernel_library", -) +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") # Description: # Kernel implementations using Neon intrinsics. diff --git a/tensorflow/core/kernels/rnn/BUILD b/tensorflow/core/kernels/rnn/BUILD index 0f0e358d9ed..0ac9b08ed15 100644 --- a/tensorflow/core/kernels/rnn/BUILD +++ b/tensorflow/core/kernels/rnn/BUILD @@ -4,8 +4,8 @@ load( "//tensorflow:tensorflow.bzl", "tf_gpu_library", - "tf_kernel_library", ) +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") load( "//tensorflow/core/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured", diff --git a/tensorflow/core/kernels/sparse/BUILD b/tensorflow/core/kernels/sparse/BUILD index bfb6c4934bb..df4afb3159b 100644 --- a/tensorflow/core/kernels/sparse/BUILD +++ b/tensorflow/core/kernels/sparse/BUILD @@ -1,11 +1,12 @@ # Description: Op kernels for sparse matrix operations. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "if_cuda_or_rocm", "tf_cc_test", - "tf_kernel_library", ) +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") package( default_visibility = ["//visibility:public"], diff --git a/tensorflow/core/kernels/special_math/BUILD b/tensorflow/core/kernels/special_math/BUILD index 9a49f8263e4..3775ab1434b 100644 --- a/tensorflow/core/kernels/special_math/BUILD +++ b/tensorflow/core/kernels/special_math/BUILD @@ -1,7 +1,4 @@ -load( - "//tensorflow:tensorflow.bzl", - "tf_kernel_library", -) +load("//tensorflow:tensorflow.bzl", "tf_kernel_library") # Implementation of Special Functions kernels. diff --git a/tensorflow/core/kernels/tensor_forest/BUILD b/tensorflow/core/kernels/tensor_forest/BUILD index 8bab8bd3974..bfa42e2dd44 100644 --- a/tensorflow/core/kernels/tensor_forest/BUILD +++ b/tensorflow/core/kernels/tensor_forest/BUILD @@ -1,6 +1,7 @@ # Description: # OpKernels for tensor forest ops. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_kernel_library") package( diff --git a/tensorflow/core/lib/bmp/BUILD b/tensorflow/core/lib/bmp/BUILD index b12424b4814..186c3a0753f 100644 --- a/tensorflow/core/lib/bmp/BUILD +++ b/tensorflow/core/lib/bmp/BUILD @@ -1,6 +1,8 @@ # Description: # bmp test data packages. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/lib/core/BUILD b/tensorflow/core/lib/core/BUILD index 491e4c5e7aa..886e3e9140c 100644 --- a/tensorflow/core/lib/core/BUILD +++ b/tensorflow/core/lib/core/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load("//tensorflow/core/platform:build_config.bzl", "tf_proto_library") load( "//tensorflow/core/platform:rules_cc.bzl", diff --git a/tensorflow/core/lib/gtl/BUILD b/tensorflow/core/lib/gtl/BUILD index 0a96a17e590..650d6f8ddf5 100644 --- a/tensorflow/core/lib/gtl/BUILD +++ b/tensorflow/core/lib/gtl/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/core/platform:rules_cc.bzl", "cc_library", diff --git a/tensorflow/core/lib/gtl/subtle/BUILD b/tensorflow/core/lib/gtl/subtle/BUILD index febe982446c..6f73f441f80 100644 --- a/tensorflow/core/lib/gtl/subtle/BUILD +++ b/tensorflow/core/lib/gtl/subtle/BUILD @@ -1,6 +1,8 @@ # Description: # gtl subtle packages. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/lib/hash/BUILD b/tensorflow/core/lib/hash/BUILD index 993ccf88341..8752aff7a92 100644 --- a/tensorflow/core/lib/hash/BUILD +++ b/tensorflow/core/lib/hash/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow:tensorflow.bzl", "if_linux_x86_64", diff --git a/tensorflow/core/lib/histogram/BUILD b/tensorflow/core/lib/histogram/BUILD index 006a829ba62..ff8c1b69281 100644 --- a/tensorflow/core/lib/histogram/BUILD +++ b/tensorflow/core/lib/histogram/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/core/platform:rules_cc.bzl", "cc_library", diff --git a/tensorflow/core/lib/io/BUILD b/tensorflow/core/lib/io/BUILD index 797e9ad1a4b..fc9148c0f04 100644 --- a/tensorflow/core/lib/io/BUILD +++ b/tensorflow/core/lib/io/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/core/platform:rules_cc.bzl", "cc_library", diff --git a/tensorflow/core/lib/llvm_rtti/BUILD b/tensorflow/core/lib/llvm_rtti/BUILD index cc8f5d5f46c..0357523dd44 100644 --- a/tensorflow/core/lib/llvm_rtti/BUILD +++ b/tensorflow/core/lib/llvm_rtti/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") + # Library for using LLVM style RTTI in TensorFlow. load( "//tensorflow/core/platform:rules_cc.bzl", diff --git a/tensorflow/core/lib/lmdb/testdata/BUILD b/tensorflow/core/lib/lmdb/testdata/BUILD index 85046175f61..188998c9797 100644 --- a/tensorflow/core/lib/lmdb/testdata/BUILD +++ b/tensorflow/core/lib/lmdb/testdata/BUILD @@ -1,6 +1,8 @@ # Description: # lmdb test data packages. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/lib/math/BUILD b/tensorflow/core/lib/math/BUILD index a095dded61c..1fcf7c0aadc 100644 --- a/tensorflow/core/lib/math/BUILD +++ b/tensorflow/core/lib/math/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/core/platform:rules_cc.bzl", "cc_library", diff --git a/tensorflow/core/lib/monitoring/BUILD b/tensorflow/core/lib/monitoring/BUILD index bbed9a58452..fc1ab3fb59b 100644 --- a/tensorflow/core/lib/monitoring/BUILD +++ b/tensorflow/core/lib/monitoring/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/core/platform:rules_cc.bzl", "cc_library", diff --git a/tensorflow/core/lib/png/testdata/BUILD b/tensorflow/core/lib/png/testdata/BUILD index 0885aee2e6f..7331a0f7d2c 100644 --- a/tensorflow/core/lib/png/testdata/BUILD +++ b/tensorflow/core/lib/png/testdata/BUILD @@ -1,6 +1,8 @@ # Description: # PNG test data packages. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/lib/psnr/BUILD b/tensorflow/core/lib/psnr/BUILD index 386f1a5bd06..196047c14c9 100644 --- a/tensorflow/core/lib/psnr/BUILD +++ b/tensorflow/core/lib/psnr/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") + package( default_visibility = [ "//tensorflow/core:__pkg__", diff --git a/tensorflow/core/lib/random/BUILD b/tensorflow/core/lib/random/BUILD index 88d2f0280f1..54670302e98 100644 --- a/tensorflow/core/lib/random/BUILD +++ b/tensorflow/core/lib/random/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/core/platform:rules_cc.bzl", "cc_library", diff --git a/tensorflow/core/lib/ssim/BUILD b/tensorflow/core/lib/ssim/BUILD index 7d9b72b11b0..a3e8ec48e96 100644 --- a/tensorflow/core/lib/ssim/BUILD +++ b/tensorflow/core/lib/ssim/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") + package( default_visibility = [ "//tensorflow/core:__pkg__", diff --git a/tensorflow/core/lib/strings/BUILD b/tensorflow/core/lib/strings/BUILD index 15dc7fbfe7e..c855b278afe 100644 --- a/tensorflow/core/lib/strings/BUILD +++ b/tensorflow/core/lib/strings/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/core/platform:rules_cc.bzl", "cc_library", diff --git a/tensorflow/core/nccl/BUILD b/tensorflow/core/nccl/BUILD index e49b89de1f5..a076d388b70 100644 --- a/tensorflow/core/nccl/BUILD +++ b/tensorflow/core/nccl/BUILD @@ -2,7 +2,13 @@ # Wrap NVIDIA (https://github.com/NVIDIA/nccl) NCCL with tensorflow ops. # APIs are meant to change over time. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") load("//tensorflow:tensorflow.bzl", "if_cuda_or_rocm", "tf_copts") load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda") load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm") diff --git a/tensorflow/core/ops/compat/BUILD b/tensorflow/core/ops/compat/BUILD index 47ab66cd944..7bfe06f2806 100644 --- a/tensorflow/core/ops/compat/BUILD +++ b/tensorflow/core/ops/compat/BUILD @@ -2,6 +2,7 @@ # For keeping the history of OpDefs for every major version of TensorFlow, # to validate that we don't make backwards-incompatible changes. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_binary", diff --git a/tensorflow/core/ops/compat/ops_history_v1/BUILD b/tensorflow/core/ops/compat/ops_history_v1/BUILD index dfd7dab25bf..808031e7e5d 100644 --- a/tensorflow/core/ops/compat/ops_history_v1/BUILD +++ b/tensorflow/core/ops/compat/ops_history_v1/BUILD @@ -3,6 +3,8 @@ # to validate that we don't make backwards-incompatible changes in particular # for v1. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/ops/compat/ops_history_v2/BUILD b/tensorflow/core/ops/compat/ops_history_v2/BUILD index a7462807779..dfc10238c29 100644 --- a/tensorflow/core/ops/compat/ops_history_v2/BUILD +++ b/tensorflow/core/ops/compat/ops_history_v2/BUILD @@ -3,6 +3,8 @@ # to validate that we don't make backwards-incompatible changes in particular # for v2. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/platform/BUILD b/tensorflow/core/platform/BUILD index 80d0a916fe3..ffbd55ab9b5 100644 --- a/tensorflow/core/platform/BUILD +++ b/tensorflow/core/platform/BUILD @@ -54,6 +54,9 @@ load("@bazel_skylib//:bzl_library.bzl", "bzl_library") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_cc_test_gpu") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable") @@ -202,6 +205,7 @@ cc_library( cc_library( name = "cuda", hdrs = ["cuda.h"], + compatible_with = [], deps = [ ":platform", "//tensorflow/stream_executor/cuda:cuda_activation_header", diff --git a/tensorflow/core/platform/cloud/BUILD b/tensorflow/core/platform/cloud/BUILD index 4b934fbac8d..f52c97282bd 100644 --- a/tensorflow/core/platform/cloud/BUILD +++ b/tensorflow/core/platform/cloud/BUILD @@ -1,6 +1,7 @@ # Description: # Cloud file system implementation. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "if_windows", diff --git a/tensorflow/core/platform/cloud/testdata/BUILD b/tensorflow/core/platform/cloud/testdata/BUILD index c7f1b8ef29b..88097d2214b 100644 --- a/tensorflow/core/platform/cloud/testdata/BUILD +++ b/tensorflow/core/platform/cloud/testdata/BUILD @@ -1,5 +1,7 @@ # Cloud test data files. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/platform/default/BUILD b/tensorflow/core/platform/default/BUILD index 40883cb69d4..a1f23563cf5 100644 --- a/tensorflow/core/platform/default/BUILD +++ b/tensorflow/core/platform/default/BUILD @@ -1,10 +1,8 @@ # Tensorflow default + linux implementations of tensorflow/core/platform libraries. load("@bazel_skylib//:bzl_library.bzl", "bzl_library") +load("//tensorflow:tensorflow.bzl", "filegroup") load("//tensorflow:tensorflow.bzl", "if_not_windows", "tf_copts") -load( - "//tensorflow/core/platform:rules_cc.bzl", - "cc_library", -) +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") package( default_visibility = [ @@ -49,6 +47,7 @@ cc_library( name = "cuda_libdevice_path", srcs = ["cuda_libdevice_path.cc"], hdrs = ["//tensorflow/core/platform:cuda_libdevice_path.h"], + compatible_with = [], tags = [ "manual", "no_oss", @@ -337,6 +336,7 @@ cc_library( name = "rocm_rocdl_path", srcs = ["rocm_rocdl_path.cc"], hdrs = ["//tensorflow/core/platform:rocm_rocdl_path.h"], + compatible_with = [], tags = [ "manual", "no_oss", diff --git a/tensorflow/core/platform/default/build_config.bzl b/tensorflow/core/platform/default/build_config.bzl index 08ef8a6628a..3b9f89232a2 100644 --- a/tensorflow/core/platform/default/build_config.bzl +++ b/tensorflow/core/platform/default/build_config.bzl @@ -668,6 +668,7 @@ def tf_additional_core_deps(): clean_dep("//tensorflow:linux_s390x"): [], clean_dep("//tensorflow:windows"): [], clean_dep("//tensorflow:no_hdfs_support"): [], + clean_dep("//tensorflow:with_tpu_support"): [], "//conditions:default": [ clean_dep("//tensorflow/core/platform/hadoop:hadoop_file_system"), ], diff --git a/tensorflow/core/platform/default/build_config/BUILD b/tensorflow/core/platform/default/build_config/BUILD index 2e2ef2c766c..83eadf2c460 100644 --- a/tensorflow/core/platform/default/build_config/BUILD +++ b/tensorflow/core/platform/default/build_config/BUILD @@ -1,6 +1,7 @@ # Description: # Platform-specific build configurations. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_copts", "tf_cuda_library") load("//tensorflow/core/platform:build_config_root.bzl", "if_static") diff --git a/tensorflow/core/platform/hadoop/BUILD b/tensorflow/core/platform/hadoop/BUILD index 49d9e9975cf..7f24ba3d07e 100644 --- a/tensorflow/core/platform/hadoop/BUILD +++ b/tensorflow/core/platform/hadoop/BUILD @@ -1,6 +1,7 @@ # Description: # Hadoop file system implementation. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_test", @@ -15,6 +16,7 @@ cc_library( name = "hadoop_file_system", srcs = ["hadoop_file_system.cc"], hdrs = ["hadoop_file_system.h"], + compatible_with = [], deps = [ "//tensorflow/core:lib", "//tensorflow/core:lib_internal", diff --git a/tensorflow/core/platform/windows/BUILD b/tensorflow/core/platform/windows/BUILD index dddb4b9aed4..235900ac759 100644 --- a/tensorflow/core/platform/windows/BUILD +++ b/tensorflow/core/platform/windows/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") + # Tensorflow windows-specific implementations of tensorflow/core/platform libraries. load( "//tensorflow:tensorflow.bzl", diff --git a/tensorflow/core/profiler/BUILD b/tensorflow/core/profiler/BUILD index 543a7d6467b..527d538b75c 100644 --- a/tensorflow/core/profiler/BUILD +++ b/tensorflow/core/profiler/BUILD @@ -1,4 +1,6 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") load("//tensorflow:tensorflow.bzl", "tf_cc_binary") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/core/platform:build_config.bzl", "tf_additional_all_protos", "tf_proto_library") package( diff --git a/tensorflow/core/profiler/convert/BUILD b/tensorflow/core/profiler/convert/BUILD index 35c3a8761b0..1af3b170b18 100644 --- a/tensorflow/core/profiler/convert/BUILD +++ b/tensorflow/core/profiler/convert/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/profiler/internal/BUILD b/tensorflow/core/profiler/internal/BUILD index d484ad9eff1..bbd73edffe6 100644 --- a/tensorflow/core/profiler/internal/BUILD +++ b/tensorflow/core/profiler/internal/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "if_not_windows", "tf_cc_test") load("//tensorflow/core/platform:build_config_root.bzl", "if_static") diff --git a/tensorflow/core/profiler/internal/advisor/BUILD b/tensorflow/core/profiler/internal/advisor/BUILD index f6ab03a0100..c5e592525d9 100644 --- a/tensorflow/core/profiler/internal/advisor/BUILD +++ b/tensorflow/core/profiler/internal/advisor/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/profiler/internal/cpu/BUILD b/tensorflow/core/profiler/internal/cpu/BUILD index d8c84425e2b..6fb518d413e 100644 --- a/tensorflow/core/profiler/internal/cpu/BUILD +++ b/tensorflow/core/profiler/internal/cpu/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/profiler/internal/testdata/BUILD b/tensorflow/core/profiler/internal/testdata/BUILD index e2701e7b978..577e3d2032d 100644 --- a/tensorflow/core/profiler/internal/testdata/BUILD +++ b/tensorflow/core/profiler/internal/testdata/BUILD @@ -1,6 +1,8 @@ # Description: # Profiler test data packages. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/profiler/lib/BUILD b/tensorflow/core/profiler/lib/BUILD index 67eb9923986..fd209259c3b 100644 --- a/tensorflow/core/profiler/lib/BUILD +++ b/tensorflow/core/profiler/lib/BUILD @@ -1,5 +1,11 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/core/platform:build_config_root.bzl", "if_static") load("//tensorflow:tensorflow.bzl", "if_not_android", "tf_cc_test", "tf_cuda_library") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_pybind_cc_library_wrapper") package( diff --git a/tensorflow/core/profiler/rpc/BUILD b/tensorflow/core/profiler/rpc/BUILD index 81861b95a3e..834ef1304ac 100644 --- a/tensorflow/core/profiler/rpc/BUILD +++ b/tensorflow/core/profiler/rpc/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_external_workspace_visible") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_pybind_cc_library_wrapper") # buildifier: disable=same-origin-load diff --git a/tensorflow/core/profiler/rpc/client/BUILD b/tensorflow/core/profiler/rpc/client/BUILD index eb7f236d587..00792405c70 100644 --- a/tensorflow/core/profiler/rpc/client/BUILD +++ b/tensorflow/core/profiler/rpc/client/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") load("//tensorflow:tensorflow.bzl", "tf_pybind_cc_library_wrapper") # buildifier: disable=same-origin-load diff --git a/tensorflow/core/profiler/rpc/oss/BUILD b/tensorflow/core/profiler/rpc/oss/BUILD index 12bc92a68e8..7e30302b718 100644 --- a/tensorflow/core/profiler/rpc/oss/BUILD +++ b/tensorflow/core/profiler/rpc/oss/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_grpc_cc_dependency") package( diff --git a/tensorflow/core/profiler/utils/BUILD b/tensorflow/core/profiler/utils/BUILD index d4b64b089d3..68697d32b9c 100644 --- a/tensorflow/core/profiler/utils/BUILD +++ b/tensorflow/core/profiler/utils/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/public/BUILD b/tensorflow/core/public/BUILD index e440735ed3a..f4c11601bb6 100644 --- a/tensorflow/core/public/BUILD +++ b/tensorflow/core/public/BUILD @@ -1,3 +1,6 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( default_visibility = [ "//tensorflow/core:__subpackages__", diff --git a/tensorflow/core/summary/BUILD b/tensorflow/core/summary/BUILD index 44c9b4b8b17..d39abb0b9ec 100644 --- a/tensorflow/core/summary/BUILD +++ b/tensorflow/core/summary/BUILD @@ -1,6 +1,7 @@ # Description: # C++ implementation code for the summary writing APIs. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "tf_cc_binary", diff --git a/tensorflow/core/tpu/BUILD b/tensorflow/core/tpu/BUILD index c5a5910f4a5..cd5beec6296 100644 --- a/tensorflow/core/tpu/BUILD +++ b/tensorflow/core/tpu/BUILD @@ -1,5 +1,6 @@ # Description: Utilities for TPU Operations +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "if_windows", diff --git a/tensorflow/core/tpu/graph_rewrite/BUILD b/tensorflow/core/tpu/graph_rewrite/BUILD index bffb44c1b97..36c3b6205e1 100644 --- a/tensorflow/core/tpu/graph_rewrite/BUILD +++ b/tensorflow/core/tpu/graph_rewrite/BUILD @@ -1,5 +1,7 @@ # Contains graph rewrites for TPU runtimes and optimizations. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( default_visibility = [ "//tensorflow/core/tpu:__subpackages__", diff --git a/tensorflow/core/tpu/kernels/BUILD b/tensorflow/core/tpu/kernels/BUILD index f35f7151222..d180a6990ca 100644 --- a/tensorflow/core/tpu/kernels/BUILD +++ b/tensorflow/core/tpu/kernels/BUILD @@ -1,5 +1,6 @@ # TPU Kernel Implementations +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow/core/platform:build_config.bzl", "tf_proto_library_cc", diff --git a/tensorflow/core/tpu/kernels/xla/BUILD b/tensorflow/core/tpu/kernels/xla/BUILD index f55583a570b..0c1bfba9c36 100644 --- a/tensorflow/core/tpu/kernels/xla/BUILD +++ b/tensorflow/core/tpu/kernels/xla/BUILD @@ -1,5 +1,7 @@ # XLA Ops for TPUs +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/tpu/ops/BUILD b/tensorflow/core/tpu/ops/BUILD index 5b49f5abc78..a85b599ab31 100644 --- a/tensorflow/core/tpu/ops/BUILD +++ b/tensorflow/core/tpu/ops/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( default_visibility = ["//visibility:public"], licenses = ["notice"], # Apache 2.0 diff --git a/tensorflow/core/util/BUILD b/tensorflow/core/util/BUILD index d7f4464d8b0..ab909078cf0 100644 --- a/tensorflow/core/util/BUILD +++ b/tensorflow/core/util/BUILD @@ -16,7 +16,14 @@ load( "tf_cuda_only_cc_test", "tf_kernel_library", ) + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable") + +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_version_info_genrule") load( "//third_party/mkl:build_defs.bzl", @@ -598,6 +605,7 @@ tf_kernel_library( name = "cuda_solvers", srcs = ["cuda_solvers.cc"], hdrs = ["cuda_solvers.h"], + compatible_with = [], # @local_config_cuda//cuda:cusolver_static, //third_party/eigen3:blas, # and //third_party/libf2c all contain various parts of BLAS, LAPACK, # and f2c helper functions in global namespace. Tell the compiler to @@ -621,6 +629,7 @@ tf_kernel_library( name = "rocm_solvers", srcs = ["rocm_solvers.cc"], hdrs = ["rocm_solvers.h"], + compatible_with = [], visibility = ["//tensorflow/core/kernels:friends"], deps = [ "//tensorflow/core:framework", @@ -639,6 +648,7 @@ tf_kernel_library( name = "cuda_sparse", srcs = if_cuda(["cuda_sparse.cc"]) + if_rocm(["rocm_sparse.cc"]), hdrs = ["cuda_sparse.h"], + compatible_with = [], deps = [ ":cuda_solvers", "//tensorflow/core:framework", diff --git a/tensorflow/core/util/ctc/BUILD b/tensorflow/core/util/ctc/BUILD index aa00a210f79..086a046f5c1 100644 --- a/tensorflow/core/util/ctc/BUILD +++ b/tensorflow/core/util/ctc/BUILD @@ -2,6 +2,8 @@ # is a type of seq2seq loss. The libraries in this directory # implement the CTC loss and a number of CTC decoders. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_tests") package( diff --git a/tensorflow/core/util/proto/BUILD b/tensorflow/core/util/proto/BUILD index a47a64982e7..d971b6383cb 100644 --- a/tensorflow/core/util/proto/BUILD +++ b/tensorflow/core/util/proto/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/util/rpc/BUILD b/tensorflow/core/util/rpc/BUILD index 1cd90283d5c..c1b8869b8d2 100644 --- a/tensorflow/core/util/rpc/BUILD +++ b/tensorflow/core/util/rpc/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") package( diff --git a/tensorflow/core/util/sparse/BUILD b/tensorflow/core/util/sparse/BUILD index 6d0e3d0b4af..4f621a6376d 100644 --- a/tensorflow/core/util/sparse/BUILD +++ b/tensorflow/core/util/sparse/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/core/util/tensor_bundle/BUILD b/tensorflow/core/util/tensor_bundle/BUILD index 36114256216..fa2b3d25e29 100644 --- a/tensorflow/core/util/tensor_bundle/BUILD +++ b/tensorflow/core/util/tensor_bundle/BUILD @@ -1,15 +1,21 @@ # Description: # Tensor bundle: a module to efficiently serialize and deserialize tensors. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", - "cc_header_only_library", "if_not_windows", "if_windows", "tf_cc_test", "tf_copts", ) +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "cc_header_only_library") + +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "filegroup") + package( default_visibility = ["//visibility:public"], licenses = ["notice"], # Apache 2.0 diff --git a/tensorflow/core/util/tensor_bundle/testdata/old_string_tensors/BUILD b/tensorflow/core/util/tensor_bundle/testdata/old_string_tensors/BUILD index 9f7c7c65ba5..b3f1eea2fc4 100644 --- a/tensorflow/core/util/tensor_bundle/testdata/old_string_tensors/BUILD +++ b/tensorflow/core/util/tensor_bundle/testdata/old_string_tensors/BUILD @@ -1,6 +1,8 @@ # Description: # Old string tensors data packages. +load("//tensorflow:tensorflow.bzl", "filegroup") + package( licenses = ["notice"], # Apache 2.0 ) diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index ac6abb25103..2738f976894 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -5,6 +5,9 @@ load("//tensorflow:tensorflow.bzl", "py_strict_library") load("//tensorflow:tensorflow.bzl", "cc_header_only_library", "if_mlir", "if_not_windows", "if_xla_available", "py_test", "py_tests", "tf_cc_shared_object", "tf_cc_test", "tf_cuda_library", "tf_enable_mlir_bridge", "tf_gen_op_wrapper_py") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") + # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_monitoring_python_deps") @@ -416,6 +419,7 @@ cc_library( name = "cost_analyzer_lib", srcs = ["grappler/cost_analyzer.cc"], hdrs = ["grappler/cost_analyzer.h"], + compatible_with = get_compatible_with_cloud(), deps = [ "//tensorflow/core:lib", "//tensorflow/core/grappler/costs:analytical_cost_estimator", @@ -530,6 +534,7 @@ tf_python_pybind_extension( # Necessary for the pywrap inclusion below. tf_pybind_cc_library_wrapper( name = "tfcompile_headers_lib", + compatible_with = [], deps = [ "//tensorflow/compiler/aot:tfcompile_lib", ], @@ -5929,6 +5934,7 @@ tf_cuda_library( name = "tf_session_helper", srcs = ["client/tf_session_helper.cc"], hdrs = ["client/tf_session_helper.h"], + compatible_with = [], deps = [ ":construction_fails_op", ":ndarray_tensor", diff --git a/tensorflow/python/profiler/internal/BUILD b/tensorflow/python/profiler/internal/BUILD index 5eeaf96448d..e471a4b97f6 100644 --- a/tensorflow/python/profiler/internal/BUILD +++ b/tensorflow/python/profiler/internal/BUILD @@ -1,5 +1,9 @@ +# buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "cuda_py_test") +# buildifier: disable=same-origin-load +load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") + # buildifier: disable=same-origin-load load("//tensorflow:tensorflow.bzl", "tf_python_pybind_extension") load("//tensorflow:tensorflow.bzl", "py_test") @@ -137,6 +141,7 @@ cc_library( name = "python_hooks", srcs = ["python_hooks.cc"], hdrs = ["python_hooks.h"], + compatible_with = get_compatible_with_cloud(), copts = ["-fexceptions"], features = ["-use_header_modules"], # Incompatible with -fexceptions. visibility = ["//visibility:public"], diff --git a/tensorflow/stream_executor/BUILD b/tensorflow/stream_executor/BUILD index 22aa60a70a4..33ef97f6712 100644 --- a/tensorflow/stream_executor/BUILD +++ b/tensorflow/stream_executor/BUILD @@ -4,6 +4,8 @@ # Throughout this file, all targets are built with the standard crosstool and # do not link against restricted binary blobs. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "tf_cc_test") load("//tensorflow/core/platform:build_config.bzl", "tf_proto_library") load("//tensorflow/core/platform:build_config_root.bzl", "if_static") diff --git a/tensorflow/stream_executor/cuda/BUILD b/tensorflow/stream_executor/cuda/BUILD index dccdab8877e..ea65d7aee5c 100644 --- a/tensorflow/stream_executor/cuda/BUILD +++ b/tensorflow/stream_executor/cuda/BUILD @@ -2,6 +2,7 @@ # CUDA-platform specific StreamExecutor support code. load("//tensorflow:tensorflow.bzl", "tf_cc_test", "tf_copts", "tf_cuda_cc_test") +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends", @@ -17,6 +18,10 @@ load( "//tensorflow/core/platform:build_config_root.bzl", "tf_cuda_tests_tags", ) +load( + "//tensorflow/core/platform:rules_cc.bzl", + "cc_library", +) package( default_visibility = [":friends"], diff --git a/tensorflow/stream_executor/gpu/BUILD b/tensorflow/stream_executor/gpu/BUILD index 9744fc82593..6328fa44a74 100644 --- a/tensorflow/stream_executor/gpu/BUILD +++ b/tensorflow/stream_executor/gpu/BUILD @@ -1,6 +1,7 @@ # Description: # GPU-platform specific StreamExecutor support code. +load("//tensorflow:tensorflow.bzl", "filegroup") load( "//tensorflow/stream_executor:build_defs.bzl", "if_gpu_is_configured", @@ -9,8 +10,12 @@ load( "//tensorflow/core/platform/default:cuda_build_defs.bzl", "if_cuda_is_configured", ) -load("//tensorflow:tensorflow.bzl", "tf_copts") +load("//tensorflow:tensorflow.bzl", "if_tpu", "tf_copts") load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured") +load( + "//tensorflow/core/platform:rules_cc.bzl", + "cc_library", +) package( default_visibility = [ @@ -65,8 +70,10 @@ cc_library( "//tensorflow/stream_executor:device_options", "//tensorflow/stream_executor/lib", "//tensorflow/stream_executor/platform", - "@local_config_cuda//cuda:cuda_headers", - ], + ] + if_tpu( + if_false = ["@local_config_cuda//cuda:cuda_headers"], + if_true = [], + ), ) cc_library( diff --git a/tensorflow/stream_executor/host/BUILD b/tensorflow/stream_executor/host/BUILD index e0ac5e5367b..a48e1cdf5b0 100644 --- a/tensorflow/stream_executor/host/BUILD +++ b/tensorflow/stream_executor/host/BUILD @@ -1,6 +1,8 @@ # Description: # Host-platform specific StreamExecutor support code. +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends") load("//tensorflow:tensorflow.bzl", "tf_cc_test") diff --git a/tensorflow/stream_executor/lib/BUILD b/tensorflow/stream_executor/lib/BUILD index 76fe0ed94e3..d0f57112471 100644 --- a/tensorflow/stream_executor/lib/BUILD +++ b/tensorflow/stream_executor/lib/BUILD @@ -1,3 +1,5 @@ +load("//tensorflow:tensorflow.bzl", "filegroup") +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow:tensorflow.bzl", "if_windows", "tf_cc_test") load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends") diff --git a/tensorflow/stream_executor/platform/BUILD b/tensorflow/stream_executor/platform/BUILD index 37bec24dfe5..d2172934b1e 100644 --- a/tensorflow/stream_executor/platform/BUILD +++ b/tensorflow/stream_executor/platform/BUILD @@ -1,3 +1,4 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load("//tensorflow/core/platform:build_config.bzl", "tf_platform_deps") load("//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends") diff --git a/tensorflow/stream_executor/platform/default/BUILD b/tensorflow/stream_executor/platform/default/BUILD index 032dc518f1a..ce9731e994f 100644 --- a/tensorflow/stream_executor/platform/default/BUILD +++ b/tensorflow/stream_executor/platform/default/BUILD @@ -1,9 +1,10 @@ +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") +load("//tensorflow:tensorflow.bzl", "tf_copts") + licenses(["notice"]) # Apache 2.0 package(default_visibility = ["//tensorflow/stream_executor:__subpackages__"]) -load("//tensorflow:tensorflow.bzl", "tf_copts") - cc_library( name = "platform", textual_hdrs = ["initialize.h"], @@ -18,8 +19,12 @@ cc_library( "//conditions:default": ["dlopen_checker_stub.cc"], }), hdrs = ["dso_loader.h"], + compatible_with = [], copts = tf_copts(), - tags = ["nobuilder"], + tags = [ + "manual", + "nobuilder", + ], deps = [ "//tensorflow/stream_executor:platform", "//tensorflow/stream_executor/lib", diff --git a/tensorflow/stream_executor/rocm/BUILD b/tensorflow/stream_executor/rocm/BUILD index bd4c45382f8..39e73ae453c 100644 --- a/tensorflow/stream_executor/rocm/BUILD +++ b/tensorflow/stream_executor/rocm/BUILD @@ -1,11 +1,13 @@ # Description: # ROCm-platform specific StreamExecutor support code. +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") load( "//tensorflow/stream_executor:build_defs.bzl", "stream_executor_friends", ) load("//tensorflow:tensorflow.bzl", "tf_copts") +load("//tensorflow:tensorflow.bzl", "filegroup") load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm_is_configured") load("//tensorflow/core/platform:build_config_root.bzl", "if_static") diff --git a/tensorflow/stream_executor/tpu/BUILD b/tensorflow/stream_executor/tpu/BUILD index 83aabf312d9..ea878294687 100644 --- a/tensorflow/stream_executor/tpu/BUILD +++ b/tensorflow/stream_executor/tpu/BUILD @@ -1,5 +1,7 @@ # Description: StreamExecutor Interface for TPUs +load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") + package( default_visibility = ["//tensorflow/core/tpu:__subpackages__"], licenses = ["notice"], # Apache 2.0 @@ -138,8 +140,10 @@ cc_library( ], deps = [ ":c_api_conversions", + ":c_api_decl", ":status_helper", ":tpu_executor_c_api_hdrs", + ":tpu_topology_external", "//tensorflow/core:lib", "//tensorflow/core/tpu:tpu_api", "//tensorflow/stream_executor", diff --git a/tensorflow/tensorflow.bzl b/tensorflow/tensorflow.bzl index a695db365aa..4b03e960d3a 100644 --- a/tensorflow/tensorflow.bzl +++ b/tensorflow/tensorflow.bzl @@ -261,6 +261,13 @@ def if_nccl(if_true, if_false = []): "//conditions:default": if_true, }) +def if_tpu(if_true, if_false = []): + """Shorthand for select()ing whether to build for TPUs.""" + return select({ + str(Label("//tensorflow:with_tpu_support")): if_true, + "//conditions:default": if_false, + }) + # Linux systems may required -lrt linker flag for e.g. clock_gettime # see https://github.com/tensorflow/tensorflow/issues/15129 def lrt_if_needed(): @@ -321,6 +328,7 @@ def tf_copts( (if_not_windows(["-fno-exceptions"]) if not allow_exceptions else []) + if_cuda(["-DGOOGLE_CUDA=1"]) + if_nvcc(["-DTENSORFLOW_USE_NVCC=1"]) + + if_tpu(["-DLIBTFTPU"]) + if_xla_available(["-DTENSORFLOW_USE_XLA=1"]) + if_tensorrt(["-DGOOGLE_TENSORRT=1"]) + if_mkl(["-DINTEL_MKL=1", "-DENABLE_MKLDNN_V1", "-DENABLE_INTEL_MKL_BFLOAT16"]) + @@ -1478,6 +1486,7 @@ def tf_kernel_library( copts = None, gpu_copts = None, is_external = False, + compatible_with = None, **kwargs): """A rule to build a TensorFlow OpKernel. @@ -1565,6 +1574,7 @@ def tf_kernel_library( linkstatic = 1, # Needed since alwayslink is broken in bazel b/27630669 alwayslink = alwayslink, deps = deps, + compatible_with = compatible_with, **kwargs ) @@ -2775,7 +2785,7 @@ def tf_python_pybind_extension( testonly = testonly, ) -def tf_pybind_cc_library_wrapper(name, deps, visibility = None): +def tf_pybind_cc_library_wrapper(name, deps, visibility = None, **kwargs): """Wrapper for cc_library and proto dependencies used by tf_python_pybind_extension. This wrapper ensures that cc libraries' and protos' headers are made @@ -2783,7 +2793,7 @@ def tf_pybind_cc_library_wrapper(name, deps, visibility = None): linked case. The symbols in these deps symbols should be linked to, and exported by, the core pywrap_tensorflow_internal.so """ - cc_header_only_library(name = name, deps = deps, visibility = visibility) + cc_header_only_library(name = name, deps = deps, visibility = visibility, **kwargs) def if_cuda_or_rocm(if_true, if_false = []): """Shorthand for select()'ing whether to build for either CUDA or ROCm. @@ -2863,13 +2873,6 @@ def tf_enable_mlir_bridge(): "//conditions:default": [], }) -def if_tpu(if_true, if_false = []): - """Shorthand for select()ing whether to build for TPUs.""" - return select({ - str(Label("//tensorflow:with_tpu_support")): if_true, - "//conditions:default": if_false, - }) - def tfcompile_target_cpu(): return "" @@ -2918,3 +2921,9 @@ def filegroup(**kwargs): def genrule(**kwargs): native.genrule(**kwargs) + +def internal_hlo_deps(): + return [] + +def internal_tfrt_deps(): + return [] diff --git a/tensorflow/tools/graph_transforms/BUILD b/tensorflow/tools/graph_transforms/BUILD index b900867a2d3..12dfe038e8b 100644 --- a/tensorflow/tools/graph_transforms/BUILD +++ b/tensorflow/tools/graph_transforms/BUILD @@ -9,6 +9,7 @@ load( "tf_copts", "tf_py_test", ) +load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") package( default_visibility = ["//visibility:public"], @@ -25,6 +26,7 @@ cc_library( hdrs = [ "transform_utils.h", ], + compatible_with = get_compatible_with_cloud(), copts = tf_copts(), visibility = ["//visibility:public"], deps = [ diff --git a/tensorflow/tools/optimization/BUILD b/tensorflow/tools/optimization/BUILD index 73a8a15b121..14db4f77232 100644 --- a/tensorflow/tools/optimization/BUILD +++ b/tensorflow/tools/optimization/BUILD @@ -6,6 +6,7 @@ load( "tf_cc_binary", "tf_cuda_library", ) +load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") package( default_visibility = ["//visibility:public"], @@ -18,6 +19,7 @@ tf_cuda_library( name = "optimization_pass_runner_lib", srcs = ["optimization_pass_runner.cc"], hdrs = ["optimization_pass_runner.h"], + compatible_with = get_compatible_with_cloud(), deps = [ "//tensorflow/core:core_cpu", "//tensorflow/core:core_cpu_base",