Hi, @aca88 I believe your example would be taken care of using the Merge Composite pass before partitioning. You can imagine that after running this pass add+conv2d for the blue compiler would be represented by a single node. The partitioning would then happen as you described. However, this would force this result, there is no consideration for the 'optimal' partitioning for the time being.
Currently there is ongoing work in the graph partitioning area. I don't believe the new partitioning mechanism has support for Composite functions just yet, although this is something being looked into. @matt-arm @manupa-arm may be able to comment more. --- [Visit Topic](https://discuss.tvm.ai/t/relay-improved-graph-partitioning-algorithm/5830/18) 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/1614dc2692e11cbc5cf1a5a83b200c86caa5aec29ad3261fee4313eff0d0706c).