Skip to content

Commit 9d4cea6

Browse files
[NCCL] Fix ras build issue in OSS.
PiperOrigin-RevId: 726180935
1 parent a933a2d commit 9d4cea6

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

third_party/nccl/archive.BUILD

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,10 @@ cc_library(
192192
"src/transport/coll_net.cc",
193193
"src/transport/net.cc",
194194
"src/enqueue.cc",
195+
# RAS is a health-checking system (starting from NCCL 2.24:
196+
# https://docs.nvidia.com/deeplearning/nccl/user-guide/docs/troubleshooting/ras.html
197+
# It's not reqired for NCCL to work.
198+
"src/ras/client.cc",
195199
],
196200
) + [
197201
# Required for header inclusion checking (see
@@ -200,6 +204,7 @@ cc_library(
200204
# from the virtual includes directory.
201205
"src/include/collectives.h",
202206
"src/nccl.h",
207+
"src/ras/ras_internal.h",
203208
],
204209
hdrs = ["src/nccl.h"],
205210
include_prefix = "third_party/nccl",

0 commit comments

Comments
 (0)