add back missing proxy-counter updates (#2052)

[ROCm/rccl commit: da183596cd]
이 커밋은 다음에 포함됨:
isaki001
2025-11-25 15:22:34 -06:00
커밋한 사람 GitHub
부모 566671910a
커밋 cf11e2f39f
+3
파일 보기
@@ -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++;