
    ȅi                     `   S SK r S SKrS SKJr  S SKrSqS rS r\S 5       r " S S5      r	 " S	 S
\R                  5      r " S S5      rSS jr\SS j5       rS rS r " S S\5      r\" \ R$                  \   \5      \ R$                  \'   S SKJrJrJrJrJrJrJrJrJrJrJrJ r J!r!J"r"  g)    N)contextmanagerTc                      Sq g )NF"__allow_nonbracketed_mutation_flag     Q/home/james-whalen/.local/lib/python3.13/site-packages/torch/backends/__init__.pydisable_global_flagsr
      s    ).&r   c                      [         (       + $ Nr   r   r   r	   flags_frozenr      s    111r   c               #   8   #    [         n Sq  S v   U q g ! U q f = f7f)NTr   )olds    r	   __allow_nonbracketed_mutationr      s&      -C)-&1-0*S*s   	 c                   &    \ rS rSrS rS rS rSrg)ContextProp$   c                     Xl         X l        g r   gettersetter)selfr   r   s      r	   __init__ContextProp.__init__%   s    r   c                 "    U R                  5       $ r   )r   )r   objobjtypes      r	   __get__ContextProp.__get__)   s    {{}r   c                 v    [        5       (       d  U R                  U5        g [        SUR                   S35      e)Nznot allowed to set zM flags after disable_global_flags; please use flags() context manager instead)r   r   RuntimeError__name__)r   r   vals      r	   __set__ContextProp.__set__,   s9    ~~KK%cll^ 4Y Y r   r   N)r"   
__module____qualname____firstlineno__r   r   r$   __static_attributes__r   r   r	   r   r   $   s    r   r   c                   .   ^  \ rS rSrU 4S jrS rSrU =r$ )
PropModule6   c                 0   > [         TU ]  U5        Xl        g r   )superr   m)r   r/   name	__class__s      r	   r   PropModule.__init__7   s    r   c                 8    U R                   R                  U5      $ r   )r/   __getattribute__)r   attrs     r	   __getattr__PropModule.__getattr__;   s    vv&&t,,r   )r/   )r"   r&   r'   r(   r   r6   r)   __classcell__r1   s   @r	   r+   r+   6   s    - -r   r+   c                   4   ^  \ rS rSrS rU 4S jrS rSrU =r$ )_FP32Precision?   c                     Xl         X l        g r   backendop)r   r?   r@   s      r	   r   _FP32Precision.__init__@   s    r   c                    > US:X  a6  [         R                  R                  U R                  U R                  U5        g US;   a  [
        TU ]  X5        g [        SU-   5      e)Nfp32_precisionr>   Unknown attribute )torch_C_set_fp32_precision_setterr?   r@   r.   __setattr__AttributeError)r   r0   valuer1   s      r	   rH   _FP32Precision.__setattr__D   sQ    ##HH//dgguM&&G, !5!<==r   c                     US:X  a4  [         R                  R                  U R                  U R                  5      $ [        SU-   5      e)NrC   rD   )rE   rF   _get_fp32_precision_getterr?   r@   rI   )r   r0   s     r	   r6   _FP32Precision.__getattr__L   s;    ##8866t||TWWMM !5!<==r   r>   )	r"   r&   r'   r(   r   rH   r6   r)   r8   r9   s   @r	   r;   r;   ?   s    >> >r   r;   c                     [         R                  R                  SS5      4nU b!  [         R                  R                  SSU 5        U$ )Ngenericall)rE   rF   rM   rG   )_fp32_precision
orig_flagss     r	   	set_flagsrT   S   s=    ((55iGIJ"++IuoNr   c              #   .  #    [        5          [        U 5      nS S S 5         S v   [        5          [        W6   S S S 5        g ! , (       d  f       N/= f! , (       d  f       g = f! [        5          [        W6   S S S 5        f ! , (       d  f       f = f= f7fr   )r   rT   )rC   rS   s     r	   flagsrV   Z   sc     	&	(~.
 
)#*,z" -, 
)	(
 -,*,z" -,,s_   BABA$ 
B	A	B
AB
A!B$B/	B8	B
BBBc                    ^ ^ U U4S jnU$ )Nc                  D   > [         R                  R                  T T5      $ r   )rE   rF   rM   r>   s   r	   inner)_get_fp32_precision_getter.<locals>.innerf   s    xx227B??r   r   r?   r@   rY   s   `` r	   rM   rM   e   s    @ Lr   c                    ^ ^ U U4S jnU$ )Nc                 F   > [         R                  R                  TTU 5      $ r   )rE   rF   rG   )	precisionr?   r@   s    r	   rY   )_set_fp32_precision_setter.<locals>.innerm   s    xx227B	JJr   r   r[   s   `` r	   rG   rG   l   s    K Lr   c                   B    \ rS rSr\" \" SS5      \" SS5      5      rSrg)GenericModules   rP   rQ   r   N)	r"   r&   r'   r(   r   rM   rG   rC   r)   r   r   r	   ra   ra   s   s      "9e4"9e4Nr   ra   )cpucudacudnn
cusparseltkleidiaimhamiopenmklmkldnnmpsnnpackopenmp
opt_einsum	quantized)none)#systypes
contextlibr   rE   r   r
   r   r   r   
ModuleTyper+   r;   rT   rV   rM   rG   ra   modulesr"   torch.backendsrc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   r   r   r	   <module>rx      s    
  %  &* "/
2 1 1 $-!! -> >( # #J  &ckk(&;XFH     r   