Add another Rome model (#497)

[ROCm/rccl commit: 02096c9936]
This commit is contained in:
Wenkai Du
2022-02-12 10:30:16 -08:00
committed by GitHub
parent 9c3189589f
commit 5b697e40db
5 changed files with 117 additions and 2 deletions
+1 -1
View File
@@ -467,7 +467,7 @@ static ncclResult_t getLoopInfo(struct ncclInfo* info) {
return ncclSuccess;
}
RCCL_PARAM(IntraNetThreshold, "RCCL_INTRANET_THRESHOLD", 8388608);
RCCL_PARAM(IntraNetThreshold, "INTRANET_THRESHOLD", 8388608);
static ncclResult_t computeColl(struct ncclInfo* info /* input */, struct ncclWorkElem* work, struct ncclProxyArgs* proxyArgs /* output */) {
work->comm = info->comm->devComm;