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
    • tvm_tirx.runtime
      • tvm_tirx.runtime.DataTypeCode
      • tvm_tirx.runtime.Device
      • tvm_tirx.runtime.MemoryScope
      • tvm_tirx.runtime.Tensor
      • tvm_tirx.runtime.load_module
      • tvm_tirx.runtime.tensor
      • tvm_tirx.runtime.empty
      • tvm_tirx.runtime.cpu
      • tvm_tirx.runtime.cuda
      • tvm_tirx.runtime.opencl
      • tvm_tirx.runtime.vulkan
      • tvm_tirx.runtime.metal
      • tvm_tirx.runtime.vpi
      • tvm_tirx.runtime.rocm
      • tvm_tirx.runtime.ext_dev
      • tvm_tirx.runtime.hexagon
      • tvm_tirx.runtime.enabled
      • tvm_tirx.runtime.num_threads
    • tvm_tirx.ir
    • tvm_tirx.ir.prim
    • tvm_tirx.ir.transform
    • tvm_tirx.sym
    • tvm_tirx.script
    • tvm_tirx.script.parser
    • tvm_tirx.script.ir_builder
    • tvm_tirx.target
    • tvm_tirx.tirx
    • tvm_tirx.tirx.op
    • tvm_tirx.tirx.analysis
    • tvm_tirx.tirx.transform
    • tvm_tirx.tirx.layout
    • tvm_tirx.tirx.tile
    • tvm_tirx.tirx.script
    • tvm_tirx.codegen
    • tvm_tirx.backend
    • tvm_tirx.backend.cuda
    • tvm_tirx.backend.cuda.op
    • tvm_tirx.backend.cuda.script
    • tvm_tirx.backend.cuda.layout
    • tvm_tirx.backend.cuda.lang
    • tvm_tirx.backend.cuda.transform
    • tvm_tirx.backend.rocm
    • tvm_tirx.backend.metal
    • tvm_tirx.backend.metal.op
    • tvm_tirx.backend.opencl
    • tvm_tirx.backend.adreno
    • tvm_tirx.backend.hexagon
    • tvm_tirx.backend.vulkan
    • tvm_tirx.backend.webgpu
    • tvm_tirx.backend.trn
    • tvm_tirx.backend.trn.op
    • tvm_tirx.backend.trn.transform
    • tvm_tirx.backend.rust
    • tvm_tirx.support
    • tvm_tirx.testing
    • tvm_ffi structural utilities
    • tvm_tirx.compile
  • C++ API
  • Rust API

Developer Manual

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

tvm_tirx.runtime.DataTypeCode

Contents

  • DataTypeCode
    • DataTypeCode.INT
    • DataTypeCode.UINT
    • DataTypeCode.FLOAT
    • DataTypeCode.HANDLE
    • DataTypeCode.BFLOAT
    • DataTypeCode.BOOL
    • DataTypeCode.Float8E3M4
    • DataTypeCode.Float8E4M3
    • DataTypeCode.Float8E4M3B11FNUZ
    • DataTypeCode.Float8E4M3FN
    • DataTypeCode.Float8E4M3FNUZ
    • DataTypeCode.Float8E5M2
    • DataTypeCode.Float8E5M2FNUZ
    • DataTypeCode.Float8E8M0FNU
    • DataTypeCode.Float6E2M3FN
    • DataTypeCode.Float6E3M2FN
    • DataTypeCode.Float4E2M1FN

tvm_tirx.runtime.DataTypeCode#

class tvm_tirx.runtime.DataTypeCode(*values)#

Bases: IntEnum

DLDataTypeCode code in DLTensor.

INT = 0#
UINT = 1#
FLOAT = 2#
HANDLE = 3#
BFLOAT = 4#
BOOL = 6#
Float8E3M4 = 7#
Float8E4M3 = 8#
Float8E4M3B11FNUZ = 9#
Float8E4M3FN = 10#
Float8E4M3FNUZ = 11#
Float8E5M2 = 12#
Float8E5M2FNUZ = 13#
Float8E8M0FNU = 14#
Float6E2M3FN = 15#
Float6E3M2FN = 16#
Float4E2M1FN = 17#

previous

tvm_tirx.runtime

next

tvm_tirx.runtime.Device

Contents
  • DataTypeCode
    • DataTypeCode.INT
    • DataTypeCode.UINT
    • DataTypeCode.FLOAT
    • DataTypeCode.HANDLE
    • DataTypeCode.BFLOAT
    • DataTypeCode.BOOL
    • DataTypeCode.Float8E3M4
    • DataTypeCode.Float8E4M3
    • DataTypeCode.Float8E4M3B11FNUZ
    • DataTypeCode.Float8E4M3FN
    • DataTypeCode.Float8E4M3FNUZ
    • DataTypeCode.Float8E5M2
    • DataTypeCode.Float8E5M2FNUZ
    • DataTypeCode.Float8E8M0FNU
    • DataTypeCode.Float6E2M3FN
    • DataTypeCode.Float6E3M2FN
    • DataTypeCode.Float4E2M1FN

By Apache TVM Community