Does tvm support one subgraph for Vitis AI?

Questions
#1

Hi Everyone, I would like to ask if there is a way to run more than one subgraph on DPU based on auto scheduling technique. As I can see in PYXIR that it does partitioning only for the largest subgraph of the deep learning model to run on the DPU.

Thank you in advance!

#2

It’s separate issues:

  • Vitis-AI now only supports one subgraph.
  • It’s not related to auto-scheduling. The subgraph being offloaded to Vitis-AI will be optimized (i.e., scheduled) by it instead of TVM.

Also cc @jtuyls