diff --git a/projects/rccl/src/transport/net.cc b/projects/rccl/src/transport/net.cc index 87fc9ce2ff..928a6a924f 100644 --- a/projects/rccl/src/transport/net.cc +++ b/projects/rccl/src/transport/net.cc @@ -378,7 +378,7 @@ ncclResult_t netRecvProxy(struct ncclProxyArgs* args) { if (done) { args->head += args->sliceSteps; if (args->protocol == NCCL_PROTO_SIMPLE) { - if (resources->useGdr) ncclNetFlush(resources->netRecvComm, localBuff+buffSlot*stepSize, size, mhandle); + if (resources->useGdr) NCCLCHECK(ncclNetFlush(resources->netRecvComm, localBuff+buffSlot*stepSize, size, mhandle)); resources->hostRecvMem->tail = args->head; } args->idle = 0;