diff --git a/configure.py b/configure.py index 9ca614f8f9b..9da49b628d3 100644 --- a/configure.py +++ b/configure.py @@ -991,6 +991,8 @@ def main(): 'with_gcp_support', False, 'gcp') set_build_var(environ_cp, 'TF_NEED_HDFS', 'Hadoop File System', 'with_hdfs_support', False, 'hdfs') + set_build_var(environ_cp, 'TF_NEED_S3', 'Amazon S3 File System', + 'with_s3_support', True, 's3') set_build_var(environ_cp, 'TF_ENABLE_XLA', 'XLA JIT', 'with_xla_support', False, 'xla') set_build_var(environ_cp, 'TF_NEED_GDR', 'GDR', 'with_gdr_support', diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 56d09390232..1620bb5f2a0 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -185,6 +185,12 @@ config_setting( visibility = ["//visibility:public"], ) +config_setting( + name = "with_s3_support", + values = {"define": "with_s3_support=true"}, + visibility = ["//visibility:public"], +) + config_setting( name = "with_xla_support", values = {"define": "with_xla_support=true"}, diff --git a/tensorflow/core/platform/default/build_config.bzl b/tensorflow/core/platform/default/build_config.bzl index 51d37291eed..2c14ea917c0 100644 --- a/tensorflow/core/platform/default/build_config.bzl +++ b/tensorflow/core/platform/default/build_config.bzl @@ -465,6 +465,11 @@ def tf_additional_core_deps(): "//tensorflow/core/platform/hadoop:hadoop_file_system", ], "//conditions:default": [], + }) + select({ + "//tensorflow:with_s3_support": [ + "//tensorflow/contrib/s3:s3_file_system", + ], + "//conditions:default": [], }) # TODO(jart, jhseu): Delete when GCP is default on.