
    sgE                     R    d dl mZ d dlZd dlmZmZmZ g dZd Zd Z	edd       Z
y)	    )contextmanagerN)__allow_nonbracketed_mutationContextProp
PropModule)is_availableflags	set_flagsc                  *    t        j                         S )z4Return whether PyTorch is built with NNPACK support.)torch_nnpack_available     Q/var/www/html/venv/lib/python3.12/site-packages/torch/backends/nnpack/__init__.pyr   r      s    ""$$r   c                     t         j                  j                         f}t         j                  j                  |        |S )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              #   (  K   t               5  t        |       }ddd       	 d t               5  t        d          ddd       y# 1 sw Y   0xY w# 1 sw Y   yxY w# t               5  t        d          ddd       w # 1 sw Y   w xY wxY ww)z9Context manager for setting if nnpack is enabled globallyNr   )r   r	   )enabledr   s     r   r   r      s      
'	( (w'
(%*, 	%jm$	% 	%( (
	% 	%*, 	%jm$	% 	% 	%s_   BABA  
BA	BABAB B+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   