From d64011394d804344675652dace19f4e8c8d22d06 Mon Sep 17 00:00:00 2001
From: HeKa <mofhejia@163.com>
Date: Fri, 3 Jan 2025 22:40:11 +0800
Subject: [PATCH] rules_nccl@0.0.1 (#3525)

A simple module for compiling and linking local installed NCCL library.

Release: https://github.com/MoFHeka/rules_nccl/releases/tag/v0.0.1

Author: @MoFHeka
---
 modules/rules_nccl/0.0.1/MODULE.bazel  | 26 ++++++++++++++++++++++++++
 modules/rules_nccl/0.0.1/presubmit.yml | 10 ++++++++++
 modules/rules_nccl/0.0.1/source.json   |  5 +++++
 modules/rules_nccl/metadata.json       | 17 +++++++++++++++++
 4 files changed, 58 insertions(+)
 create mode 100644 modules/rules_nccl/0.0.1/MODULE.bazel
 create mode 100644 modules/rules_nccl/0.0.1/presubmit.yml
 create mode 100644 modules/rules_nccl/0.0.1/source.json
 create mode 100644 modules/rules_nccl/metadata.json

diff --git a/modules/rules_nccl/0.0.1/MODULE.bazel b/modules/rules_nccl/0.0.1/MODULE.bazel
new file mode 100644
index 00000000000..076d100f0de
--- /dev/null
+++ b/modules/rules_nccl/0.0.1/MODULE.bazel
@@ -0,0 +1,26 @@
+"""
+A rule for import nccl library
+"""
+
+module(
+    name = "rules_nccl",
+    version = "0.0.1",
+    compatibility_level = 1,
+)
+
+nccl = use_extension("@rules_nccl//nccl:extensions.bzl", "nccl_dependencie")
+nccl.nccl_dependencie(
+    name = "local_nccl",
+    nccl_path = "",
+)
+use_repo(nccl, "local_nccl")
+
+bazel_dep(name = "rules_cuda", version = "0.2.1", dev_dependency = True)
+cuda = use_extension("@rules_cuda//cuda:extensions.bzl", "toolchain", dev_dependency = True)
+cuda.local_toolchain(
+    name = "local_cuda",
+)
+use_repo(cuda, "local_cuda")
+
+bazel_dep(name = "bazel_skylib", version = "1.7.1")
+bazel_dep(name = "platforms", version = "0.0.10")
diff --git a/modules/rules_nccl/0.0.1/presubmit.yml b/modules/rules_nccl/0.0.1/presubmit.yml
new file mode 100644
index 00000000000..5ff5ab58b65
--- /dev/null
+++ b/modules/rules_nccl/0.0.1/presubmit.yml
@@ -0,0 +1,10 @@
+matrix:
+  bazel:
+    - 6.x
+    - 7.x
+    - 8.x
+tasks:
+  verify_targets_linux:
+    name: Verify build targets
+    bazel: ${{ bazel }}
+    platform: ubuntu2004
diff --git a/modules/rules_nccl/0.0.1/source.json b/modules/rules_nccl/0.0.1/source.json
new file mode 100644
index 00000000000..8d6086abf6e
--- /dev/null
+++ b/modules/rules_nccl/0.0.1/source.json
@@ -0,0 +1,5 @@
+{
+    "integrity": "sha256-sReOaNA1jhcy4tPGLQgYU/IEi8diZ7dBJXm7zZaRN14=",
+    "strip_prefix": "",
+    "url": "https://github.com/MoFHeka/rules_nccl/releases/download/v0.0.1/rules_nccl-v0.0.1.tar.gz"
+}
\ No newline at end of file
diff --git a/modules/rules_nccl/metadata.json b/modules/rules_nccl/metadata.json
new file mode 100644
index 00000000000..58d70eecef2
--- /dev/null
+++ b/modules/rules_nccl/metadata.json
@@ -0,0 +1,17 @@
+{
+  "homepage": "https://github.com/MoFHeka/rules_nccl",
+  "maintainers": [
+    {
+      "email": "mofhejia@163.com",
+      "github": "MoFHeka",
+      "name": "Jia He"
+    }
+  ],
+  "repository": [
+    "github:MoFHeka/rules_nccl"
+  ],
+  "versions": [
+    "0.0.1"
+  ],
+  "yanked_versions": {}
+}
\ No newline at end of file