2
0

Fix GPU to NIC mapping in tree (#573)

* Fix GPU to NIC mapping in tree

* Update tuning table
Este cometimento está contido em:
Wenkai Du
2022-07-03 20:52:52 -07:00
cometido por GitHub
ascendente a89a9966aa
cometimento 00af1f64e9
3 ficheiros modificados com 23 adições e 18 eliminações
+4 -6
Ver ficheiro
@@ -477,7 +477,7 @@ static ncclResult_t getAlgoInfo(struct ncclInfo* info, int collNetTypeSupport, i
}
}
} else if (comm->topo->nodes[GPU].nodes[0].gpu.gcn == 910 && comm->topo->tuning == 4 &&
((comm->nNodes == 2 && info->nBytes == 33554432) || (comm->nNodes <= 4 && info->nBytes == 67108864))) {
comm->nNodes >= 4 && info->nBytes >= 4294967296) {
static int userTuneInput = -2;
if (userTuneInput == -2) {
const char *protoStr = getenv("NCCL_PROTO");
@@ -487,14 +487,12 @@ static ncclResult_t getAlgoInfo(struct ncclInfo* info, int collNetTypeSupport, i
else
userTuneInput = 1;
}
if (userTuneInput) {
info->nChannels = nc;
if (!userTuneInput) {
// always respect user settings
info->nChannels = nc;
} else {
// use ring simple with reduced channels on gfx90a for specific data sizes
// use ring simple on gfx90a for specific data sizes
info->protocol = NCCL_PROTO_SIMPLE;
info->algorithm = NCCL_ALGO_RING;
info->nChannels = nc/2;
}
} else {
info->nChannels = nc;