Skip to content

Commit ba1f6b8

Browse files
committed
remove unnecessary include, add doc for the rest of params
1 parent 56f82eb commit ba1f6b8

File tree

2 files changed

+17
-2
lines changed

2 files changed

+17
-2
lines changed

python/tvm/meta_schedule/schedule_rule/multi_level_tiling.py

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff 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__(

src/meta_schedule/schedule_rule/multi_level_tiling.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
#include <vector>
2626

2727
#include "../utils.h"
28-
#include "tvm/meta_schedule/schedule_rule.h"
2928

3029
namespace tvm {
3130
namespace tir {

0 commit comments

Comments
 (0)