Merge remote-tracking branch 'nccl/master' into develop

[ROCm/rccl commit: da3b197d6c]
This commit is contained in:
Wenkai Du
2020-07-01 16:51:25 -07:00
5 changed files with 15 additions and 8 deletions
+4 -1
View File
@@ -146,7 +146,10 @@ void* ncclAsyncThreadPreconnect(void* args_) {
NCCL_API(ncclResult_t, ncclGroupEnd);
ncclResult_t ncclGroupEnd() {
if (ncclGroupMode == 0) return ncclInvalidUsage;
if (ncclGroupMode == 0) {
WARN("ncclGroupEnd: not in a group call.");
return ncclInvalidUsage;
}
ncclGroupMode--;
if (ncclGroupMode > 0) return ncclSuccess;
int savedDev;