tvm
|
▼ include | |
▼ tvm | |
► arith | |
analyzer.h | Algebra expression simplifications |
bound.h | Bound deducers |
int_set.h | Integer set |
int_solver.h | Integer constraints data structures and solvers |
iter_affine_map.h | Iterator quasi-affine mapping patterns |
pattern.h | Expression pattern detectors |
► ir | |
analysis.h | |
attrs.h | Helpers for attribute objects |
diagnostic.h | A new diagnostic interface for TVM error reporting |
env_func.h | Serializable global function used in IR |
expr.h | Base expr nodes in TVM |
function.h | Function nodes |
global_info.h | GlobalInfo are globally static object that are referred by the IR itself |
global_var_supply.h | GlobalVarSupply that can be used to generate unique |
instrument.h | |
module.h | IRModule that holds the functions and type definitions |
name_supply.h | NameSupply that can be used to generate unique variable names |
op.h | Primitive operators(builtin intrinsics) and registry for them |
replace_global_vars.h | A utility to replace GlobalVar instances across all TVM IR types in an IRMdoule |
source_map.h | A map from source names to source code |
transform.h | |
type.h | IR/AST nodes for the unified type system in TVM |
type_functor.h | A way to defined arbitrary function signature with dispatch on types |
► meta_schedule | |
► schedule | |
► cuda | |
thread_bind.h | |
► generic | |
winograd.h | |
arg_info.h | |
builder.h | |
cost_model.h | |
database.h | |
extracted_task.h | |
feature_extractor.h | |
measure_callback.h | |
measure_candidate.h | |
mutator.h | |
postproc.h | |
profiler.h | |
runner.h | |
schedule_rule.h | |
search_strategy.h | |
space_generator.h | |
task_scheduler.h | |
tune_context.h | |
► node | |
attr_registry_map.h | Attribute map used in registry |
cast.h | Value casting helpers |
functor.h | Defines the Functor data structures |
node.h | Definitions and helper macros for IR/AST nodes |
reflection.h | Reflection utilities for IR/AST nodes |
repr_printer.h | Printer class to print repr string of each AST/IR nodes |
script_printer.h | |
serialization.h | |
structural_equal.h | Structural equality comparison |
structural_hash.h | |
► relax | |
► attrs | |
ccl.h | Attributes for ccl operators |
create.h | Attributes for tensor creation operators |
datatype.h | Attributes for datatype operators |
distributed.h | Attributes for redistribute and annotate_sharding operators |
image.h | Attributes for image operators |
index.h | Attributes for indexing operators |
linear_algebra.h | Attributes for linear algebra operators |
manipulate.h | Attributes for tensor manipulation operators |
nn.h | Attributes for neural network operators |
op.h | Attributes for relax specific operators |
qdq.h | Attributes for quantize/dequantize operators |
sampling.h | Attributes for sampling operators |
search.h | Attributes for search operators |
sorting.h | Attributes for sorting operators |
statistical.h | Attributes for statistical operators |
► distributed | |
axis_group_graph.h | |
global_info.h | Data structure for distributed inference |
struct_info.h | Struct info for DTensor (Distributed Tensor) |
transform.h | Relax distributed specific transformation passes |
analysis.h | The set of Relax specific analysis on IR |
backend.h | Relax backend specific transformation passes |
binding_rewrite.h | An IR rewriter to easily add/remove/replace bindings (statements) |
block_builder.h | The utility for constructing Relax binding blocks |
dataflow_matcher.h | A pattern matcher for matching dataflow properties |
dataflow_pattern.h | A pattern language for matching dataflow properties |
dataflow_pattern_functor.h | Functors and visitors for dataflow patterns |
exec_builder.h | |
expr.h | |
expr_functor.h | A more powerful visitor which enables defining arbitrary function signatures with type based dispatch on first argument |
nested_msg.h | Helper container to store nested message for robust tuple-aware analysis |
op_attr_types.h | Data structures that can appear in operator attributes |
struct_info.h | |
struct_info_functor.h | Functors and visitors for struct info |
tir_pattern.h | Data Structure of TIR Pattern used for matching |
transform.h | Relax specific transformation passes |
type.h | Relax Types |
utils.h | Utility classes and functions for working with the Relax IR |
► runtime | |
► contrib | |
papi.h | |
► disco | |
builtin.h | |
cuda_ipc_memory.h | |
disco_worker.h | This file defines a worker in Disco. A worker can be launched in a separate thread or process as long as the channel supports bi-directional communication in-between the worker and the controler |
session.h | This file serves as the entry point of Disco and defines key data structures and interfaces |
► memory | |
memory_manager.h | Abstract device memory management API |
► vm | |
builtin.h | Builtin runtime APIs |
bytecode.h | The bytecode for the virtual machine |
executable.h | |
ndarray_cache_support.h | |
vm.h | |
base.h | |
builtin_fp16.h | Functions for conversion between fp32 and fp16 |
c_backend_api.h | TVM runtime backend API |
data_type.h | |
device_api.h | Abstract device memory management API |
int_tuple.h | Defines tuple of integers |
module.h | Runtime container of the functions generated by TVM, This is used to support dynamically link, load and save functions from different convention under unified API |
ndarray.h | A device-independent managed NDArray abstraction |
nvtx.h | |
object.h | A managed object in the TVM runtime |
packed_func.h | Type-erased function used across TVM API |
profiling.h | Runtime profiling including timers |
serializer.h | Serializer extension to support TVM data types Include this file to enable serialization of DLDataType, DLDevice |
threading_backend.h | Utilities for manipulating thread pool threads |
► script | |
► ir_builder | |
► ir | |
frame.h | |
ir.h | |
► relax | |
frame.h | |
ir.h | |
► tir | |
frame.h | |
ir.h | |
base.h | |
► printer | |
doc.h | |
ir_docsifier.h | |
ir_docsifier_functor.h | |
► support | |
parallel_for.h | An implementation to run loop in parallel |
random_engine.h | Random number generator. It provides a generic interface consistent with std::uniform_random_bit_generator |
with.h | RAII wrapper function to enter and exit a context object similar to python's with syntax |
► target | |
codegen.h | Translates IRModule to runtime::Module |
tag.h | Target tag registry |
target.h | Compilation target object |
target_info.h | Various information about target |
target_kind.h | Target kind registry |
virtual_device.h | A compile time representation for where data is to be stored at runtime, and how to compile code to compute it |
► te | |
operation.h | Operation node can generate one or multiple Tensors |
tensor.h | Dataflow tensor object |
► tir | |
► schedule | |
instruction.h | |
schedule.h | |
state.h | This file defines ScheduleState, the core data structure of TensorIR scheduling |
trace.h | |
analysis.h | Analysis utilities and passes for TIR |
block_dependence_info.h | Define BlockDependenceInfoNode that uses the BlockScope and StmtSRef objects to store the block level dependences |
block_scope.h | Definition of two pillar data structure for TensorIR scheduling: StmtSRef, BlockScope |
buffer.h | Symbolic n-dimensional array, to represent a memory buffer |
builtin.h | TIR builtin intrinsics |
data_layout.h | Layout expression to describe the data organization of a tensor. And BijectiveLayout to mapping two data layouts between each other |
data_type_rewriter.h | Rewrite the data type of expressions |
expr.h | TIR expressions |
expr_functor.h | Functors for tir expressions |
function.h | TIR Function |
index_map.h | Defines a remapping of buffer indices |
op.h | Common operators defined for Expr |
op_attr_types.h | Attribute types in the Op registry for TIR ops |
stmt.h | TIR statements |
stmt_functor.h | Functors for tir stmts utility functions to call common functors |
transform.h | TIR specific transformation passes |
utils.h | |
var.h | Variables in the TIR |
► topi | |
► contrib | |
cublas.h | External function interface to cuBLAS libraries |
rocblas.h | |
► detail | |
array_utils.h | Utility functions for handling arrays |
broadcast.h | Detail broadcast |
constant_utils.h | Utility functions for handling constants in TVM expressions |
extern.h | Helpers for using external functions |
fuse.h | Fuse operation |
pad_utils.h | Padding helpers |
ravel_unravel.h | Index ravel and unraval operations |
strided_slice.h | Utility functions for strided_slice op |
tensor_utils.h | Utility functions for handling tensor |
► nn | |
bias_add.h | Bias_add op constructions |
bnn.h | Binary op constructions |
dense.h | Dense op constructions |
dilate.h | Dilate op constructions |
flatten.h | Softmax op constructions |
group_norm.h | Group normalization op constructions |
instance_norm.h | Instance normalization op constructions |
layer_norm.h | Layer normalization op constructions |
local_response_norm.h | Local response normalization op constructions |
mapping.h | Mapping op constructions |
pooling.h | Pooling op constructions |
rms_norm.h | Root mean square normalization op constructions |
softmax.h | Softmax op constructions |
► vision | |
reorg.h | Reorg op constructions |
broadcast.h | Broadcast op constructions |
einsum.h | Einstein summation op |
elemwise.h | Elementwise op constructions |
nn.h | NN op constructions |
reduction.h | Reduction op constructors |
tags.h | External function interface to rocBLAS libraries |
transform.h | Transform op constructors |
utils.h | Topi utility function |