Update MSCCL++ register/deregister (#1523)
* erase handle key from mscclpp communicator during deregistration * remove check on buffer size being a multiple of 32 from registration/deregistration routines since these checks are applied during enqueue * add check for greater than zero buffer size in mscclpp registration
Этот коммит содержится в:
@@ -116,7 +116,7 @@ index 022d398..468fcf2 100644
|
||||
+ if (outIt != comm->channelOutInfos.end()) {
|
||||
+ comm->channelOutInfos.erase(outIt);
|
||||
+ }
|
||||
+
|
||||
+ comm->handleKeys.erase(handle);
|
||||
+ free(handle);
|
||||
+ }
|
||||
+ return ncclSuccess;
|
||||
|
||||
Ссылка в новой задаче
Block a user