diff options
author | 2017-03-01 17:59:22 -0800 | |
---|---|---|
committer | 2017-03-01 18:08:24 -0800 | |
commit | 3e975ea978bac4d861bb09328b06f3c316212611 (patch) | |
tree | 79bac044c9723df8443495eb962c2dd98a2ed421 /third_party/nccl.BUILD | |
parent | 8043a27ed77f59bb68409070f2bfa01df0e04b89 (diff) |
Merge changes from github.
Change: 148954491
Diffstat (limited to 'third_party/nccl.BUILD')
-rw-r--r-- | third_party/nccl.BUILD | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/third_party/nccl.BUILD b/third_party/nccl.BUILD index bb460a05e0..06b9b8ff68 100644 --- a/third_party/nccl.BUILD +++ b/third_party/nccl.BUILD @@ -43,6 +43,24 @@ cc_library( "-Iexternal/nccl_archive/src", "-O3", ] + cuda_default_copts(), + linkopts = select({ + "@%ws%//tensorflow:android": [ + "-pie", + ], + "@%ws%//tensorflow:darwin": [ + "-Wl,-framework", + "-Wl,CoreFoundation", + "-Wl,-framework", + "-Wl,Security", + ], + "@%ws%//tensorflow:ios": [], + "@%ws%//tensorflow:windows": [ + "ws2_32.lib", + ], + "//conditions:default": [ + "-lrt", + ], + }), visibility = ["//visibility:public"], deps = ["@local_config_cuda//cuda:cuda_headers"], ) |