mbs-octoml opened a new pull request, #12085: URL: https://github.com/apache/tvm/pull/12085
Device planning is halfway through the transition to using the virtual_device_ field on every expression node to capture device/target/etc info. In the meantime it is necessary to derive from a 'device aware' visitor so as to track device information. In Collage this is not feasible, so as a stop gap allow the map from expression nodes to virtual devices to be reconstructed as a stand alone map. This code can be removed once expr->virtual_device() is the canonical representation. -- 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]
