Ë
    èÿæiÒ  ã                   óÖ   — d dl mZ d dlmZmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZmZ  G d„ d«      Zd	efd
„Ze
d„ «       Z eed¬«      d„ «       Ze
d„ «       Z eedd¬«      d„ «       Zy)é    )Útypes)ÚoverloadÚoverload_method)Ú	signature)Ú	nvvmutils)Ú	intrinsic)Ú
grid_groupÚ	GridGroupc                   ó   — e Zd ZdZdd„Zy)r
   z0A cooperative group representing the entire gridNc                   ó   — y)zSynchronize this grid groupN© r   ó    úb/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/cuda/cg.pyÚsynczGridGroup.sync   s   � r   )ÚreturnN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r
   r
   	   s
   „ Ù:ô*r   r
   r   c                  ó   — t        «       S )zGet the current grid group.)r
   r   r   r   Ú	this_gridr      s
   € ä‹;Ðr   c                 ó.   — t        t        «      }d„ }||fS )Nc                 ó¦   — | j                  t        j                  d«      }|j                  }|j	                  t        j                  |«      |f«      S )Né   )Úget_constantr   Úint32ÚmoduleÚcallr   Ú declare_cudaCGGetIntrinsicHandle)ÚcontextÚbuilderÚsigÚargsÚoneÚmods         r   Úcodegenz_this_grid.<locals>.codegen   sE   € Ø×"Ñ"¤5§;¡;°Ó2ˆØ�n‰nˆØ�|‰|Ü×6Ñ6°sÓ;ØˆFóð 	r   )r   r	   )Ú	typingctxr"   r&   s      r   Ú
_this_gridr(      s   € ä
”JÓ
€Còð �ˆ<Ðr   Úcuda)Útargetc                  ó   — d„ } | S )Nc                  ó   — t        «       S ©N)r(   r   r   r   Úimplz_ol_this_grid.<locals>.impl%   s
   € Ü‹|Ðr   r   )r.   s    r   Ú_ol_this_gridr/   #   s   € òð €Kr   c                 óD   — t        t        j                  |«      }d„ }||fS )Nc                 ó®   — | j                  t        j                  d«      }|j                  }|j	                  t        j                  |«      g |¢|‘­«      S )Nr   )r   r   r   r   r   r   Údeclare_cudaCGSynchronize)r    r!   r"   r#   Úflagsr%   s         r   r&   z!_grid_group_sync.<locals>.codegen/   sK   € Ø×$Ñ$¤U§[¡[°!Ó4ˆØ�n‰nˆØ�|‰|Ü×/Ñ/°Ó4ØˆdˆN�E‰Nóð 	r   )r   r   r   )r'   Úgroupr"   r&   s       r   Ú_grid_group_syncr5   +   s#   € ä
”E—K‘K Ó
'€Còð �ˆ<Ðr   r   c                 ó   — d„ }|S )Nc                 ó   — t        | «      S r-   )r5   )r4   s    r   r.   z!_ol_grid_group_sync.<locals>.impl;   s   € Ü Ó&Ð&r   r   )r4   r.   s     r   Ú_ol_grid_group_syncr8   9   s   € ò'ð €Kr   N)Ú
numba.corer   Únumba.core.extendingr   r   Únumba.core.typingr   Ú
numba.cudar   Únumba.cuda.extendingr   Únumba.cuda.typesr	   r
   ÚGridGroupClassr   r(   r/   r5   r8   r   r   r   Ú<module>r@      s�   ðÝ ß :Ý 'Ý  Ý *ß D÷*ñ *ð�9ó ð
 ñ
ó ð
ñ 
ˆ)˜FÔ#ñó $ðð ñ
ó ð
ñ � °Ô7ñó 8ñr   