Skip to content

Conversation

@Lunderberg
Copy link
Contributor

Prior to this commit, private non-primitive relax functions would be discarded by FuseTIR. If any calls to these functions exist, the resulting IRModule would be ill-formed. This commit updates FuseTIR so that it only applies updates to functions with attr::kPrimitive, and calls into those functions.

To retain backwards compatibility, DeadCodeElimination is applied as a post-processing step.

Prior to this commit, private non-primitive relax functions would be
discarded by `FuseTIR`.  If any calls to these functions exist, the
resulting `IRModule` would be ill-formed.  This commit updates
`FuseTIR` so that it only applies updates to functions with
`attr::kPrimitive`, and calls into those functions.

To retain backwards compatibility, `DeadCodeElimination` is applied as
a post-processing step.
@tqchen
Copy link
Member

tqchen commented Feb 14, 2024

cc @Hzfengsy

@Hzfengsy Hzfengsy merged commit 274c368 into apache:main Feb 14, 2024
@Lunderberg Lunderberg deleted the bugfix_fusetir_preserve_private_relax_functions branch February 14, 2024 15:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants