vinx13 commented on a change in pull request #9306:
URL: https://github.com/apache/tvm/pull/9306#discussion_r734757408
##########
File path: src/printer/tvmscript_printer.cc
##########
@@ -1049,15 +1068,24 @@ Doc TVMScriptPrinter::PrintBlockBody(const BlockNode*
op) {
return body;
}
-Doc TVMScriptPrinter::VisitStmt_(const BlockRealizeNode* op) {
- const auto* block_op = op->block.as<BlockNode>();
- // print block name and block vars
- Doc doc;
+/*!
+ * \brief Print the name of a block
+ * \param block_op The block node to be printed
+ */
+Doc TVMScriptPrinter::PrintBlockName(const BlockNode* block_op) {
+ Doc doc = PrintOptionalInfo(GetRef<Stmt>(block_op));
Review comment:
move this to `TVMScriptPrinter::VisitStmt_(const BlockRealizeNode* op)`,
as this is not related to block name
--
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]
