Skip to content

Commit acc4fa7

Browse files
committed
manual update of Enums.h header for trtllm-gen
1 parent 6334da9 commit acc4fa7

File tree

1 file changed

+17
-0
lines changed
  • include/flashinfer/trtllm/gemm/trtllmGen_gemm_export

1 file changed

+17
-0
lines changed

include/flashinfer/trtllm/gemm/trtllmGen_gemm_export/Enums.h

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,23 @@ enum class TileScheduler {
9797

9898
////////////////////////////////////////////////////////////////////////////////////////////////////
9999

100+
enum class CtaSwizzleType : uint32_t {
101+
// Rasterize CTAs along the M dimension.
102+
RasterizeAlongM = 0,
103+
// Rasterize CTAs along the N dimension.
104+
RasterizeAlongN,
105+
// Swizzle CTAs in zig-zag pattern along M dimension, Zig-zag width is 2.
106+
ZigZagAlongM2,
107+
// Swizzle CTAs in zig-zag pattern along N dimension, Zig-zag width is 2.
108+
ZigZagAlongN2,
109+
// Swizzle CTAs in zig-zag pattern along M dimension, Zig-zag width is 4.
110+
ZigZagAlongM4,
111+
// Swizzle CTAs in zig-zag pattern along N dimension, Zig-zag width is 4.
112+
ZigZagAlongN4,
113+
};
114+
115+
////////////////////////////////////////////////////////////////////////////////////////////////////
116+
100117
// Helper functions to check the SplitK type.
101118

102119
#define SPLIT_K_FUNCTION(Mode) \

0 commit comments

Comments
 (0)