
    shE                     T    S SK Jr  S SKrS SKJrJrJr  / SQrS rS r	\SS j5       r
g)	    )contextmanagerN)__allow_nonbracketed_mutationContextProp
PropModule)is_availableflags	set_flagsc                  ,    [         R                  " 5       $ )z4Return whether PyTorch is built with NNPACK support.)torch_nnpack_available     r/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/backends/nnpack/__init__.pyr   r      s    ""$$r   c                     [         R                  R                  5       4n[         R                  R                  U 5        U$ )z!Set if nnpack is enabled globally)r   _C_get_nnpack_enabled_set_nnpack_enabled)_enabled
orig_flagss     r   r	   r	      s/    ((..02J	HH  *r   c              #   F  #    [        5          [        U 5      nSSS5         Sv   [        5          [        WS   5        SSS5        g! , (       d  f       N5= f! , (       d  f       g= f! [        5          [        WS   5        SSS5        f ! , (       d  f       f = f= f7f)z9Context manager for setting if nnpack is enabled globallyNr   )r   r	   )enabledr   s     r   r   r      sm      
'	(w'
 
)%*,jm$ -, 
)	(
 -,*,jm$ -,,s_   B!AB!A* 
B!A	B!
AB!
A'#B!*B5B	B
BBB!)F)
contextlibr   r   torch.backendsr   r   r   __all__r   r	   r   r   r   r   <module>r      s6    %  Q Q 1%
 % %r   