2
0

[P2P] Have connIdx for both send and recv (#1524)

[ROCm/rccl commit: 387c973b5d]
Este cometimento está contido em:
Bertan Dogancay
2025-02-04 11:53:20 -05:00
cometido por GitHub
ascendente a40d4eb960
cometimento b52af8d803
5 ficheiros modificados com 26 adições e 19 eliminações
+3 -2
Ver ficheiro
@@ -247,7 +247,7 @@ struct alignas(16) ncclDevWorkP2p {
uint8_t sendProtoLL:1, recvProtoLL:1;
uint8_t sendRegistered:1, recvRegistered:1;
uint8_t connIndex:2;
uint8_t sendConnIndex:2, recvConnIndex:2;
};
// Compute the subset of the data transfer corresponding to the given part index.
@@ -441,7 +441,8 @@ struct ncclCollTrace {
uint8_t nSendChannels;
uint8_t nRecvChannels;
uint8_t channelBase;
uint8_t connIndex;
uint8_t sendConnIndex:2;
uint8_t recvConnIndex:2;
uint8_t sendProtoLL:1;
uint8_t recvProtoLL:1;
} p2p;