Fix topo_explorer compatibility and capture WarpSize (#1743)

Этот коммит содержится в:
Mustafa Abduljabbar
2025-06-16 15:18:35 +03:00
коммит произвёл GitHub
родитель ba97c9c18b
Коммит fb4ad82d0d
2 изменённых файлов: 12 добавлений и 4 удалений
+2
Просмотреть файл
@@ -255,11 +255,13 @@ int main(int argc,char* argv[])
node_model = network.GetNode(i);
assert(node_model!=0);
initTransportsRank_3(&comm[i], allGather3Data, treeGraph[i], ringGraph[i], collNetGraph[i], nvlsGraph[i]);
CUDACHECK(hipDeviceGetAttribute(&comm[i].WarpSize, hipDeviceAttributeWarpSize, comm[i].cudaDev));
}
for (uint64_t len = 8; len <= 4294967296L; len *= 2) {
struct ncclInfo info;
float minTime = 3600000000.0;
info.comm = &comm[0];
info.coll = ncclFuncAllReduce;
// Find algorithm / protocol.
int algorithm = -1;