Understanding TVM/Relay's PartitionGraph()(mod) function

The recent PR should fix this:

See this unit test:

https://github.com/apache/incubator-tvm/blob/main/tests/python/relay/test_pass_annotate_target.py#L358

1 Like