README.md 816 Bytes
Newer Older
1
# Code Organization
tqchen committed
2

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

6
## Modules
Siva committed
7
- common: Internal common utilities.
8 9 10 11
- api: API function registration.
- lang: The definition of DSL related data structure.
- arithmetic: Arithmetic expression and set simplification.
- op: The detail implementations about each operation(compute, scan, placeholder).
Siva committed
12
- schedule: The operations on the schedule graph before converting to IR.
13
- pass: The optimization pass on the IR structure.
Siva committed
14
- codegen: The code generator.
15 16 17 18
- runtime: Minimum runtime related codes.
- autotvm: The auto-tuning module.
- relay: Implementation of Relay. The second generation of NNVM, a new IR for deep learning frameworks.
- contrib: Contrib extension libraries.