AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add another Rome model (#497)

Browse Source
This commit is contained in:
Wenkai Du
2022-02-12 10:30:16 -08:00
committed by GitHub
parent f3c2cafd9d
commit 02096c9936
5 changed files with 117 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/enqueue.cc
+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;