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.layout

Contents

Functions

tvm_tirx.backend.trn.layout#

Trainium-specific TIRx layout helpers.

Functions#

is_trainium_layout(layout)

Return whether a layout uses only Trainium memory axes.

previous

tvm_tirx.backend.trn.axis.register

next

tvm_tirx.backend.trn.layout.is_trainium_layout

Contents

Functions

By Apache TVM Community