
    9j              	       \    d dl Z d dlZd dlZd dlmZ d dlmZ e	 d	dededed   fd       Zy)
    N)	Generator)contextmanageroutputmagic_trace_cachereturn)NNNc              #   &  K   t        j                         }t         j                  j                  |      s@t	        d|        t        j                  dd|ddg       t        j                  dd|g       |dd	t        |      d
| g}t        j                  |t
        j                  d      }|j                  t        d      	 |j                  j                         }t	        |       d|v rn+	 d  |j                  t        j                         |j!                         }|j                  =t	        |j                  j#                                |j                  j%                          |dk7  rt'        d|       y # |j                  t        j                         |j!                         }|j                  =t	        |j                  j#                                |j                  j%                          |dk7  rt'        d|       w xY ww)NzDownloading magic_trace to: wgetz-Oz-qzNhttps://github.com/janestreet/magic-trace/releases/download/v1.0.2/magic-tracechmodz+xattachz-pidz-ozutf-8)stderrencodingzExpected stderr to be non-NoneAttachedr   zmagic_trace exited abnormally: )osgetpidpathexistsprint
subprocessrunstrPopenPIPEr   AssertionErrorreadlinesend_signalsignalSIGINTwaitreadclose
ValueError)r   r   pidargspxrs          Y/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/functorch/dim/magic_trace.pymagic_tracer(      s     ))+C77>>+,,->,?@A!`	
 	'89:xS4HDjooHAxx=>>
HHa?	 
	D	fmm$FFH88!((--/"HHNN6>qcBCC  	
fmm$FFH88!((--/"HHNN6>qcBCC s    C.H1F 5BHBHH)z	trace.fxtz/tmp/magic-trace)	r   r   r   collections.abcr   
contextlibr   r   r(        r'   <module>r-      sP    
   % % 8J"D"D25"D "D "Dr,   