diff --git a/src/transport/net.cc b/src/transport/net.cc index 87fc9ce..928a6a9 100644 --- a/src/transport/net.cc +++ b/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;