Hi,
I’ve been playing around with relax infrastructure for a while and I wanted to ask a couple questions related to the relay to relax translator and its design/future directions.
-
First of all, I see that the
relay_translator.py
is in thetvm/relax/testing
directory. Is this because therelay_translator
is a temporary solution until relax brings its own frontend converters and adopts most/if not all the relay optimizations. -
If its not meant for testing, is there a plan to make it a core part of relax and port it to a c++ implementation since relax is planned as an optional module for now and might stay that way for a while.
One last related doubt. I see that the PassPrefix
passes are run before translating to relax, but FuseOps is not run. Does that mean there’s a separate fusion pass in relax and if so, is there a plan to only work with relax fusion.
Thanks,
Anirudh