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

File device_api.h

Contents

  • Definition (tvm/runtime/device_api.h)
  • Includes
  • Included By
  • Namespaces
  • Classes
  • Enums
  • Functions
  • Typedefs
  • Variables

File device_api.h#

↰ Parent directory (tvm/runtime)

Abstract device memory management API.

Contents

  • Definition (tvm/runtime/device_api.h)

  • Includes

  • Included By

  • Namespaces

  • Classes

  • Enums

  • Functions

  • Typedefs

  • Variables

Definition (tvm/runtime/device_api.h)#

  • Program Listing for File device_api.h

Includes#

  • string

  • tvm/ffi/any.h

  • tvm/ffi/container/tensor.h

  • tvm/ffi/device.h

  • tvm/ffi/error.h

  • tvm/ffi/optional.h

  • tvm/ffi/string.h

  • tvm/runtime/base.h (File base.h)

  • tvm/runtime/memory_scope.h (File memory_scope.h)

Included By#

  • File tensor.h

  • File doc.h

  • File target.h

  • File type.h

Namespaces#

  • Namespace tvm

  • Namespace tvm::runtime

Classes#

  • Class DeviceAPI

Enums#

  • Enum TVMDeviceExtType

Functions#

  • Function tvm::runtime::DLDeviceType2Str

  • Function tvm::runtime::operator<<(std::ostream&, DLDevice)

  • Function tvm::runtime::RuntimeEnabled

Typedefs#

  • Typedef tvm::Device

  • Typedef TVMStreamHandle

Variables#

  • Variable tvm::runtime::kAllocAlignment

  • Variable tvm::runtime::kMaxStackAlloca

  • Variable tvm::runtime::kTempAllocaAlignment

Contents
  • Definition (tvm/runtime/device_api.h)
  • Includes
  • Included By
  • Namespaces
  • Classes
  • Enums
  • Functions
  • Typedefs
  • Variables

By Apache TVM Community