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

  • support: Internal support utilities.
  • runtime: Minimum runtime related codes.
  • node: base infra for IR/AST nodes that is dialect independent.
  • ir: Common IR infrastructure.
  • tir: Tensor-level IR.
  • te: tensor expression DSL
  • arith: Arithmetic expression and set simplification.
  • relay: Relay IR, high-level optimization.
  • autotvm: The auto-tuning module.
  • contrib: Contrib extension libraries.
  • api: API function registration.
  • driver: Compilation driver APIs.