
    Wj              	       j    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dS )    N)	Generator)contextmanager	trace.fxt/tmp/magic-traceoutputmagic_trace_cachereturn)NNNc              #   @  K   t          j                    }t           j                            |          sBt	          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
                                        }t	          |           d|v rn.	 d V  |                    t          j                   |                                }|j
        ?t	          |j
                                                   |j
                                         |dk    rt'          d|           d S # |                    t          j                   |                                }|j
        ?t	          |j
                                                   |j
                                         |dk    rt'          d|           w xY 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-NoneT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          ^/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/functorch/dim/magic_trace.pymagic_tracer*      s      )++C7>>+,, ;@->@@AAA!`	
 	
 	
 	'89:::xS4HDjoHHHAx=>>>Ha??	
	D	fm$$$FFHH8!(--//"""HNN66BqBBCCC 6 	
fm$$$FFHH8!(--//"""HNN66BqBBCCCCCCCs   3F
 
BH)r   r   )	r   r   r   collections.abcr   
contextlibr   r   r*        r)   <module>r/      s    
			      % % % % % % % % % % % % 8J"D "D"D25"D "D "D "D "D "D "Dr.   