ganler commented on code in PR #12507:
URL: https://github.com/apache/tvm/pull/12507#discussion_r950643008
##########
src/tir/schedule/trace.cc:
##########
@@ -476,6 +476,8 @@ TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable)
.set_dispatch<TraceNode>([](const ObjectRef& obj, ReprPrinter* p) {
const auto* self = obj.as<TraceNode>();
ICHECK_NOTNULL(self);
+ p->stream << "# from tvm import tir" << std::endl;
+ p->stream << "def apply_trace(sch: tir.Schedule) -> None:" << std::endl;
Review Comment:
Just a minor suggestion that frequently calling IO flushing via `std::endl`
for shorter strings might be less effective. So putting a `std::flush` in the
end and keep using `<< '\n'` for newlines other than `<< std::endl` in previous
places should be better generally.
--
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]
