
    N j              	       `    S SK r S SKrS SKrS SKJr  S SKJr  \ S	S\S\S\S   4S jj5       rg)
    N)	Generator)contextmanageroutputmagic_trace_cachereturn)NNNc              #   B  #    [         R                  " 5       n[         R                  R                  U5      (       dB  [	        SU 35        [
        R                  " SSUSS/5        [
        R                  " SSU/5        USS	[        U5      S
U /n[
        R                  " U[
        R                  SS9nUR                  c  [        S5      e UR                  R                  5       n[	        U5        SU;   a  OM.   S v   UR                  [        R                  5        UR!                  5       nUR                  b=  [	        UR                  R#                  5       5        UR                  R%                  5         US:w  a  ['        SU 35      eg ! UR                  [        R                  5        UR!                  5       nUR                  b=  [	        UR                  R#                  5       5        UR                  R%                  5         US:w  a  ['        SU 35      ef = f7f)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          j/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/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    C8H;F ?B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,   