Skip to content

Commit 0bb846d

Browse files
Regina8023meta-codesync[bot]
authored andcommitted
Fix ncclBroadcast ctran algo selection
Summary: If algo!=orig, we should use `ctranBroadcast` Differential Revision: D85484348 fbshipit-source-id: 2dfb09a19a97c326800f0c9cb814e4a5f338bf84
1 parent 7b6be08 commit 0bb846d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comms/ncclx/v2_27/src/collectives.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ ncclResult_t ncclBroadcast(const void* sendbuff, void* recvbuff, size_t count, n
143143
}
144144

145145
SetCudaDevRAII setCudaDev(comm->cudaDev);
146-
if ((NCCL_BROADCAST_ALGO == NCCL_BROADCAST_ALGO::ctran) &&
146+
if ((NCCL_BROADCAST_ALGO != NCCL_BROADCAST_ALGO::orig) &&
147147
ctranBroadcastSupport(comm->ctranComm_.get())) {
148148
return metaCommToNccl(ctranBroadcast(
149149
sendbuff, recvbuff, count, ncclToMetaComm(datatype), root, comm->ctranComm_.get(), stream));

0 commit comments

Comments
 (0)