diff --git a/src/transport/net.cc b/src/transport/net.cc index 2c349fae10..23e8ebce3d 100644 --- a/src/transport/net.cc +++ b/src/transport/net.cc @@ -1481,6 +1481,8 @@ static ncclResult_t sendProxyProgress(struct ncclProxyState* proxyState, struct TRACE(NCCL_NET, "sendProxy [%ld/%d/%d] request %p done", sub->done, buffSlot, sub->nsteps, sub->requests[buffSlot]); sub->done += args->sliceSteps; ncclProfilerStopProxyStepEvent(s, args, doneStepId); + facebook_rccl::updateProxyOpCounter(proxyState->proxyTrace, sub->traceKey, + facebook_rccl::ProxyCounterTypes::DONE, sub->done); if (resources->shared == 0) { volatile uint64_t* sendHead = resources->gdcSync ? resources->gdcSync : &resources->sendMem->head; *sendHead = sub->base + sub->done; @@ -1823,6 +1825,7 @@ static ncclResult_t recvProxyProgress(struct ncclProxyState* proxyState, struct int doneStepId = sub->done; sub->done += args->sliceSteps; ncclProfilerStopProxyStepEvent(s+i, args, doneStepId); + facebook_rccl::updateProxyOpCounter(proxyState->proxyTrace, sub->traceKey, facebook_rccl::ProxyCounterTypes::DONE, sub->done); args->idle = 0; if (sub->done == sub->nsteps) { args->done++;