[GitHub] [tvm] junrushao commented on a diff in pull request #12450: [TIR][Schedule] enhance compute_at and reverse_compute_at primitive to choose possible position

[GitBox <gi...@apache.org>]

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: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org