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.testing.env.has_gpu

Contents

has_gpu()

tvm_tirx.testing.env.has_gpu#

tvm_tirx.testing.env.has_gpu()#

True if any GPU backend (cuda/rocm/opencl/metal/vulkan) is present.

Return type:: bool

previous

tvm_tirx.testing.env.has_llvm

next

tvm_tirx.testing.env.has_multi_gpu

Contents

has_gpu()

By Apache TVM Community