
    !&h                     4    d dl Z d dlZ G d d      Zd Zd Zy)    Nc                       e Zd ZdZd Zd Zy)TmaDescKernelParam   c           
         t        j                  | j                  t         j                  d      | _        t        |      t        |      k(  sJ dt        |      cxk  rdk  sJ  J | j                  j                         dz  dk(  sJ t        |      dk(  r`t        j                  j                  j                  j                  j                  ||d   |d   || j                  j                                y t        j                  j                  j                  j                  j                  ||d   |d   |d   |d   || j                  j                                y )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        z/var/www/pru.catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/triton/tools/experimental_descriptor.py__init__zTmaDescKernelParam.__init__	   s'   KK 2 2%++eT	4yC
O+++CI""""""yy!!#b(A---t9>NN!!((..EEc4PQ7T^_`TacoFJiiFXFXFZ\ NN!!((..EEc4PQ7TXYZT[]ghi]jFPQRmUacgclclcucucwy    c                 6    | j                   j                         S N)r   r   )r   s    r    tma_desc_cpu_ptrz#TmaDescKernelParam.tma_desc_cpu_ptr   s    yy!!##r"   N)__name__
__module____qualname__r   r!   r%    r"   r    r   r      s    My$r"   r   c                 "    t        | |g|g|      S r$   r   )r   dim	block_dimr   s       r    create_1d_tma_descriptorr.      s    cC59+|DDr"   c                 &    t        | ||g||g|      S r$   r+   )r   dim1dim0
block_dim1
block_dim0r   s         r    create_2d_tma_descriptorr4      s    cD$<*j1I<XXr"   )r   r   r   r.   r4   r)   r"   r    <module>r5      s!     $ $*EYr"   