Relanding change to add config to enable S3 file system support.

Pass --config=s3 argument to Bazel to build with S3 file system support.
Change was originally rolled back due to a failure it caused in
//tensorflow/core/kernels:control_flow_ops_test on Macs which is now fixed.

PiperOrigin-RevId: 171579378
This commit is contained in:
Michael Case 2017-10-09 13:31:15 -07:00 committed by TensorFlower Gardener
parent 0ac688a18c
commit 7e4e336ce5
3 changed files with 13 additions and 0 deletions

View File

@ -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',

View File

@ -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"},

View File

@ -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.