junrushao commented on code in PR #12450:
URL: https://github.com/apache/tvm/pull/12450#discussion_r955644743
##########
python/tvm/tir/schedule/schedule.py:
##########
@@ -1303,6 +1304,11 @@ def compute_at(
preserve_unit_loops: bool
Whether to keep the trivial loops whose extents are 1
+ index: int
+ The block index of the loop body subtree blocks
+ -1 means inserted into the last possible insertion point
+ -2 means inserted into the first possible insertion point
Review Comment:
```suggestion
The block index of the loop body subtree blocks:
- `index = -1` means inserted into the last possible insertion
point;
- `index = -2` means inserted into the first possible insertion
point;
- Otherwise, `index` is a nonnegative number that indicates the
insertion point
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]