U
    ªÊ¹j1  ã                   @   s6   d dl Z d dlZd dlZd dlmZ eddd„ƒZdS )é    N)Úcontextmanagerú	trace.fxtú/tmp/magic-tracec              	   c   sÞ   t  ¡ }t j |¡sFtd|› �ƒ t dd|ddg¡ t dd|g¡ |dd	t|ƒd
| g}tj|tj	dd�}|j
 ¡ }t|ƒ d|krlqŠqlz
d V  W 5 | tj¡ | ¡ }t|j
 ¡ ƒ |j
 ¡  |dkrØtd|› �ƒ‚X d S )NzDownloading magic_trace to: Zwgetz-Oz-qzNhttps://github.com/janestreet/magic-trace/releases/download/v1.0.2/magic-traceÚchmodz+xÚattachz-pidz-ozutf-8)ÚstderrÚencodingZAttachedr   zmagic_trace exited abnormally: )ÚosÚgetpidÚpathÚexistsÚprintÚ
subprocessÚrunÚstrÚPopenÚPIPEr   ÚreadlineÚsend_signalÚsignalÚSIGINTÚwaitÚreadÚcloseÚ
ValueError)ÚoutputZmagic_trace_cacheÚpidÚargsÚpÚxÚr© r!   úX/var/www/html/bwave/python/venv/lib/python3.8/site-packages/functorch/dim/magic_trace.pyÚmagic_trace   s4    ûÿ	


r#   )r   r   )r	   r   r   Ú
contextlibr   r#   r!   r!   r!   r"   Ú<module>   s
   