Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
Tile Language
0.1.6.post1 documentation
Light Logo Dark Logo
Tile Language
0.1.6.post1 documentation

GET STARTED

  • Installation Guide
  • The Tile Language: A Brief Introduction

TUTORIALS

  • Debugging Tile Language Programs
  • Auto-Tuning Techniques for Performance Optimization

DEEP LEARNING OPERATORS

  • ElementWise Operators
  • General Matrix-Vector Multiplication (GEMV)
  • General Matrix-Matrix Multiplication with Tile Library
  • 🚀 Write High Performance FlashMLA with TileLang on Hopper

COMPILER INTERNALS

  • LetStmt Inlining in TileLang
  • InjectFenceProxy Pass

API Reference

  • tilelang
    • tilelang.autotuner
      • tilelang.autotuner.capture
      • tilelang.autotuner.param
      • tilelang.autotuner.tuner
    • tilelang.cache
      • tilelang.cache.kernel_cache
    • tilelang.carver
      • tilelang.carver.analysis
      • tilelang.carver.arch
        • tilelang.carver.arch.arch_base
        • tilelang.carver.arch.cdna
        • tilelang.carver.arch.cpu
        • tilelang.carver.arch.cuda
        • tilelang.carver.arch.driver
          • tilelang.carver.arch.driver.cuda_driver
        • tilelang.carver.arch.metal
      • tilelang.carver.common_schedules
      • tilelang.carver.matmul_analysis
      • tilelang.carver.roller
        • tilelang.carver.roller.bestfit
        • tilelang.carver.roller.hint
        • tilelang.carver.roller.node
        • tilelang.carver.roller.policy
          • tilelang.carver.roller.policy.common
          • tilelang.carver.roller.policy.default
          • tilelang.carver.roller.policy.tensorcore
        • tilelang.carver.roller.rasterization
        • tilelang.carver.roller.shape_inference
          • tilelang.carver.roller.shape_inference.common
          • tilelang.carver.roller.shape_inference.tir
      • tilelang.carver.template
        • tilelang.carver.template.base
        • tilelang.carver.template.conv
        • tilelang.carver.template.elementwise
        • tilelang.carver.template.flashattention
        • tilelang.carver.template.gemv
        • tilelang.carver.template.general_reduce
        • tilelang.carver.template.matmul
      • tilelang.carver.utils
    • tilelang.common
      • tilelang.common.transform_kind
    • tilelang.contrib
      • tilelang.contrib.cc
      • tilelang.contrib.dlpack
      • tilelang.contrib.hipcc
      • tilelang.contrib.nvcc
      • tilelang.contrib.nvrtc
      • tilelang.contrib.rocm
    • tilelang.engine
      • tilelang.engine.callback
      • tilelang.engine.lower
      • tilelang.engine.param
      • tilelang.engine.phase
    • tilelang.env
    • tilelang.intrinsics
      • tilelang.intrinsics.mfma_layout
      • tilelang.intrinsics.mfma_macro_generator
      • tilelang.intrinsics.mma_layout
      • tilelang.intrinsics.mma_macro_generator
      • tilelang.intrinsics.utils
      • tilelang.intrinsics.wgmma_macro_generator
    • tilelang.ir
    • tilelang.jit
      • tilelang.jit.adapter
        • tilelang.jit.adapter.base
        • tilelang.jit.adapter.ctypes
          • tilelang.jit.adapter.ctypes.adapter
        • tilelang.jit.adapter.cython
          • tilelang.jit.adapter.cython.adapter
        • tilelang.jit.adapter.dlpack
        • tilelang.jit.adapter.libgen
        • tilelang.jit.adapter.nvrtc
          • tilelang.jit.adapter.nvrtc.adapter
        • tilelang.jit.adapter.torch
          • tilelang.jit.adapter.torch.metal
        • tilelang.jit.adapter.utils
        • tilelang.jit.adapter.wrapper
      • tilelang.jit.env
      • tilelang.jit.kernel
      • tilelang.jit.param
    • tilelang.language
      • tilelang.language.allocate
      • tilelang.language.atomic
      • tilelang.language.builtin
      • tilelang.language.copy
      • tilelang.language.customize
      • tilelang.language.experimental
        • tilelang.language.experimental.gemm_sp
      • tilelang.language.fastmath
      • tilelang.language.fill
      • tilelang.language.frame
      • tilelang.language.gemm
      • tilelang.language.kernel
      • tilelang.language.logical
      • tilelang.language.math_intrinsics
      • tilelang.language.overrides
      • tilelang.language.parallel
      • tilelang.language.persistent
      • tilelang.language.pipeline
      • tilelang.language.print
      • tilelang.language.proxy
      • tilelang.language.reduce
      • tilelang.language.tir
        • tilelang.language.tir.entry
        • tilelang.language.tir.ir
        • tilelang.language.tir.op
      • tilelang.language.utils
      • tilelang.language.warpgroup
    • tilelang.layout
      • tilelang.layout.fragment
      • tilelang.layout.gemm_sp
      • tilelang.layout.layout
      • tilelang.layout.swizzle
    • tilelang.math
    • tilelang.primitives
      • tilelang.primitives.gemm
        • tilelang.primitives.gemm.base
        • tilelang.primitives.gemm.gemm_mma
    • tilelang.profiler
      • tilelang.profiler.bench
    • tilelang.quantize
      • tilelang.quantize.lop3
      • tilelang.quantize.mxfp
      • tilelang.quantize.quantization
      • tilelang.quantize.utils
    • tilelang.testing
    • tilelang.tileop
      • tilelang.tileop.gemm
        • tilelang.tileop.gemm.gemm_base
        • tilelang.tileop.gemm.gemm_mma
        • tilelang.tileop.gemm.gemm_wgmma
    • tilelang.tools
      • tilelang.tools.Analyzer
      • tilelang.tools.plot_layout
    • tilelang.transform
      • tilelang.transform.add_bufstore_wrapper
      • tilelang.transform.pass_config
      • tilelang.transform.simplify
    • tilelang.utils
      • tilelang.utils.deprecated
      • tilelang.utils.device
      • tilelang.utils.language
      • tilelang.utils.sparse
      • tilelang.utils.target
      • tilelang.utils.tensor

Privacy

  • Privacy
Back to top
View this page

tilelang.utils.device¶

Attributes¶

IS_CUDA

IS_MPS

Functions¶

get_current_device()

Module Contents¶

tilelang.utils.device.IS_CUDA¶
tilelang.utils.device.IS_MPS¶
tilelang.utils.device.get_current_device()¶
Next
tilelang.utils.language
Previous
tilelang.utils.deprecated
Copyright © 2025-2025, Tile Lang Contributors
Made with Sphinx and @pradyunsg's Furo
On this page
  • tilelang.utils.device
    • Attributes
    • Functions
    • Module Contents
      • IS_CUDA
      • IS_MPS
      • get_current_device()