|
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 | |
| tensor_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 |
| 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 |
| tensor.h | A device-independent managed Tensor abstraction |
| 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 |
1.9.1