add 256 channels support

[ROCm/rccl commit: 506f16c506]
2024-05-15 16:58:28 -05:00
parent 1934d0f377
commit b0b2aa1166
11 changed files with 95 additions and 53 deletions
@@ -143,7 +143,7 @@ struct ncclProxyOps {

 struct ncclProxySharedP2p {
  int refcount;
-  int size;
+  int64_t size;
  char* cudaBuff;
  char* hostBuff;
  // CUDA IPC
@@ -331,4 +331,4 @@ ncclResult_t ncclProxyShmUnlink(struct ncclComm* comm);
 ncclResult_t ncclProxyDestroy(struct ncclComm* comm);

 ncclResult_t mscclSaveProxy(struct ncclComm* comm, struct ncclChannel* channel, int type, int peer, struct ncclProxyOp* op, int connIndex);
-#endif
+#endif