Skip to content

Commit

Permalink
Fix printing of schedule operations (apache#8949)
Browse files Browse the repository at this point in the history
- Add printing of factor/nparts in "split".
- Print correct operation name in "fuse".
  • Loading branch information
Krzysztof Parzyszek authored and ylc committed Jan 13, 2022
1 parent 62f3dd0 commit 5c9782a
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion src/te/schedule/schedule_lang.cc
Original file line number Diff line number Diff line change
Expand Up @@ -778,11 +778,18 @@ TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable)
p->Print(op->outer);
p->stream << ", inner=";
p->Print(op->inner);
if (op->factor.defined()) {
p->stream << ", factor=";
p->Print(op->factor);
} else {
p->stream << ", nparts=";
p->Print(op->nparts);
}
p->stream << ')';
})
.set_dispatch<FuseNode>([](const ObjectRef& node, ReprPrinter* p) {
auto* op = static_cast<const FuseNode*>(node.get());
p->stream << "split(";
p->stream << "fuse(";
p->stream << "outer=";
p->Print(op->outer);
p->stream << ", inner=";
Expand Down

0 comments on commit 5c9782a

Please sign in to comment.