jtuyls commented on a change in pull request #7577:
URL: https://github.com/apache/tvm/pull/7577#discussion_r591478453
##########
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:
Yes, I am thinking of adding `init` key with corresponding
initialization function like for example:
```
"vitis-ai": {
"config_key": "relay.ext.vitis_ai.options",
"init": init_for_vitis_ai,
"pass_pipeline": partition_for_vitis_ai,
},
```
I will work on something and add it to the PR. Are you ok with "init" or do
you prefer some other 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.
For queries about this service, please contact Infrastructure at:
[email protected]
