2.7.5-1

Minor fixes for A100 platforms. Add a WARN for invalid GroupEnd call.
2020-06-22 09:36:20 -07:00
parent 5949d96f36
commit 01afd20a77
4 changed files with 8 additions and 5 deletions
@@ -145,7 +145,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;