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

[ROCm/rccl commit: 387c973b5d]
This commit is contained in:
Bertan Dogancay
2025-02-04 11:53:20 -05:00
zatwierdzone przez GitHub
rodzic a40d4eb960
commit b52af8d803
5 zmienionych plików z 26 dodań i 19 usunięć
+3 -2
Wyświetl plik
@@ -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;