diff options
Diffstat (limited to 'tensorflow/core/BUILD')
-rw-r--r-- | tensorflow/core/BUILD | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 0fc610a76e..324183c053 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -84,12 +84,14 @@ load( "//tensorflow/core:platform/default/build_config.bzl", "tf_proto_library", "tf_proto_library_cc", + "tf_additional_core_deps", + "tf_additional_lib_defines", + "tf_additional_lib_deps", "tf_additional_lib_hdrs", "tf_additional_lib_srcs", "tf_additional_minimal_lib_srcs", "tf_additional_proto_hdrs", "tf_additional_proto_srcs", - "tf_additional_lib_deps", "tf_additional_stream_executor_srcs", "tf_additional_cupti_wrapper_deps", "tf_additional_libdevice_data", @@ -1126,12 +1128,13 @@ cc_library( "platform/tracing.h", ], copts = tf_copts(), + defines = tf_additional_lib_defines(), linkopts = ["-ldl"], - deps = [ + deps = tf_additional_lib_deps() + [ ":lib_proto_parsing", ":protos_all_cc", - "//tensorflow/core/platform/default/build_config:platformlib", "//third_party/eigen3", + "//tensorflow/core/platform/default/build_config:platformlib", "@zlib_archive//:zlib", ], ) @@ -1351,7 +1354,7 @@ tf_cuda_library( ":protos_all_cc", "//third_party/eigen3", "//tensorflow/core/kernels:required", - ] + tf_additional_lib_deps(), + ] + tf_additional_core_deps(), alwayslink = 1, ) |