msccl: enable basic collective trace (#959)
To avoid increasing number of kernels, colltrace is only enabled with RCCL_MSCCL_FORCE_FULLOPS=1
This commit is contained in:
@@ -36,6 +36,7 @@ struct mscclSchedulerParam {
|
||||
int nRanks;
|
||||
bool scheduled;
|
||||
mscclAlgoHandle_t handle;
|
||||
uint64_t opCount;
|
||||
};
|
||||
|
||||
typedef struct {
|
||||
|
||||
@@ -239,7 +239,7 @@ struct mscclWork {
|
||||
int nChunksPerLoop;
|
||||
bool hasReduce;
|
||||
bool redOpArgIsPtr;
|
||||
uint32_t pad[1];
|
||||
uint32_t fnIndex;
|
||||
};
|
||||
static_assert(sizeof(struct mscclWork) % 16 == 0, "mscclWork needs to be 16B aligned");
|
||||
|
||||
|
||||
Reference in New Issue
Block a user