[PR] [Relax] Remove name requirement from MergeCompositeFunctions [tvm]

["Lunderberg (via GitHub)" <gi...@apache.org>]

Lunderberg opened a new pull request, #17212:
URL: https://github.com/apache/tvm/pull/17212

   Prior to this commit, the `relax.transform.MergeCompositeFunctions` required the module to contain a single Relax function named `"main"`. Lack of this function would result in an error when `mod->GetGlobalVar("main")` was called.  Presence of any other Relax function would cause an error, since they were not collected into the `CompositeGroupsBuilder`.
   
   This commit updates `MergeCompositeFunctions` to be indepedendent of the names of the Relax functions in an IRModule.  The transform now updates all Relax functions that do not have the `attr::kPrimitive` or `attr::kCodegen` attributes.
   
   Closes https://github.com/apache/tvm/issues/17210


-- 
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-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org