Misc fixes and disable binTree

This commit is contained in:
Wenkai Du
2022-09-14 00:10:19 +00:00
parent e5d2dfed34
commit a06e14e39b
3 changed files with 6 additions and 3 deletions
+1
View File
@@ -107,6 +107,7 @@ struct ncclChannel {
struct ncclRing ring;
int* devRingUserRanks;
struct ncclTree tree;
struct ncclTree binTree;
struct ncclDirect collTree;
int id; // index of this channel
uint32_t workFifoSent; // last used work index+1
+3 -2
View File
@@ -203,14 +203,15 @@ struct ncclWorkElem {
union {
uint8_t flagBits;
struct {
uint8_t isUsed:1, redOpArgIsPtr:1, regUsed:1, pad_0:1, nWarps:4;
uint8_t isUsed:1, redOpArgIsPtr:1, regUsed:1, nWarps:5;
};
};
uint8_t direct;
uint8_t bid;
uint8_t nChannels;
struct {
uint32_t root:30;
uint32_t root:28;
uint32_t pad_0:2;
uint32_t connIndex:2;
};