Fix broken build due to ncclCudaCalloc change (#2135)

[ROCm/rccl commit: 11e0f4445e]
This commit is contained in:
Wenkai Du
2026-01-09 09:22:00 -08:00
committad av GitHub
förälder 4e15dc142c
incheckning 87eec6427e
+1 -1
Visa fil
@@ -682,7 +682,7 @@ static ncclResult_t commAlloc(struct ncclComm* comm, struct ncclComm* parent, in
#ifdef ENABLE_COLLTRACE
NCCLCHECK(ncclCudaHostCalloc(&comm->collTraceTail, MAXCHANNELS));
#if defined(HIP_UNCACHED_MEMORY)
NCCLCHECK(ncclCudaCalloc(&comm->collTrace, COLLTRACE_NUM_ITEMS*MAXCHANNELS, nullptr, hipDeviceMallocUncached));
NCCLCHECK(ncclCudaCalloc(&comm->collTrace, COLLTRACE_NUM_ITEMS*MAXCHANNELS, hipDeviceMallocUncached));
#else
NCCLCHECK(ncclCudaCalloc(&comm->collTrace, COLLTRACE_NUM_ITEMS*MAXCHANNELS));
#endif