diff options
Diffstat (limited to 'third_party/toolchains/preconfig/ubuntu14.04/nccl2')
-rwxr-xr-x | third_party/toolchains/preconfig/ubuntu14.04/nccl2/BUILD | 25 | ||||
-rw-r--r-- | third_party/toolchains/preconfig/ubuntu14.04/nccl2/WORKSPACE | 2 |
2 files changed, 27 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" """, +) diff --git a/third_party/toolchains/preconfig/ubuntu14.04/nccl2/WORKSPACE b/third_party/toolchains/preconfig/ubuntu14.04/nccl2/WORKSPACE new file mode 100644 index 0000000000..1e6662ac91 --- /dev/null +++ b/third_party/toolchains/preconfig/ubuntu14.04/nccl2/WORKSPACE @@ -0,0 +1,2 @@ +# DO NOT EDIT: automatically generated WORKSPACE file for nccl_configure rule +workspace(name = "local_config_nccl") |