File tree Expand file tree Collapse file tree 1 file changed +17
-0
lines changed
include/flashinfer/trtllm/gemm/trtllmGen_gemm_export Expand file tree Collapse file tree 1 file changed +17
-0
lines changed Original file line number Diff line number Diff line change @@ -97,6 +97,23 @@ enum class TileScheduler {
97
97
98
98
// //////////////////////////////////////////////////////////////////////////////////////////////////
99
99
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
+
100
117
// Helper functions to check the SplitK type.
101
118
102
119
#define SPLIT_K_FUNCTION (Mode ) \
You can’t perform that action at this time.
0 commit comments