|
| | analysis.h |
| | Analysis utilities for Schedulable TensorIR (S-TIR).
|
| |
| | data_layout.h |
| | SLayout expression to describe the data organization of a tensor. And SBijectiveLayout to mapping two data layouts between each other.
|
| |
| | function.h |
| |
| | iter_var.h |
| | Axis metadata for schedulable tensor computations.
|
| |
| | random_engine.h |
| | Random number generator. It provides a generic interface consistent with std::uniform_random_bit_generator
|
| |
| | sblock_dependence_info.h |
| | Define SBlockDependenceInfoObj that uses the SBlockScope and StmtSRef objects to store the block level dependences.
|
| |
| | sblock_scope.h |
| | Definition of two pillar data structure for TensorIR scheduling: StmtSRef, SBlockScope.
|
| |
| | stmt.h |
| | S-TIR (Schedulable TIR) statements and attributes.
|
| |
| | stmt_functor.h |
| | Native traversal and statement dispatch for schedulable TIR.
|
| |
| | stmt_functor_with_analyzer.h |
| | Statement and expression functors with an analyzer context.
|
| |
| | tensor_intrin.h |
| | Tensor intrinsics for schedulable TIR.
|
| |
| | transform.h |
| | S-TIR specific transformation passes.
|
| |
| | utils.h |
| |
