tilelang.carver.arch.driver.ascend_driver
=========================================

.. py:module:: tilelang.carver.arch.driver.ascend_driver


Functions
---------

.. autoapisummary::

   tilelang.carver.arch.driver.ascend_driver.get_ascend_device_properties
   tilelang.carver.arch.driver.ascend_driver.get_device_name
   tilelang.carver.arch.driver.ascend_driver.get_num_cube_cores
   tilelang.carver.arch.driver.ascend_driver.get_num_vector_cores
   tilelang.carver.arch.driver.ascend_driver.get_total_memory
   tilelang.carver.arch.driver.ascend_driver.get_l2_cache_size


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

.. py:function:: get_ascend_device_properties(device_id = 0)

   Get raw torch_npu device properties object for the given device.

   Returns None if torch_npu or the NPU device is not available.


.. py:function:: get_device_name(device_id = 0)

   Get the device name (e.g. 'Ascend950PR_9589').


.. py:function:: get_num_cube_cores(device_id = 0)

   Get the number of AI Cube cores (matmul units) on the device.


.. py:function:: get_num_vector_cores(device_id = 0)

   Get the number of AI Vector cores (SIMD units) on the device.


.. py:function:: get_total_memory(device_id = 0, format = 'bytes')

   Get the total HBM memory of the device in bytes, kb, mb, or gb.


.. py:function:: get_l2_cache_size(device_id = 0, format = 'bytes')

   Get the L2 cache size of the device in bytes, kb, or mb.