a
    h1                     @   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|v rlqqlzLd V  W |tj | }t|j
  |j
  |dkrtd| nF|tj | }t|j
  |j
  |dkrtd| 0 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!   U/var/www/html/assistant/venv/lib/python3.9/site-packages/functorch/dim/magic_trace.pymagic_trace   s@    	



r#   )r   r   )r	   r   r   
contextlibr   r#   r!   r!   r!   r"   <module>   s
   