23 lines
484 B
C
23 lines
484 B
C
|
#ifndef NOTIFY_DISPATCH_TILING_H
|
||
|
|
#define NOTIFY_DISPATCH_TILING_H
|
||
|
|
|
||
|
|
#include "kernel_tiling/kernel_tiling.h"
|
||
|
|
|
||
|
|
struct NotifyDispatchInfo {
|
||
|
|
uint32_t rankSize;
|
||
|
|
uint32_t rankId;
|
||
|
|
uint32_t localRankSize;
|
||
|
|
uint32_t localRankId;
|
||
|
|
uint32_t sendCount;
|
||
|
|
uint32_t numTokens;
|
||
|
|
uint32_t aivNum;
|
||
|
|
uint64_t totalUbSize;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct NotifyDispatchTilingData {
|
||
|
|
Mc2InitTiling mc2InitTiling;
|
||
|
|
Mc2CcTiling mc2CcTiling1;
|
||
|
|
NotifyDispatchInfo notifyDispatchInfo;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|