Add macro for channel mask offset (#1467)
[ROCm/rccl commit: a9d6e7661c]
This commit is contained in:
کامیت شده توسط
GitHub
والد
300f954185
کامیت
dd3fc22531
@@ -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];
|
||||
|
||||
مرجع در شماره جدید
Block a user