I was going to suggest using `MergeCompilerRegions`, but I saw you already using it. So I like your current approach. Sending many small functions to torch sounds like a non-trivial overhead, and I think " piece things back together into a graph" is essentially what `MergeCompilerRegions` does at Relay level.
-- You are receiving this because you are subscribed to this thread. Reply to this email directly or view it on GitHub: https://github.com/apache/tvm/pull/7401#issuecomment-919993664