AI/rocm-systems
2
0
Fork 0
Koodi Ongelmat Pull-pyynnöt Actions Paketit Projektit Julkaisut Wiki Toiminta

Disable context tracking for the current version. (#1839)

Selaa lähdekoodia
This commit is contained in:
Arm Patinyasakdikul
2025-08-04 10:48:00 -05:00
committed by GitHub
vanhempi 0e7d7da55d
commit 6fc228e247
1 muutettua tiedostoa jossa 4 lisäystä ja 2 poistoa
Download Patch File Download Diff File Expand all files Collapse all files
src/init.cc
+4 -2
Näytä tiedosto
@@ -501,7 +501,8 @@ static ncclResult_t commFree(ncclComm_t comm) {
NCCLCHECK(ncclNetFinalize(comm)); NCCLCHECK(ncclNetFinalize(comm));
NCCLCHECK(ncclNetPluginUnload(comm)); NCCLCHECK(ncclNetPluginUnload(comm));
ncclCudaContextDrop(comm->context); // Disable until we validate NCCL_LAUNCH_IMPLICIT_ORDER support.
//ncclCudaContextDrop(comm->context);
free(comm); free(comm);
@@ -599,7 +600,8 @@ static ncclResult_t commAlloc(struct ncclComm* comm, struct ncclComm* parent, in
comm->lastStream = nullptr; comm->lastStream = nullptr;
CUDACHECK(cudaGetDevice(&comm->cudaDev)); CUDACHECK(cudaGetDevice(&comm->cudaDev));
NCCLCHECK(ncclCudaContextTrack(&comm->context)); // Disable until we validate NCCL_LAUNCH_IMPLICIT_ORDER support.
//NCCLCHECK(ncclCudaContextTrack(&comm->context));
NCCLCHECK(getBusId(comm->cudaDev, &comm->busId)); NCCLCHECK(getBusId(comm->cudaDev, &comm->busId));
char busId[]="0000:00:00.0"; char busId[]="0000:00:00.0";