cblmemo commented on code in PR #14673:
URL: https://github.com/apache/tvm/pull/14673#discussion_r1241023473
##########
python/tvm/tir/schedule/schedule.py:
##########
@@ -402,6 +402,46 @@ def sample_perfect_tile(
)
)
+ @type_checked
+ def sample_partitioned_tile(
+ self,
+ loop: LoopRV,
+ n: int,
+ partition_pos: int = 0,
+ innerpart_factor: int = 1,
+ decision: Optional[List[int]] = None,
+ ) -> List[ExprRV]:
+ """Sample the factors to a partitioned tile for a specific loop
+
+ Parameters
+ ----------
+ loop : LoopRV
+ The loop to be tiled
+ n : int
+ The number of tiles to be sampled
+ partition_pos : int
+ The position to partition tiles to two parts
+ innerpart_factor : int
+ The factor of the second part
+ decision: Optional[List[int]]
+ The sampling decision, if any
+
+ Returns
+ -------
+ result : List[ExprRV]
+ A list of length `n`, the random partitioned tile sizes sampled
+ """
+ return list(
Review Comment:
The `sample_perfect_tile` primitive also use this explicit conversion
--
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]
