
    (Th                     4    S SK r S SKr " S S5      rS rS rg)    Nc                   $    \ rS rSrSrS rS rSrg)TmaDescKernelParam      c           
         [         R                  " U R                  [         R                  SS9U l        [        U5      [        U5      :X  d   eS[        U5      s=::  a  S::  d   e   eU R                  R                  5       S-  S:X  d   e[        U5      S:X  a_  [        R                  R                  R                  R                  R                  XS   US   UU R                  R                  5       5        g [        R                  R                  R                  R                  R                  XS   US   US   US   X@R                  R                  5       5        g )Ncpu)dtypedevice      @   r   )torchemptyTMA_DESC_SIZEuint8desclendata_ptrtritonruntimedriveractiveutilsfill_1d_tma_descriptorfill_2d_tma_descriptor)selfptrdims
block_dimselement_sizes        \/var/www/auris/envauris/lib/python3.13/site-packages/triton/tools/experimental_descriptor.py__init__TmaDescKernelParam.__init__	   s    KK 2 2%++eT	4yC
O+++CI""""""yy!!#b(A---t9>NN!!((..EEcPQ7T^_`TacoFJiiFXFXFZ\ NN!!((..EEcPQ7TXYZT[]ghi]jFPQRmUaclclcucucwy    c                 6    U R                   R                  5       $ N)r   r   )r   s    r!   tma_desc_cpu_ptr#TmaDescKernelParam.tma_desc_cpu_ptr   s    yy!!##r$   )r   N)__name__
__module____qualname____firstlineno__r   r"   r'   __static_attributes__ r$   r!   r   r      s    My$r$   r   c                      [        X/U/U5      $ r&   r   )r   dim	block_dimr    s       r!   create_1d_tma_descriptorr3      s    c59+|DDr$   c                 "    [        XU/X4/U5      $ r&   r0   )r   dim1dim0
block_dim1
block_dim0r    s         r!   create_2d_tma_descriptorr9      s    c$<*1I<XXr$   )r   r   r   r3   r9   r.   r$   r!   <module>r:      s!     $ $*EYr$   