Optimize alltoall for 64 GPUs and above for gfx942 (#1828)

Add pxn and p2p net chunksize mi300x tuning

[ROCm/rccl commit: 4ce3df8d3a]
Bu işleme şunda yer alıyor:
Mustafa Abduljabbar
2025-07-30 15:14:43 -04:00
işlemeyi yapan: GitHub
ebeveyn cca5172260
işleme cafd7a5126
4 değiştirilmiş dosya ile 43 ekleme ve 6 silme
+7 -1
Dosyayı Görüntüle
@@ -41,6 +41,11 @@ typedef enum RcclTunableColls {
#define RCCL_PROTOCOL_FACTOR_IDX 2
#define RCCL_PROTOCOL_THREAD_THRESHOLD_IDX 3
typedef enum {
RCCL_VALUE_UNSET = -2,
RCCL_VALUE_INVALID = -1
} rcclValueState_t;
#ifdef RCCL_EXPOSE_STATIC
#define RCCL_STATIC_EXPOSE_CHECK()
#else
@@ -78,7 +83,8 @@ void rcclUpdateThreadThreshold(struct ncclComm* comm, size_t const& nBytes, stru
ncclResult_t rcclGetAlgoInfo(struct ncclComm* comm, ncclFunc_t coll, uint64_t count, ncclDataType_t dataType,
int collNetSupport, int nvlsSupport, int numPipeOps,
int* algo, int* protocol, int* maxChannels);
void rcclSetPxn(struct ncclComm* comm, int& rcclPxnDisable);
void rcclSetP2pNetChunkSize(struct ncclComm* comm, int& rcclP2pNetChunkSize);
ncclResult_t rcclFuncMaxSendRecvCount(ncclFunc_t func, int nRanks, size_t count, size_t& maxCount);
ncclResult_t commSetUnrollFactor(struct ncclComm* comm);
#endif