I was playing around with the new external codegen & composite functions and noticed this flow fails inside of AnnotateTarget. Wondering if it is because of this [TODO](https://github.com/apache/incubator-tvm/blob/021213832cb98703dda54f631215ac17fbabff7b/src/relay/transforms/annotate_target.cc#L70) ``` mod = tvm.IRModule.from_expr(f) mod = MergeComposite(pattern_table)(mod) mod = AnnotateTarget("my_external_compiler")(mod) mod = PartitionGraph()(mod) ```
``` TVMError: Check failed: ref->template IsInstance<typename SubRef: :ContainerType>(): Downcast from relay.Function to relay.Op failed. ``` I'm interested in offloading some subgraphs to my external codegen, lets say Conv2d+bias+relu. Is this the right mechanism to use? I can annotate individual ops for external codegen if I skip MergeComposite, but after this I'm not sure how to merge the annotated ops. Thanks @zhiics @comaniac @matt-arm --- [Visit Topic](https://discuss.tvm.ai/t/external-codegen-status-of-annotating-composite-functions/6150/1) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/f8adbd3d3700b160627378bdd48bf228f78d306a4375df58e2c9118d0e5ea5ae).
