jtuyls commented on a change in pull request #7577:
URL: https://github.com/apache/tvm/pull/7577#discussion_r589327224
##########
File path: python/tvm/driver/tvmc/composite_target.py
##########
@@ -22,7 +22,7 @@
from tvm.relay.op.contrib.arm_compute_lib import partition_for_arm_compute_lib
from tvm.relay.op.contrib.ethosn import partition_for_ethosn
from tvm.relay.op.contrib.vitis_ai import partition_for_vitis_ai
-from tvm.contrib.target import vitis_ai
+from tvm.contrib.target import vitis_ai # pylint: disable=unused-import
Review comment:
The `relay.ext.vitis_ai` codegen is registered there and won't be found
if we don't import the module:
https://github.com/apache/tvm/blob/5d5bbfb1ed2281c3e9b2719be6643281d60022fb/python/tvm/contrib/target/vitis_ai.py#L64
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]