diff --git a/projects/rccl/src/init.cc b/projects/rccl/src/init.cc index 6eb6916eb0..02fdccdffc 100644 --- a/projects/rccl/src/init.cc +++ b/projects/rccl/src/init.cc @@ -662,7 +662,7 @@ static ncclResult_t devCommSetup(ncclComm_t comm) { #endif #ifdef ENABLE_PROFILING - NCCLCHECK(ncclCudaCalloc(&tmpCommAndChans.comm.devProf, MAXCHANNELS*PROFILE_NUM_LAUNCHES), comm->sideStream); + NCCLCHECK(ncclCudaCalloc(&tmpCommAndChans.comm.devProf, MAXCHANNELS*PROFILE_NUM_LAUNCHES, comm->sideStream)); #endif NCCLCHECKGOTO(ncclCudaMemcpyAsync(devCommAndChans, &tmpCommAndChans, 1, comm->sharedRes->deviceStream.cudaStream), ret, fail);