This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 943861c85c [TIR][Schedule] TileWithTensorIntrin skip incorrect
ComputeInline for input-padding (#16239)
943861c85c is described below
commit 943861c85c6e72a6b87498642107b88803b26a1f
Author: XFPlus <[email protected]>
AuthorDate: Tue Dec 19 23:17:15 2023 +0800
[TIR][Schedule] TileWithTensorIntrin skip incorrect ComputeInline for
input-padding (#16239)
[TIR][Schedule] TileWithTensorInstrin Skip some ComputeInline for some
producer if not padded
---
src/tir/schedule/transform.cc | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/tir/schedule/transform.cc b/src/tir/schedule/transform.cc
index 62662a4ce1..33ec876352 100644
--- a/src/tir/schedule/transform.cc
+++ b/src/tir/schedule/transform.cc
@@ -318,10 +318,12 @@ Optional<LoopRV> TileWithTensorIntrin(const
tir::Schedule& sch, const tir::Block
auto producers = sch->GetProducers(block_rv);
for (const auto& producer : producers) {
auto original_producers = sch->GetProducers(producer);
- ICHECK_EQ(original_producers.size(), 1u);
+ // NOTICE: there may not all producers padded.
// Inline the original producer into the padding block. This ensures
that the new producer
// has the padded shape.
- sch->ComputeInline(original_producers[0]);
+ if (original_producers.size() == 1u) {
+ sch->ComputeInline(original_producers[0]);
+ }
}
auto consumers = sch->GetConsumers(block_rv);
for (const auto& consumer : consumers) {
