
    pi                         U d dl Z d dlmZ d dlZd dlmZmZmZ 	 d	dZe	 d
d       Z	 G d de      Z
 e
e j                  e   e      e j                  e<   eed<   y)    N)contextmanager)__allow_nonbracketed_mutationContextProp
PropModulec                     t         j                  j                         f}| t         j                  j                  |        |S N)torch_C_get_miopen_immediate_set_miopen_immediate)
_immediate
orig_flagss     `/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/torch/backends/miopen/__init__.py	set_flagsr   	   s7     ((0024J&&z2    c              #     K   t               5  t        |       }d d d        	 d  t               5  t          d d d        y # 1 sw Y   *xY w# 1 sw Y   y xY w# t               5  t          d d d        w # 1 sw Y   w xY wxY wwr   )r   r   )	immediater   s     r   flagsr      s      
'	( 



# +, 	#z"	# 	#
 
	# 	#*, 	#z"	# 	# 	#s_   BABA 
B	A	BABABB%	A7.	B7B <BBc                        e Zd Z fdZ eej                  j                  ej                  j                        Z	 xZ
S )MiopenModulec                 &    t         |   ||       y r   )super__init__)selfmname	__class__s      r   r   zMiopenModule.__init__(   s    D!r   )__name__
__module____qualname__r   r   r	   r
   r   r   r   __classcell__)r   s   @r   r   r   '   s-    " &&(F(FIr   r   r   r   )F)sys
contextlibr   r	   torch.backendsr   r   r   r   r   r   modulesr   bool__annotations__ r   r   <module>r)      sh    
 %  Q Q  # #(:  %S[[%:HEH  r   