diff --git a/src/enqueue.cc b/src/enqueue.cc index 1e7384a8c..f2a1e5709 100644 --- a/src/enqueue.cc +++ b/src/enqueue.cc @@ -3008,6 +3008,7 @@ ncclResult_t ncclEnqueueCheck(struct ncclInfo* info) { NCCLCHECKGOTO(taskAppend(info->comm, info), ret, fail); + info->comm->opCount++; exit: if (devOld != -1) CUDACHECK(cudaSetDevice(devOld)); ncclGroupErrCheck(ret); diff --git a/src/proxy.cc b/src/proxy.cc index 1bb213f51..387738c2e 100644 --- a/src/proxy.cc +++ b/src/proxy.cc @@ -1013,7 +1013,6 @@ ncclResult_t ncclProxyStart(struct ncclComm* comm) { ops->nextOps = ops->nextOpsEnd = -1; ops->count = 0; } - comm->opCount++; TIME_STOP(1); return ncclSuccess; }