Apache TVM TIRx Documentation

Welcome to the documentation for TVM TIRx. You can get started by reading the Get Started section, then reading through the other sections.

Installation

To try out TVM TIRx, install it via pip or uv:

pip install --upgrade --force-reinstall apache-tvm-tirx
# add the cuda-bindings dependency:
# pip install --upgrade --force-reinstall "apache-tvm-tirx[cuda]"

Table of Contents

Get Started

• Quick Start

IR Guides

• IR Overview
• Types and Expressions
• Statements
• Operators, Calls, and Regions
• Functions and Modules
• Printing and Parsing
• Structural Traversal, Rewriting, and Comparison
• Symbolic Integer Analysis
• Working with IR in C++
• Working with IR in Rust

Programming Guides

• Programming Overview
• Language Basics
• Control Flow
• Types, Tensors, and Memory
• Metaprogramming
• Calling CUDA and PTX
• Data Layout
• Tile Operations
• Compiling and Inspecting
• Profiling with IKET

Extending TIRx: everything is a plugin

• Getting Started with Rust Extensions
• Defining Types, Operators, and Passes
• Using and Compiling the Extension

Reference

• Python API
• C++ API
• Rust API

Developer Manual

• Build from Source
• Build This Doc Site
• CUDA Backend Development
• Reproduce CI/CD
• Release Process