File tree Expand file tree Collapse file tree 2 files changed +17
-2
lines changed
python/tvm/meta_schedule/schedule_rule
src/meta_schedule/schedule_rule Expand file tree Collapse file tree 2 files changed +17
-2
lines changed Original file line number Diff line number Diff line change @@ -93,7 +93,23 @@ class MultiLevelTilingWithIntrin(ScheduleRule):
9393 intrin_name : str
9494 The name of a tensor intrinsic, must be registerd via TensorIntrin.register(...) beforehand
9595
96- The rest of parameters are the same as MultiLevelTiling above.
96+ structure : str
97+ The tiling structure. Recommended:
98+ - 'SSRSRS' on CPU
99+ - 'SSSRRSRS' on GPU
100+ tile_bind : Optional[List[str]]
101+ For each level of tiles, which thread axis it is bound to. Recommended:
102+ - None on CPU
103+ - [blockIdx.x, vthread.x, threadIdx.x] on GPU
104+ max_innermost_factor : Optional[int]
105+ The maximum size of the innermost factor. None means no limit
106+ vector_load_lens : Optional[List[int]]
107+ The length of vector lane in vectorized cooperative fetching.
108+ None means disable vectorization
109+ reuse_read : Optional[ReuseType]
110+ Data reuse configuration for reading. None means no reuse.
111+ reuse_write : Optional[ReuseType]
112+ Data reuse configuration for writing. None means no reuse.
97113 """
98114
99115 def __init__ (
Original file line number Diff line number Diff line change 2525#include < vector>
2626
2727#include " ../utils.h"
28- #include " tvm/meta_schedule/schedule_rule.h"
2928
3029namespace tvm {
3130namespace tir {
You can’t perform that action at this time.
0 commit comments