Skip to main content
Back to top
Ctrl
+
K
TVM Homepage
Community
Download
Getting Started
Overview
Installing TVM
Install from PyPI
Install from Source
Docker Images
Quick Start
IRModule
TVM Errors
How To
Importing Models from ML Frameworks
End-to-End Optimize Model
Customize Optimization
Optimize Large Language Model
Cross Compilation and RPC
Export and Load Relax Executables
Mix Python/PyTorch with TVM Using BasePyModule
Bring Your Own Codegen
Deep Dive
Design and Architecture
Operator Fusion
Code Generation
External Library Dispatch (BYOC)
TVM Runtime System
Introduction to Module Serialization
Relax Virtual Machine
Pass Infrastructure
TVMScript
Device/Target Interactions
TensorIR
Tensor Program Abstraction
Understand TensorIR Abstraction
TensorIR Creation
Transformation
DLight: Rule-Based GPU Scheduling
MetaSchedule: Search-Based Auto-Tuning
Relax
Graph Abstraction for ML Models
Understand Relax Abstraction
Dataflow Pattern Language (DPL)
Relax Creation
Transformation
TIRx
Overview
Getting Started
Installation
Your first kernel
Programming Guide
CUDA Programming Guide
Defining a function
Data types and expressions
Tensors and memory
Control flow
CUDA C++/PTX intrinsics
Compiling and inspecting
Parser utilities
In-kernel profiling with CudaProfiler
Tensor Layout
Tile Operations
Explicit CUDA tile patterns
ld / st
ldmatrix / stmatrix
mov
add / sub / mul
div / rcp / sqrt
min / max
fma
cvt
ex2 / lg2
abs / neg
cp.async
cp.async.bulk: shared to cluster shared
cp.async.bulk.tensor / cp.reduce.async.bulk.tensor
tcgen05.ld / tcgen05.st
tcgen05.cp
mma.sync
tcgen05.mma
ewise: caller-defined scalar compositions
Reduction Instructions
Compiler Internals
TIRx lowering pipeline
Tensor Instruction Lowering
Backend Internals
API Reference
Core TVMScript
Tensor Instruction Authoring API
CUDA Authoring and Support APIs
CUDA backend configuration
Native CUDA compiler arguments
CUDA kernel configuration reference
Direct PTX Instructions
Trainium Authoring and Support APIs
Core TIRx IR
Layout IR
Thread Scope Enum Tags
Compilation
Compiler Transforms
Analysis
Tensor Instruction Registration
Backend Extension API
API Reference
Python API
tvm_tirx.ir
tvm_tirx.sym
tvm_tirx.ir.transform.instrument
tvm_tirx.ir.transform
tvm_tirx.target
tvm_tirx.codegen
tvm.driver
tvm_tirx.testing
tvm.exec
tvm.support
tvm_tirx.runtime
tvm.runtime.vm
tvm.runtime.disco
tvm.relax
tvm.relax.analysis
tvm.relax.backend
tvm.relax.block_builder
tvm.relax.distributed
tvm.relax.frontend
tvm.relax.op
tvm.relax.testing
tvm.relax.training
tvm.relax.transform
tvm.relax.dpl
tvm.s_tir.analysis
tvm.s_tir.schedule
tvm.s_tir.transform
tvm.s_tir.dlight
tvm.s_tir.backend
tvm.s_tir.tensor_intrin
tvm_tirx.script
tvm_tirx.script.parser
tvm_tirx.script.ir_builder
tvm_tirx.script.printer
tvm.te
tvm.topi
tvm.s_tir.meta_schedule
tvm.rpc
tvm.contrib
Other APIs
About
Contributor Guide
TVM Community Guidelines
Submit a Pull Request
Code Reviews
Committer Guide
Documentation
Code Guide and Tips
Testing TVM
Git Usage Tips
Using TVM’s CI
Release Process
Error Handling Guide
Publications
Security Guide
Index
Index
Repository
Show source
Suggest edit
Open issue
.rst
.pdf
Get Started
Get Started
#
IRModule
IRModule
Quick Start
Quick Start
