Skip to content

Commit 584d054

Browse files
Krzysztof Parzyszekylc
authored andcommitted
Fix printing of schedule operations (apache#8949)
- Add printing of factor/nparts in "split". - Print correct operation name in "fuse".
1 parent 029e3b6 commit 584d054

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/te/schedule/schedule_lang.cc

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -778,11 +778,18 @@ TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable)
778778
p->Print(op->outer);
779779
p->stream << ", inner=";
780780
p->Print(op->inner);
781+
if (op->factor.defined()) {
782+
p->stream << ", factor=";
783+
p->Print(op->factor);
784+
} else {
785+
p->stream << ", nparts=";
786+
p->Print(op->nparts);
787+
}
781788
p->stream << ')';
782789
})
783790
.set_dispatch<FuseNode>([](const ObjectRef& node, ReprPrinter* p) {
784791
auto* op = static_cast<const FuseNode*>(node.get());
785-
p->stream << "split(";
792+
p->stream << "fuse(";
786793
p->stream << "outer=";
787794
p->Print(op->outer);
788795
p->stream << ", inner=";

0 commit comments

Comments
 (0)