Add another Rome model (#483)

此提交包含在:
Wenkai Du
2022-01-05 09:26:31 -08:00
提交者 GitHub
父節點 bbbb35ceec
當前提交 f8d0775a6f
共有 5 個檔案被更改,包括 142 行新增 和 2 行删除
+1 -1
查看文件
@@ -509,7 +509,6 @@ ncclResult_t ncclTopoTrimSystem(struct ncclTopoSystem* system, struct ncclComm*
}
}
if (ret) {
system->type |= RCCL_TOPO_GDR_ALL;
bool allXgmi = true;
// don't trim NICs unless all GPUs are connected by XGMI
for (int i = 0; i < system->nodes[GPU].count && allXgmi; i++) {
@@ -524,6 +523,7 @@ ncclResult_t ncclTopoTrimSystem(struct ncclTopoSystem* system, struct ncclComm*
}
if (!allXgmi) {
remove = 0;
system->type |= RCCL_TOPO_GDR_ALL;
INFO(NCCL_GRAPH, "GDR is available on all GPUs");
}
}
+28
查看文件
@@ -434,6 +434,32 @@ static struct rcclRomeModel rome_model_65 = {
.netGdrLevel = 5,
};
static struct rcclRomeModel rome_model_66 = {
.nGpus = 8, .nCpus = 2, .nNics = 0, .nLinks = 3,
.gpuIds = { 0x29000, 0x2c000, 0x2f000, 0x32000, 0xad000, 0xb0000, 0xb3000, 0xb6000, },
.nicIds = { },
.gpuNuma = { 1, 1, 1, 1, 3, 3, 3, 3, },
.nicNuma = { },
.connMatrix = { 0, 1, 0, 0, 1, 0, 1, 0, 1, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 0, 1, 0, 0, 1, 1, 0, 0, 1, 0, 0, 1, 0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1, 1, 0, 0, 1, 1, 0, 1, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 1, 1, 0, },
.gdrLevel = { },
.pattern = "4040",
.ringBase = "0 6 7 5 4 2 3 1|1 3 2 4 5 7 6 0|0 1 7 6 2 3 5 4|4 5 3 2 6 7 1 0",
.netGdrLevel = -2,
};
static struct rcclRomeModel rome_model_67 = {
.nGpus = 8, .nCpus = 2, .nNics = 4, .nLinks = 3,
.gpuIds = { 0x29000, 0x2c000, 0x2f000, 0x32000, 0xad000, 0xb0000, 0xb3000, 0xb6000, },
.nicIds = { 0x1d000, 0x1e000, 0xa1000, 0xa2000, },
.gpuNuma = { 1, 1, 1, 1, 3, 3, 3, 3, },
.nicNuma = { 1, 1, 3, 3, },
.connMatrix = { 0, 1, 0, 0, 1, 0, 1, 0, 1, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 0, 1, 0, 0, 1, 1, 0, 0, 1, 0, 0, 1, 0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1, 1, 0, 0, 1, 1, 0, 1, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 1, 1, 0, },
.gdrLevel = { 4, 4, 4, 4, 6, 6, 6, 6, 4, 4, 4, 4, 6, 6, 6, 6, 6, 6, 6, 6, 4, 4, 4, 4, 6, 6, 6, 6, 4, 4, 4, 4, },
.pattern = "4242",
.ringBase = "N3 7 6 0 1 3 2 4 5 N2|N2 5 4 2 3 1 0 6 7 N3|N1 2 3 5 4 0 1 7 6 N3|N2 4 5 3 2 6 7 1 0 N0|N1 3 2 4 5 7 6 0 1 N0|N0 1 0 6 7 5 4 2 3 N1|N0 0 1 7 6 2 3 5 4 N2|N3 6 7 1 0 4 5 3 2 N1",
.netGdrLevel = -2,
};
static struct rcclRomeModel romeTopoModels[] = {
rome_model_22,
rome_model_25,
@@ -465,6 +491,8 @@ static struct rcclRomeModel romeTopoModels[] = {
rome_model_62,
rome_model_63,
rome_model_65,
rome_model_66,
rome_model_67,
};
/* Parse user defined rings. Format is like :