We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 029e3b6 commit 584d054Copy full SHA for 584d054
src/te/schedule/schedule_lang.cc
@@ -778,11 +778,18 @@ TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable)
778
p->Print(op->outer);
779
p->stream << ", inner=";
780
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
+ }
788
p->stream << ')';
789
})
790
.set_dispatch<FuseNode>([](const ObjectRef& node, ReprPrinter* p) {
791
auto* op = static_cast<const FuseNode*>(node.get());
- p->stream << "split(";
792
+ p->stream << "fuse(";
793
p->stream << "outer=";
794
795
0 commit comments