electriclilies opened a new pull request #8914: URL: https://github.com/apache/tvm/pull/8914
In this PR, I add `InferType` to the end of `LowerTEPass` so we don't have to call it after we call `LowerTEPass`. I also remove `per_target_modules_` from the interpreter. To do this, I did have to fix a few bugs, namely: - Make sure we copy the attributes of the `IRModule` when we soft-copy the `IRModule` - In the type inferencer, we lookup `GlobalVars` by name, not by pointer address - In the type inferencer, if we encounter a `tir::PrimFunc`, we the function it instead of failing (this allows us to call InferType on the unified module which contains both `relay::Functions` and `tir::PrimFuncs`) This is forked from #8886, so there are also extraneous changes in here. They will go away after that is merged and I rebase. -- 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. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
