experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use //tensorflow:with_numa_support instead of enumerating all non-numa OSs.

Browse Source 
PiperOrigin-RevId: 289919322
Change-Id: I7b82f78724904445f69dbf4fdb13456db6dcb976
This commit is contained in:
A. Unique TensorFlower 2020-01-15 12:59:47 -08:00 committed by TensorFlower Gardener
parent 86fa42f516
commit 37f0ac13bd
3 changed files with 27 additions and 19 deletions
tensorflow
core/platform/default
BUILD
tools
lib_package
BUILD
pip_package
BUILD

8
tensorflow/core/platform/default/BUILD
View File

@ -287,12 +287,8 @@ cc_library(
"@snappy",
] + select({
# TF Additional NUMA dependencies
"//tensorflow:android": [],
"//tensorflow:ios": [],
"//tensorflow:macos": [],
"//conditions:default": [
"@hwloc",
],
"//tensorflow:with_numa_support": ["//third_party/hwloc"],
"//conditions:default": [],
}),
)

26
tensorflow/tools/lib_package/BUILD
View File

@ -141,6 +141,7 @@ genrule(
"//third_party/hadoop:LICENSE.txt",
"//third_party/icu/data:LICENSE",
"@boringssl//:LICENSE",
"@com_google_protobuf//:LICENSE",
"@com_googlesource_code_re2//:LICENSE",
"@curl//:COPYING",
"@double_conversion//:LICENSE",
@ -150,22 +151,20 @@ genrule(
"@gemmlowp//:LICENSE",
"@gif//:COPYING",
"@highwayhash//:LICENSE",
"@hwloc//:COPYING",
"@icu//:icu4c/LICENSE",
"@libjpeg_turbo//:LICENSE.md",
"@lmdb//:LICENSE",
"@llvm-project//llvm:LICENSE.TXT",
"@llvm-project//mlir:LICENSE.TXT",
"@lmdb//:LICENSE",
"@local_config_sycl//sycl:LICENSE.text",
"@local_config_tensorrt//:LICENSE",
"@nasm//:LICENSE",
"@nsync//:LICENSE",
"@png//:LICENSE",
"@com_google_protobuf//:LICENSE",
"@six_archive//:LICENSE",
"@snappy//:COPYING",
"@sobol_data//:LICENSE",
"@zlib_archive//:zlib.h",
"@six_archive//:LICENSE",
] + select({
"//tensorflow:android": [],
"//tensorflow:ios": [],
@ -189,6 +188,11 @@ genrule(
"@libxsmm_archive//:LICENSE.md",
],
"//conditions:default": [],
}) + select({
"//tensorflow:with_numa_support": [
"//third_party/hwloc:COPYING",
],
"//conditions:default": [],
}) + if_cuda([
"@cub_archive//:LICENSE.TXT",
"@local_config_nccl//:LICENSE",
@ -215,6 +219,7 @@ genrule(
"//third_party/hadoop:LICENSE.txt",
"//third_party/icu/data:LICENSE",
"@boringssl//:LICENSE",
"@com_google_protobuf//:LICENSE",
"@com_googlesource_code_re2//:LICENSE",
"@curl//:COPYING",
"@double_conversion//:LICENSE",
@ -223,8 +228,9 @@ genrule(
"@fft2d//:fft2d/readme2d.txt",
"@gemmlowp//:LICENSE",
"@gif//:COPYING",
"@grpc//:LICENSE",
"@grpc//third_party/address_sorting:LICENSE",
"@highwayhash//:LICENSE",
"@hwloc//:COPYING",
"@icu//:icu4j/main/shared/licenses/LICENSE",
"@libjpeg_turbo//:LICENSE.md",
"@llvm-project//llvm:LICENSE.TXT",
@ -235,13 +241,10 @@ genrule(
"@nasm//:LICENSE",
"@nsync//:LICENSE",
"@png//:LICENSE",
"@com_google_protobuf//:LICENSE",
"@six_archive//:LICENSE",
"@snappy//:COPYING",
"@sobol_data//:LICENSE",
"@zlib_archive//:zlib.h",
"@grpc//:LICENSE",
"@grpc//third_party/address_sorting:LICENSE",
"@six_archive//:LICENSE",
] + select({
"//tensorflow:android": [],
"//tensorflow:ios": [],
@ -265,6 +268,11 @@ genrule(
"@libxsmm_archive//:LICENSE.md",
],
"//conditions:default": [],
}) + select({
"//tensorflow:with_numa_support": [
"//third_party/hwloc:COPYING",
],
"//conditions:default": [],
}) + if_cuda([
"@cub_archive//:LICENSE.TXT",
"@local_config_nccl//:LICENSE",

12
tensorflow/tools/pip_package/BUILD
View File

@ -131,6 +131,7 @@ filegroup(
"@astor_archive//:LICENSE",
"@boringssl//:LICENSE",
"@com_google_absl//:LICENSE",
"@com_google_protobuf//:LICENSE",
"@com_googlesource_code_re2//:LICENSE",
"@curl//:COPYING",
"@double_conversion//:LICENSE",
@ -144,29 +145,27 @@ filegroup(
"@gemmlowp//:LICENSE",
"@gif//:COPYING",
"@highwayhash//:LICENSE",
"@hwloc//:COPYING",
"@icu//:icu4c/LICENSE",
"@kissfft//:COPYING",
"@libjpeg_turbo//:LICENSE.md",
"@lmdb//:LICENSE",
"@llvm-project//llvm:LICENSE.TXT",
"@llvm-project//mlir:LICENSE.TXT",
"@lmdb//:LICENSE",
"@local_config_sycl//sycl:LICENSE.text",
"@local_config_tensorrt//:LICENSE",
"@nasm//:LICENSE",
"@nsync//:LICENSE",
"@opt_einsum_archive//:LICENSE",
"@org_python_pypi_backports_weakref//:LICENSE",
"@pasta//:LICENSE",
"@pcre//:LICENCE",
"@png//:LICENSE",
"@com_google_protobuf//:LICENSE",
"@six_archive//:LICENSE",
"@snappy//:COPYING",
"@sobol_data//:LICENSE",
"@swig//:LICENSE",
"@termcolor_archive//:COPYING.txt",
"@zlib_archive//:zlib.h",
"@org_python_pypi_backports_weakref//:LICENSE",
] + select({
"//tensorflow:android": [],
"//tensorflow:ios": [],
@ -190,6 +189,11 @@ filegroup(
"@libxsmm_archive//:LICENSE.md",
],
"//conditions:default": [],
}) + select({
"//tensorflow:with_numa_support": [
"//third_party/hwloc:COPYING",
],
"//conditions:default": [],
}) + if_cuda([
"@cub_archive//:LICENSE.TXT",
"@local_config_nccl//:LICENSE",