Ë
    ùÿæi—  ã                  ó"  — d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
mZmZmZ d dlmZmZmZmZmZmZ d dlmZ erd d	lmZ dd
„Zdddœ	 	 	 	 	 	 	 	 	 	 	 dd„Zdddœ	 	 	 	 	 	 	 	 	 dd„Zddœdd„Zdd„Zdddœ	 	 	 	 	 	 	 dd„Zddœdd„Zy)é    )ÚannotationsN)ÚTYPE_CHECKING)Úassert_never)Úcpp)ÚArgNameÚBindingÚCTypeÚ
NamedCType)ÚArgumentÚFunctionSchemaÚReturnÚSelfArgumentÚTensorOptionsArgumentsÚType)Ú	concatMap)ÚSequencec                ó,   — t        j                  | «      S ©N)r   Úname)Úfuncs    úl/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torchgen/api/dispatcher.pyr   r   (   s   € Ü�8‰8�D‹>Ðó    FT)Úremove_non_owning_ref_typesÚsymintc               ó6   — t        j                  | ||||¬«      S )N)ÚmutableÚbindsr   r   )r   Úargumenttype_type)Útr   r   r   r   s        r   r   r   ,   s&   € ô × Ñ Ø	ØØØØ$?ôð r   c               óJ   — t        | j                  | j                  |||¬«      S )N)r   r   r   r   )r   ÚtypeÚis_write)Úar   r   r   s       r   Úargument_typer$   A   s(   € ô Ø	�‰Ø—
‘
ØØ$?Øôð r   ©r   c               ó0   — t        j                  | |¬«      S ©Nr%   )r   Úreturns_type)Úrsr   s     r   r(   r(   Q   s   € ä×Ñ˜B vÔ.Ð.r   c           
     óÞ   — 	 	 	 	 dd„}t        t        |t        j                  | j                  j
                  | j                  j                  | j                  j                  «      «      «      S )Nc                óö   — t        | t        «      r| gS t        | t        «      r| j                  gS t        | t        «      r.| j
                  | j                  | j                  | j                  gS t        | «       y r   )
Ú
isinstancer   r   Úargumentr   ÚdtypeÚlayoutÚdeviceÚ
pin_memoryr   )r#   s    r   Úto_argumentz"jit_arguments.<locals>.to_argumentW   s\   € ô �aœÔ"Ø�3ˆJÜ˜œ<Ô(Ø—J‘J�<ÐÜ˜Ô1Ô2Ø—G‘G˜QŸX™X q§x¡x°·±Ð>Ð>ä˜�Or   )r#   z0Argument | TensorOptionsArguments | SelfArgumentÚreturnúlist[Argument])Úlistr   Ú	itertoolsÚchainÚ	argumentsÚ
positionalÚ
kwarg_onlyÚout)r   r2   s     r   Újit_argumentsr<   V   sb   € ð
Ø;ð
à	ó
ô ÜØÜ�O‰OØ—‘×)Ñ)¨4¯>©>×+DÑ+DÀdÇnÁn×FXÑFXóó	
óð r   c               ó`   — t        t        | | j                  ||¬«      | j                  | ¬«      S )N)r   r   r   )Únctyper   r-   )r   r$   r   )r#   r   r   s      r   r-   r-   m   s5   € ô ÜØØ—&‘&Ø(CØô	
ð �V‰VØô	ð 	r   c               óT   — t        | «      D �cg c]  }t        ||¬«      ‘Œ c}S c c}w r'   )r<   r-   )r   r   r#   s      r   r8   r8   |   s(   € Ü0=¸dÔ0CÓDÑ0C¨1ŒH�Q˜vÖ&Ð0CÑDÐDùÒDs   Ž%)r   r   r3   Ústr)r   r   r   Úboolr   r   r   rA   r   rA   r3   r
   )
r#   r   r   r   r   rA   r   rA   r3   r
   )r)   zSequence[Return]r   rA   r3   r	   )r   r   r3   r4   )r#   r   r   rA   r   rA   r3   r   )r   r   r   rA   r3   zlist[Binding]) Ú
__future__r   r6   Útypingr   Útyping_extensionsr   Útorchgen.apir   Útorchgen.api.typesr   r   r	   r
   Útorchgen.modelr   r   r   r   r   r   Útorchgen.utilsr   Úcollections.abcr   r   r   r$   r(   r<   r-   r8   © r   r   Ú<module>rK      s   ðÝ "ã Ý  Ý *å ß BÓ B÷÷ õ %ñ Ý(ó&ð ).ØñØðð ðð ð	ð
 "&ðð ðð óð2 ).ØñØðð ðð "&ð	ð
 ðð óð  :>õ /ó
ð0 9>ÈdñØðØ15ðØGKðàóð 7;ö Er   