diff options
Diffstat (limited to 'tensorflow/core/platform/default/build_config.bzl')
-rw-r--r-- | tensorflow/core/platform/default/build_config.bzl | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/core/platform/default/build_config.bzl b/tensorflow/core/platform/default/build_config.bzl index ffb38a169f..0af38affd5 100644 --- a/tensorflow/core/platform/default/build_config.bzl +++ b/tensorflow/core/platform/default/build_config.bzl @@ -293,6 +293,12 @@ def tf_additional_mpi_lib_defines(): "//conditions:default": [], }) +def tf_additional_gdr_lib_defines(): + return select({ + "//tensorflow:with_gdr_support": ["TENSORFLOW_USE_GDR"], + "//conditions:default": [], + }) + def tf_pyclif_proto_library(name, proto_lib, proto_srcfile="", visibility=None, **kwargs): pass |