-
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
[Relay][VM] Fix constant folding issue in VM compiler #4077
Conversation
Is the performance now caused by pass or vm related things, like dispatching and memory allocation? |
e622648
to
b11b9d2
Compare
Yes. Most of the time spent other than packed function execution are |
1. allow pass params when compile a module 2. enhance profiler robustness
8af82d3
to
9543ede
Compare
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.
LGTM
* [Relay][VM] Fix constant folding issue in VM compiler 1. allow pass params when compile a module 2. enhance profiler robustness * remove dead code * fix lint * add get_params * fix test * don't pass params back * remove get_params * docs * move compile function to api * compile clashes with builtin name * fix compilation error * remove dead code
* [Relay][VM] Fix constant folding issue in VM compiler 1. allow pass params when compile a module 2. enhance profiler robustness * remove dead code * fix lint * add get_params * fix test * don't pass params back * remove get_params * docs * move compile function to api * compile clashes with builtin name * fix compilation error * remove dead code
This speeds up mxnet resnset18 by 4x because it fixes constant folding. Now vm is just
1.4x slower than relay graph runtime.
cc @jroesch @zhiics @icemelon9