Code Organization

Header files in include are public APIs that share across modules. There can be internal header files within each module that sit in src.

Modules

  • arith: Arithmetic expression and set simplification.
  • auto_scheduler: The template-free auto-tuning module.
  • autotvm: The template-based auto-tuning module.
  • contrib: Contrib extension libraries.
  • driver: Compilation driver APIs.
  • ir: Common IR infrastructure.
  • node: The base infra for IR/AST nodes that is dialect independent.
  • relay: Relay IR, high-level optimizations.
  • runtime: Minimum runtime related codes.
  • support: Internal support utilities.
  • target: Hardware targets.
  • tir: Tensor IR, low-level optimizations.
  • te: Tensor expression DSL.
  • topi: Tensor Operator Inventory.