Add macro for channel mask offset (#1467)

[ROCm/rccl commit: a9d6e7661c]
このコミットが含まれているのは:
Mustafa Abduljabbar
2025-01-03 08:41:41 -05:00
committed by GitHub
コミット dd3fc22531
2個のファイルの変更、8行の追加、8行の削除
+1 -1
ファイルの表示
@@ -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];