Lunderberg commented on code in PR #12720:
URL: https://github.com/apache/tvm/pull/12720#discussion_r973042996
##########
python/tvm/tir/schedule/schedule.py:
##########
@@ -2479,6 +2480,31 @@ def transform_layout(
primitive will be called in addition to the
TransformLayout primitive.
+ pad_value: Optional[Union[int, float, PrimExpr, IndexMap, Callable]]
Review Comment:
Thank you, and the docstring has been updated. I've also added two unit
tests, one that validates that an error is raised when the pad value loads from
a different buffer, and one that specifies the intended behavior for pad value
that loads from the transformed buffer. The latter is currently marked with
`pytest.mark.xfail`, as the support isn't implemented yet.
--
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]
