Update

lshmouse · Sep 24, 2024 · 5e5f04e · 5e5f04e
1 parent e1fe368
commit 5e5f04e
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/third_party/nccl/BUILD b/third_party/nccl/BUILD
@@ -0,0 +1 @@
+package(default_visibility = ["//visibility:public"])
diff --git a/third_party/nccl/README.md b/third_party/nccl/README.md
@@ -0,0 +1,2 @@
+## NCCL
+See: https://github.com/bazel-contrib/rules_cuda/blob/main/examples/WORKSPACE.bazel
diff --git a/third_party/nccl/nccl.BUILD b/third_party/nccl/nccl.BUILD
diff --git a/third_party/nccl/workspace.bzl b/third_party/nccl/workspace.bzl
@@ -0,0 +1,15 @@
+"""Loads the nccl library"""
+
+load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+
+def clean_dep(dep):
+    return str(Label(dep))
+
+def repo():
+    http_archive(
+        name = "nccl",
+        build_file = "@rules_cuda_examples//nccl:nccl.BUILD",
+        sha256 = "83b299cfc2dfe63887dadf3590b3ac2b8b2fd68ec5515b6878774eda39a697d2",
+        strip_prefix = "nccl-9814c75eea18fc7374cde884592233b6b7dc055b",
+        urls = ["https://github.com/nvidia/nccl/archive/9814c75eea18fc7374cde884592233b6b7dc055b.tar.gz"],
+    )
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		package(default_visibility = ["//visibility:public"])
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		## NCCL
		See: https://github.com/bazel-contrib/rules_cuda/blob/main/examples/WORKSPACE.bazel