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" """,
+)