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

[ROCm/rccl commit: 19105206f6]
Αυτή η υποβολή περιλαμβάνεται σε:
isaki001
2025-02-04 09:09:56 -06:00
υποβλήθηκε από GitHub
γονέας e171f59719
υποβολή d2b5ba80a7
2 αρχεία άλλαξαν με 3 προσθήκες και 3 διαγραφές
@@ -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;