Ë
    7^(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        úb/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/triton/tools/experimental_descriptor.pyÚ__init__zTmaDescKernelParam.__init__	   s'  € Ü—K‘K × 2Ñ 2¼%¿+¹+ÈeÔTˆŒ	Ü�4‹yœC 
›OÒ+Ð+Ð+Ø”C˜“IÔ" Ò"Ð"Ñ"Ð"Ð"Ø�y‰y×!Ñ!Ó# bÑ(¨AÒ-Ð-Ð-äˆt‹9˜Š>Ü�N‰N×!Ñ!×(Ñ(×.Ñ.×EÑEÀcÈ4ÐPQÉ7ÐT^Ð_`ÑTaÐcoØFJÇiÁi×FXÑFXÓFZõ\ô �N‰N×!Ñ!×(Ñ(×.Ñ.×EÑEÀcÈ4ÐPQÉ7ÐTXÐYZÑT[Ð]gÐhiÑ]jØFPÐQRÁmÐUaÐcg×clÑcl×cuÑcuÓcwõyó    c                 ó6   — | j                   j                  «       S ©N)r   r   )r   s    r    Útma_desc_cpu_ptrz#TmaDescKernelParam.tma_desc_cpu_ptr   s   € Ø�y‰y×!Ñ!Ó#Ð#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   € Ü˜c C 5¨9¨+°|ÓDÐDr"   c                 ó&   — t        | ||g||g|«      S r$   r+   )r   Údim1Údim0Ú
block_dim1Ú
block_dim0r   s         r    Úcreate_2d_tma_descriptorr4      s   € Ü˜c D¨$ <°*¸jÐ1IÈ<ÓXÐXr"   )r   r   r   r.   r4   r)   r"   r    ú<module>r5      s!   ðÛ ã ÷$ñ $ò*EóYr"   