AI/rocm-systems
2
0 Forc 0
Cód Saincheisteanna Iarratais Tarraingthe Gníomhartha Pacáistí Tionscadail Scaoileann Vicí Gníomhaíocht

Fix GPU to NIC mapping for peertopeer (#994)

Brabhsáil Foinse 
[ROCm/rccl commit: 0b53f79196]
Tá an tiomantas seo le fáil i:
pradeep-ramanna údaraithe agus GitHub tiomanta 2023-12-01 08:00:17 -08:00
1 tuismitheoir cef45b8311
tiomantas bf57487384
1 comhad athraithe
+2 -2
Íoslódáil an comhad paiste Íoslódáil Comhad Diff Leathnaigh gach comhaid Laghdaigh gach comhaid
projects/rccl/src/graph/paths.cc
+2 -2
Féach ar an gComhad
@@ -548,7 +548,7 @@ static bool rcclPathOverride(struct ncclTopoSystem* system, uint64_t distance) {
for (i = 0; i < system->nodes[GPU].count; i++) {
for (j = 0; j < system->nodes[NET].count; j++) {
if (system->nodes[NET].nodes[j].net.busId - system->nodes[GPU].nodes[i].id == distance)
if ((system->nodes[NET].nodes[j].net.busId - system->nodes[GPU].nodes[i].id == distance) || (system->nodes[GPU].nodes[i].id - system->nodes[NET].nodes[j].net.busId == distance))
break;
}
if (j >= system->nodes[NET].count)
@@ -557,7 +557,7 @@ static bool rcclPathOverride(struct ncclTopoSystem* system, uint64_t distance) {
if (i >= system->nodes[GPU].count) {
for (i = 0; i < system->nodes[GPU].count; i++) {
for (j = 0; j < system->nodes[NET].count; j++) {
if (system->nodes[NET].nodes[j].net.busId - system->nodes[GPU].nodes[i].id == distance)
if ((system->nodes[NET].nodes[j].net.busId - system->nodes[GPU].nodes[i].id == distance) || (system->nodes[GPU].nodes[i].id - system->nodes[NET].nodes[j].net.busId == distance))
system->nodes[GPU].nodes[i].paths[NET][j].type = PATH_PXB;
}
}