I’m not particularly familiar with annotation.stop_fusion, other than that it seems like something that is introduced to block the FuseOps pass. My naive solution here would be to register annotation.stop_fusion as a supported operator for your codegen. You can then later remove/ignore it.
1 Like