[GitHub] [tvm] electriclilies commented on a change in pull request #9848: [Relay] [Virtual Device] Store function result virtual device in virtual_device_ field

[GitBox <gi...@apache.org>]

electriclilies commented on a change in pull request #9848:
URL: https://github.com/apache/tvm/pull/9848#discussion_r793171304



##########
File path: src/printer/relay_text_printer.cc
##########
@@ -458,6 +458,7 @@ Doc RelayTextPrinter::PrintFunc(const Doc& prefix, const relay::Function& fn) {
   if (fn->ret_type.defined()) {
     doc << "-> " << Print(fn->ret_type) << " ";
   }
+  doc << "Virtual Device: " << Print(fn->virtual_device()) << " \n";

Review comment:
       Yeah, this was causing the parser to fail, I removed it so that it would pass.. I can add it back as a comment or just leave it out..




-- 
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: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org