Add macro for channel mask offset (#1467)

Tento commit je obsažen v:
Mustafa Abduljabbar
2025-01-03 08:41:41 -05:00
odevzdal GitHub
rodič 39483c55f8
revize a9d6e7661c
2 změnil soubory, kde provedl 8 přidání a 8 odebrání
+1 -1
Zobrazit soubor
@@ -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];