Lunderberg commented on code in PR #12720:
URL: https://github.com/apache/tvm/pull/12720#discussion_r968604370
##########
python/tvm/tir/schedule/schedule.py:
##########
@@ -2479,6 +2480,20 @@ def transform_layout(
primitive will be called in addition to the
TransformLayout primitive.
+ pad_value: Optional[Union[int, float, PrimExpr, IndexMap, Callable]]
+
+ The value to be used for any padding introduced by the
+ transformation.
+
+ If None, the transformation may not introduce padding.
+
+ If an int, float or PrimExpr, the transformation is the
+ specific value to be present in the padding.
+
+ If an IndexMap or Callable, the transformation is the
+ value to be present in the padding in terms of the
+ transformed index.
Review Comment:
Updates made to pass `Optional<IndexMap> pad_value` throughout C++ API,
mimicking how `IndexMap index_map` is passed, along with a unit test to
validate the functionality.
--
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-unsubscr...@tvm.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
