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 slow. So putting a `std::flush` in the end and
keep using `<< '\n'` for newlines other than `<< std::endl` in previous places
is usually more efficient.
--
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]
