Expand the glob for tf_additional_lib_hdrs(). This makes the set of platform-specific headers explicit, and also helps make checking in BUILD files in subdirectories of tf/core/platform easier.
This change is part of the Tensorflow Build Improvements RFC: https://github.com/tensorflow/community/pull/179
PiperOrigin-RevId: 284392848
Change-Id: I635d2051bc064d88dbdd8fb146636887adc19dbb
diff --git a/tensorflow/core/platform/default/build_config.bzl b/tensorflow/core/platform/default/build_config.bzl
index fd2d5b9..4e8f88f 100644
--- a/tensorflow/core/platform/default/build_config.bzl
+++ b/tensorflow/core/platform/default/build_config.bzl
@@ -552,21 +552,33 @@
"//conditions:default": native.glob(posix_set),
})
-def tf_additional_lib_hdrs(exclude = []):
- windows_hdrs = native.glob([
- "default/*.h",
- "windows/*.h",
- "posix/error.h",
- ], exclude = exclude + [
- "default/subprocess.h",
- "default/posix_file_system.h",
- ])
- return select({
- clean_dep("//tensorflow:windows"): windows_hdrs,
- "//conditions:default": native.glob([
- "default/*.h",
- "posix/*.h",
- ], exclude = exclude),
+def tf_additional_lib_hdrs():
+ return [
+ "//tensorflow/core/platform:default/context.h",
+ "//tensorflow/core/platform:default/cord.h",
+ "//tensorflow/core/platform:default/dynamic_annotations.h",
+ "//tensorflow/core/platform:default/integral_types.h",
+ "//tensorflow/core/platform:default/logging.h",
+ "//tensorflow/core/platform:default/mutex.h",
+ "//tensorflow/core/platform:default/mutex_data.h",
+ "//tensorflow/core/platform:default/notification.h",
+ "//tensorflow/core/platform:default/stacktrace.h",
+ "//tensorflow/core/platform:default/strong_hash.h",
+ "//tensorflow/core/platform:default/test_benchmark.h",
+ "//tensorflow/core/platform:default/tracing_impl.h",
+ "//tensorflow/core/platform:default/unbounded_work_queue.h",
+ ] + select({
+ "//tensorflow:windows": [
+ "//tensorflow/core/platform:windows/intrinsics_port.h",
+ "//tensorflow/core/platform:windows/stacktrace.h",
+ "//tensorflow/core/platform:windows/subprocess.h",
+ "//tensorflow/core/platform:windows/wide_char.h",
+ "//tensorflow/core/platform:windows/windows_file_system.h",
+ ],
+ "//conditions:default": [
+ "//tensorflow/core/platform:default/posix_file_system.h",
+ "//tensorflow/core/platform:default/subprocess.h",
+ ],
})
def tf_additional_monitoring_hdrs():