How does TOPI interact with other modules in the TVM stack (ie. Relay)?

Questions
2

First some clarifying definitions:

  • Relay/NNVM-V2: is the graph IR doing graph-level optimizations.
  • TVM: is a tensor DSL providing primitive for defining operators and how to schedule them.
  • TOPI: defines operators and scheduling mechanism using DSL provided by TVM. It’s primarily used for AutoTVM. In fact, NNVM (old) and new Relay call into TOPI.

tvm-stack
tvm-stack563×500 48.4 KB

You may find more details in

  1. Relay RFC
  2. Original NNVM design before merging into TVM
2 Likes