Hzfengsy commented on code in PR #14095:
URL: https://github.com/apache/tvm/pull/14095#discussion_r1121078571
##########
include/tvm/tir/schedule/schedule.h:
##########
@@ -642,10 +642,17 @@ class ScheduleNode : public runtime::Object {
* Algebraic symplifications, branch elimination, and other
* optimizations may assume that this precondition is met, and
* may result in incorrect results being returned.
+ *
+ * \param assume_injective_transform If set to true, the schedule primitive
will assume the
Review Comment:
```suggestion
* \param assume_injective_transform If set to true, the schedule
primitive will assume the
```
##########
python/tvm/tir/schedule/schedule.py:
##########
@@ -2645,6 +2644,8 @@ def transform_layout(
buffer: Union[Tuple[str, int], str, Buffer],
index_map: Union[IndexMap, Callable],
pad_value: Optional[Union[int, float, PrimExpr, IndexMap, Callable]] =
None,
+ *,
+ assume_injective_transform=False,
Review Comment:
```suggestion
assume_injective_transform: bool = False,
```
##########
src/tir/schedule/primitive.h:
##########
@@ -501,10 +501,15 @@ TVM_DLL void Unannotate(ScheduleState self, const
StmtSRef& sref, const String&
* \param buffer_index_type The type of the buffer index, kRead or kWrite.
* \param index_map The transformation to apply.
* \param pad_value The value to write into padding introduced by the
transformation.
+ * \param assume_injective_transform If set to true, the schedule primitive
will assume the
Review Comment:
```suggestion
* \param assume_injective_transform If set to true, the schedule primitive
will assume the
```
--
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]
