AI/rocm-systems
2
0
Atdalīts 0
Kods Problēmas Izmaiņu pieprasījumi Darbības Pakotnes Projekti Laidieni Vikivietne Aktivitāte

Tuning: fixed out-of-bound access

Pārlūkot izejas kodu 
[ROCm/rccl commit: dea50b5e11]
Šī revīzija ir iekļauta:
alexander-sannikov
2025-12-08 16:24:37 +00:00
revīziju iesūtīja Alexander Sannikov
vecāks 4ef22f973e
revīzija 1b00f1a895
1 mainīti faili ar 6 papildinājumiem un 4 dzēšanām
Lejupielādēt ielāpa failu Lejupielādēt izmaiņu failu Izvērst visus failus Savērst visus failus
projects/rccl/src/graph/tuning.cc
+6 -4
Parādīt failu
@@ -888,14 +888,16 @@ ncclResult_t ncclTopoGetAlgoTime(struct ncclComm* comm, int coll, int algorithm,
int logSize = log2i(nBytes>>6);
#if defined(__HIP_PLATFORM_AMD__) || defined(__HIPCC__)
if (logSize < 0) logSize = 0;
if (logSize > 26) logSize = 26;
if (algorithm == NCCL_ALGO_TREE) {
if (logSize < 27) bw *= rcclTuningModel[comm->topo->tuning].treeCorrectionFactor[protocol][logSize];
else bw *= rcclTuningModel[comm->topo->tuning].treeCorrectionFactor[protocol][26];
bw *= rcclTuningModel[comm->topo->tuning].treeCorrectionFactor[protocol][logSize];
}
else if (algorithm == NCCL_ALGO_RING && comm->nNodes > 1) {
if(logSize < 27) bw *= rcclTuningModel[comm->topo->tuning].ringCorrectionFactor[protocol][logSize];
else bw *= rcclTuningModel[comm->topo->tuning].ringCorrectionFactor[protocol][26];
bw *= rcclTuningModel[comm->topo->tuning].ringCorrectionFactor[protocol][logSize];
}
#else
if (algorithm == NCCL_ALGO_TREE && coll == ncclFuncAllReduce && logSize >= 0 && logSize < 23) bw *= treeCorrectionFactor[protocol][logSize];
if (algorithm == NCCL_ALGO_RING && protocol == NCCL_PROTO_SIMPLE && comm->nNodes > 1