From 4ec992fab7d90691ad43bee6f7f6c34aa219934c Mon Sep 17 00:00:00 2001 From: Sylvain Jeaugey Date: Wed, 22 Sep 2021 05:10:09 -0700 Subject: [PATCH] Fix Collnet when GDR is disabled --- src/transport/coll_net.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transport/coll_net.cc b/src/transport/coll_net.cc index 49398f9ffc..4c0e76d5bc 100644 --- a/src/transport/coll_net.cc +++ b/src/transport/coll_net.cc @@ -507,7 +507,7 @@ ncclResult_t collNetRecvProxy(struct ncclProxyArgs* args) { int sharedBuffSlot = sub->transmitted%NCCL_STEPS; int startChannel = group*COLLNET_GROUP_NSUBS; char* groupRecvAddress; - NCCLCHECK(ncclProxySharedBuffersGetCollNet(sub->connector->comm, 1, 1, sharedBuffSlot, startChannel, &groupRecvAddress)); + NCCLCHECK(ncclProxySharedBuffersGetCollNet(sub->connector->comm, p == NCCL_PROTO_SIMPLE ? resources->useGdr : 0, 1, sharedBuffSlot, startChannel, &groupRecvAddress)); char* ptr = groupRecvAddress + (s%COLLNET_GROUP_NSUBS)*args->sharedSize[sharedBuffSlot]; if (p == NCCL_PROTO_SIMPLE) { volatile void** ptrsFifo = (volatile void**)resources->recvMem->ptrsFifo;