-
Notifications
You must be signed in to change notification settings - Fork 3.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[TVMC] use target_host when it is set #6855
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I like this change.Could you add a test to verify this behavior?
@euntaik please address the review comments. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just a nit.
The test case looks not relevant at the first glance, but I guess maybe OpenCL is the more obvious backend that requires target host to be set explicitly?
# check for output types | ||
assert type(graph) is str | ||
assert type(lib) is tvm.runtime.module.Module | ||
assert type(params) is dict | ||
assert type(dumps) is dict |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Those checks look not cover anything special? To me, as long as you can compile the model without errors, then you should have used the right target host.
* [TVMC] add cl support in tvmc runner * [TVMC] use target_host when it is set * Cleanup comment and asssert device type in else case * add a test for tvmc compiler * remove unused func
* [TVMC] add cl support in tvmc runner * [TVMC] use target_host when it is set * Cleanup comment and asssert device type in else case * add a test for tvmc compiler * remove unused func
* [TVMC] add cl support in tvmc runner * [TVMC] use target_host when it is set * Cleanup comment and asssert device type in else case * add a test for tvmc compiler * remove unused func
Feed target_host to the relay.build when it is provided.