Lunderberg commented on PR #16843: URL: https://github.com/apache/tvm/pull/16843#issuecomment-2039658387
Yup, `relax::ExternFunc` is a subclass of `BaseFunc`, and can be stored directly in an `IRModule`. At some point, I want to use this to specify when the lookup of a `PackedFunc` in the global registry occurs. Currently, there's a discrepency between Relax and TIR, where Relax performs the lookup when the VirtualMachine is initialized, while TIR performs the lookup just before the call is performed. I'd like to make these more similar, with an `ExternFunc` being resolved either at initialization (declared in IRModule), at the start of a function (ExternFunc bound to a variable), or at the callsite (ExternFunc used inline). But the main impetus for that would have been the `LazyTransformParams` pass, and with the `LazyGetInput` accepting a callback parameter, there isn't as much need for it. -- 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: commits-unsubscr...@tvm.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
