Add macro for channel mask offset (#1467)

[ROCm/rccl commit: a9d6e7661c]
This commit is contained in:
Mustafa Abduljabbar
2025-01-03 08:41:41 -05:00
کامیت شده توسط GitHub
والد 300f954185
کامیت dd3fc22531
2فایلهای تغییر یافته به همراه8 افزوده شده و 8 حذف شده
@@ -48,7 +48,7 @@ struct ncclPeerInfo {
// MNNVL support
nvmlGpuFabricInfoV_t fabricInfo;
};
#define CHANNEL_MASK_OFFSET(nranks, connIndex) (nranks * (connIndex == NCCL_CONN_IDX_P2P_NET ? NCCL_CONN_IDX_P2P_NET : 0))
#define CONNECT_SIZE 128
struct ncclConnect {
char data[CONNECT_SIZE];