21 lines
412 B
C
21 lines
412 B
C
|
#ifndef DISPATCH_LAYOUT_TILING_H
|
||
|
|
#define DISPATCH_LAYOUT_TILING_H
|
||
|
|
|
||
|
|
#include "kernel_tiling/kernel_tiling.h"
|
||
|
|
|
||
|
|
struct DispatchLayoutInfo {
|
||
|
|
uint32_t numTokens;
|
||
|
|
uint32_t numRanks;
|
||
|
|
uint32_t numExperts;
|
||
|
|
uint32_t numTopk;
|
||
|
|
uint64_t totalUbSize;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct DispatchLayoutTilingData {
|
||
|
|
Mc2InitTiling mc2InitTiling;
|
||
|
|
Mc2CcTiling mc2CcTiling1;
|
||
|
|
DispatchLayoutInfo dispatchLayoutInfo;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|