Skip to main content

Ctrl+K

tvm-tirx

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

.rst

tvm_tirx.backend.trn.script

Contents

Classes

tvm_tirx.backend.trn.script#

Trainium TVMScript namespaces.

Classes#

NKINamespace()

The NKI instructions submodule.

previous

tvm_tirx.backend.trn.pipeline.finalize_device_passes_trn

next

tvm_tirx.backend.trn.script.NKINamespace

Contents

Classes

By Apache TVM Community