Skip to content

Commit 3404d8a

Browse files
get9facebook-github-bot
authored andcommitted
antlir: Add nccl as a distro dep
Summary: Ditto for nccl, which torch depends on (even though for inference we shouldn't need this?) Test Plan: CI Rollback Plan: Reviewed By: lisroach Differential Revision: D77917160 fbshipit-source-id: aa3c2b0668710eff73ce4c76a25ed328a33645bf
1 parent 6acd4f0 commit 3404d8a

2 files changed

Lines changed: 23 additions & 1 deletion

File tree

antlir/distro/deps/nccl/BUCK

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
load("//antlir/distro/deps:rpm_library.bzl", "rpm_library")
2+
3+
oncall("antlir")
4+
5+
# Only supporting cudnn9 for now.
6+
7+
rpm_library(
8+
name = "nccl",
9+
dnf_additional_repos = ["nvidia"],
10+
header_glob = [
11+
("/usr/include", "nccl.h"),
12+
],
13+
lib = "/usr/lib64/libnccl.so",
14+
rpm = select({
15+
"DEFAULT": "libnccl-devel-2.23.4-1+cuda12.6.x86_64",
16+
"ovr_config//third-party/nccl/constraints:2.18.3": "libnccl-devel-2.18.3-1+cuda12.2.x86_64",
17+
"ovr_config//third-party/nccl/constraints:2.20.3": "libnccl-devel-2.20.3-1+cuda12.3.x86_64",
18+
"ovr_config//third-party/nccl/constraints:2.23.4": "libnccl-devel-2.23.4-1+cuda12.6.x86_64",
19+
"ovr_config//third-party/nccl/constraints:2.25.1": "libnccl-devel-2.25.1-1+cuda12.8.x86_64",
20+
}),
21+
)

antlir/distro/deps/projects.bzl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# @generated SignedSource<<e121f5fdc8fe5b8107913479bf205a6c>>
1+
# @generated SignedSource<<727982a2a02f75e5fb8fbea0f23a06e2>>
22
PROJECTS = [
33
("abseil", "abseil"),
44
("binutils", "iberty"),
@@ -92,6 +92,7 @@ PROJECTS = [
9292
("llvm-fb", "clang_rt.ubsan_standalone"),
9393
("llvm-fb", "clang_rt.ubsan_standalone_cxx"),
9494
("lzo", "lzo2"),
95+
("nccl", "nccl"),
9596
("ncurses", "ncursesw"),
9697
("ncurses", "panelw"),
9798
("openjpeg", "openjpeg"),

0 commit comments

Comments
 (0)