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:
parent
0ac688a18c
commit
7e4e336ce5
@ -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',
|
||||
|
@ -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"},
|
||||
|
@ -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.
|
||||
|
Loading…
Reference in New Issue
Block a user