Этот коммит содержится в:
Wenkai Du
2020-12-22 13:28:21 -05:00
родитель b33a2cac8b
Коммит 2ddbe6646b
5 изменённых файлов: 83 добавлений и 22 удалений
+12 -1
Просмотреть файл
@@ -286,7 +286,18 @@ struct ncclCollTrace {
uint32_t data_0;
uint64_t timeStamp;
uint64_t opCount;
uint64_t data_1;
union {
uint64_t data_1;
struct {
uint16_t nThreads;
uint8_t bid;
uint8_t nChannels;
} coll;
struct {
uint16_t nThreads;
uint16_t delta;
} p2p;
};
};
static_assert(sizeof(struct ncclCollTrace) == 8*sizeof(int), "ncclCollTrace must have a pow2 size");