tilelang.utils.device
=====================

.. py:module:: tilelang.utils.device


Attributes
----------

.. autoapisummary::

   tilelang.utils.device.IS_CUDA
   tilelang.utils.device.IS_MPS


Functions
---------

.. autoapisummary::

   tilelang.utils.device.get_current_device


Module Contents
---------------

.. py:data:: IS_CUDA

.. py:data:: IS_MPS

.. py:function:: get_current_device()