Ë
    ùÿæi(  ã                   óŽ  — U d Z ddlZddlZddlmZ ddlZ G d„ d«      Z e«       Zdd„Zdd„Z	ej                  j                  «       Zi Zeee   ee   f   ed<   i Zeeee   f   ed	<   dd
„Zd„ Zd„ Zdd„Z ej.                  «       Zej.                  ed<    ej.                  «       Zej.                  ed<   d„ Zdd„Zd„ Zdd„Zy)zÄJIT-related state.

This module stores various pieces of Python-global state relating to the JIT.

This is not intended to be imported directly; please the exposed
functionalities in `torch.jit`.
é    N)ÚAnyc                   ó*   — e Zd ZdZdd„Zd„ Zdefd„Zy)ÚEnabledProxyzyStores whether the JIT is enabled or not.

    This is just a wrapper for a bool, so that we get reference semantics
    ÚreturnNc                 ó6   — | j                  dddd«      | _        y )NÚPYTORCH_JITTz> Using PyTorch JITz> PyTorch JIT DISABLED)Ú	parse_envÚenabled©Úselfs    úe/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/torch/jit/_state.pyÚ__init__zEnabledProxy.__init__   s   € Ø—~‘~Ø˜4Ð!6Ð8Pó
ˆ�ó    c                 óö   — t         j                  j                  |«      }|€|S |j                  «       dv ry|j                  «       dv ry|dk(  rt	        |«       y|dk(  rt	        |«       yt        d|› d�«      ‚)	N>   Ú1ÚyesÚtrueT>   Ú0ÚnoÚfalseFÚ1vÚ0vzUnknown setting of z. Try using 0 or 1.)ÚosÚenvironÚgetÚlowerÚprintÚ
ValueError)r   ÚnameÚdefaultÚtrue_messageÚfalse_messageÚvalues         r   r	   zEnabledProxy.parse_env   s|   € Ü—
‘
—‘˜tÓ$ˆØˆ=ØˆNØ�;‰;‹=Ð0Ñ0ØØ�[‰[‹]Ð2Ñ2ØØ�DŠ=Ü�,ÔØØ�dŠ]Ü�-Ô ØÜÐ.¨t¨fÐ4GÐHÓIÐIr   c                 ó   — | j                   S ©N)r
   r   s    r   Ú__bool__zEnabledProxy.__bool__,   s   € Ø�|‰|Ðr   ©r   N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r	   Úboolr&   © r   r   r   r      s   „ ñó

ò
Jð ˜$ô r   r   c                  ó   — dt         _        y )NF©Ú_enabledr
   r-   r   r   Údisabler1   3   s
   € Ø„HÕr   c                  ó   — dt         _        y )NTr/   r-   r   r   Úenabler3   7   s
   € Ø„HÕr   Ú_script_classesÚ_name_to_pyclassc                 óD   — |t         | <   | t        |j                  «       <   y r%   )r4   r5   Úqualified_name)Úpython_classÚscript_classs     r   Ú_add_script_classr:   F   s    € Ø$0„O�LÑ!Ø6BÔ�\×0Ñ0Ó2Ò3r   c                 ó`   — t        | dd «      }|�t        |«      } t        j                  | «      S )NÚ_jit_override_qualname)ÚgetattrÚ_get_python_classr4   r   )r8   Úoverrides     r   Ú_get_script_classr@   K   s2   € Ü�|Ð%=¸tÓD€HØÐÜ(¨Ó2ˆÜ×Ñ˜|Ó,Ð,r   c                 ó,   — t         j                  | «      S r%   )r5   r   )r7   s    r   r>   r>   R   s   € Ü×Ñ Ó/Ð/r   c                  óT   — t         j                  «        t        j                  «        y r%   )r4   Úclearr5   r-   r   r   Ú_clear_class_staterD   V   s   € Ü×ÑÔÜ×ÑÕr   Ú_jit_caching_layerÚ_jit_function_overload_cachingc                 ó„   — t         j                  | d «      }|r#|D �cg c]  }t        j                  |«      ‘Œ c}S y c c}w r%   )rF   r   Ú
_python_cuÚfind_function)ÚkeyÚ
qual_namesÚ	qual_names      r   Ú_try_get_jit_cached_overloadsrM   g   s@   € Ü/×3Ñ3°C¸Ó>€JÙÙEOÓPÁZ¸	”
×(Ñ(¨Õ3ÀZÑPÐPàùò Qs   �=c                 óP   — |D �cg c]  }|j                   ‘Œ c}t        | <   y c c}w r%   )r7   rF   )rJ   Úcompiled_fnsÚfns      r   Ú_set_jit_overload_cacherQ   o   s%   € ÙGSÓ*TÁ|À¨2×+<Ó+<À|Ñ*TÔ" 3Ò'ùÒ*Ts   …#c                 ó~   — t        | dd«      du ry t        j                  | d «      }|rt        j	                  |«      S y )NÚ __disable_jit_function_caching__FT)r=   rE   r   rH   rI   )rJ   rL   s     r   Ú_try_get_jit_cached_functionrT   s   s@   € ÜˆsÐ6¸Ó>À$ÑFØÜ"×&Ñ& s¨DÓ1€IÙÜ×'Ñ'¨	Ó2Ð2àr   c                 ó    — t        |t        j                  j                  «      st	        dt        |«      › �«      ‚|j                  t        | <   y )NzExpected ScriptFunction, got )Ú
isinstanceÚtorchÚjitÚScriptFunctionÚAssertionErrorÚtyper7   rE   )rJ   r#   s     r   Ú_set_jit_function_cacher\   }   s>   € ä�eœUŸY™Y×5Ñ5Ô6ÜÐ<¼TÀ%»[¸MÐJÓKÐKØ#×2Ñ2Ô�sÒr   r'   )r+   r   ÚweakrefÚtypingr   rW   r   r0   r1   r3   Ú_CÚCompilationUnitrH   r4   Údictr[   Ú__annotations__r5   Ústrr:   r@   r>   rD   ÚWeakKeyDictionaryrE   rF   rM   rQ   rT   r\   r-   r   r   Ú<module>re      sä   ðòó 
Û Ý ã ÷ñ ñ> ‹>€óóð �X‰X×%Ñ%Ó'€
ð /1€��d˜3‘i  c¡Ð*Ñ+Ó 0Ø)+Ð �$�s˜D ™I�~Ñ&Ó +óCò
-ò0óð 1J°×0IÑ0IÓ0KÐ �G×-Ñ-Ó KØ<U¸G×<UÑ<UÓ<WÐ  × 9Ñ 9Ó WòóUòô3r   