diff options
| author |  avijit-nervana <avijit.chakraborty@intel.com> | 2018-09-14 09:21:08 -0700 |
|---|---|---|
| committer | avijit-nervana <avijit.chakraborty@intel.com> | 2018-09-14 09:21:08 -0700 |
| commit | 41aaed7751690b0b3137dad2620656a698b3ceae (patch) | |
| tree | 00fc1a7f6be0c3968f3e674a65ca4907110ddf2d /third_party/toolchains/preconfig/ubuntu14.04/nccl2/BUILD | |
| parent | c26c5e1217944448f1f4c2b97626fc4d7d6406d3 (diff) | |
| parent | 95338704198205c1bdec1e344e103f1daf05df68 (diff) | |
Merge branch 'master' into avijit/add-cpu-backend
Diffstat (limited to 'third_party/toolchains/preconfig/ubuntu14.04/nccl2/BUILD')
| -rwxr-xr-x | third_party/toolchains/preconfig/ubuntu14.04/nccl2/BUILD | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/third_party/toolchains/preconfig/ubuntu14.04/nccl2/BUILD b/third_party/toolchains/preconfig/ubuntu14.04/nccl2/BUILD new file mode 100755 index 0000000000..96ed60d3cf --- /dev/null +++ b/third_party/toolchains/preconfig/ubuntu14.04/nccl2/BUILD @@ -0,0 +1,25 @@ +filegroup( + name = "LICENSE", + visibility = ["//visibility:public"], +) + +cc_library( + name = "nccl", + srcs = ["libnccl.so.2"], + hdrs = ["nccl.h"], + include_prefix = "third_party/nccl", + visibility = ["//visibility:public"], + deps = [ + "@local_config_cuda//cuda:cuda_headers", + ], +) + +genrule( + name = "nccl-files", + outs = [ + "libnccl.so.2", + "nccl.h", + ], + cmd = """cp "/usr/include/nccl.h" "$(@D)/nccl.h" && + cp "/usr/lib/libnccl.so.2" "$(@D)/libnccl.so.2" """, +) |