[Fix][Transform] FuseOps handling multiple src functions - Githubissues

tlc-pack / relax

Apache License 2.0

193 stars 58 forks source link

[Fix][Transform] FuseOps handling multiple src functions #427

Closed MasterJH5574 closed 1 year ago

MasterJH5574 commented 1 year ago

This PR fixes FuseOps so that it can now handle muitlple Relax functions instead of only doing fusion in function "main."

Specifically, this PR

fixes FuseOps as described above.
adds a unit test to FuseOps to ensure it works for multiple src functions.
adds a unit test to FuseTIR to ensure it works for multiple src functions.
removes the internal IsLeaf funciton in FuseOps as it duplicates with the existing util.