
    [ThQ                        S SK r S SKrS SKrS SKrS SKJr  S SKJrJrJ	r	J
r
JrJr  S SKJrJr  S SKrS SKJs  Jr  S SKJr  S SKJrJrJr  S SKJrJrJr  S SKJ r   S S	K!J"r"  S S
K#J$r$  \" S\RJ                  \RL                  5      r' " S S\RP                  5      r) " S S5      r*\	" S\+5      r, " S S\RZ                  5      r. " S S5      r/ " S S5      r0 " S S5      r1 " S S5      r2 " S S5      r3 " S S5      r4 " S S 5      r5 " S! S"\55      r6 " S# S$\55      r7 " S% S&\55      r8 " S' S(\55      r9 " S) S*\55      r: " S+ S,5      r; " S- S.5      r<g)/    N)abstractmethod)AnyCallableNewTypeOptionalTypeVarUnion)overrideSelf)TracingContext)
FakeTensorFakeTensorModeTensor)MetaConverterMetaTensorDescMetaTensorDescriberSymNode)ShapeEnv)no_dispatch	_SymNodeTc            	          ^  \ rS rSrSrS\R                  SS4U 4S jjr\S\	S\
\S\4   \
\S4   4   4S	 j5       r\S\	S\\   4S
 j5       r\S\	S\4S j5       r\S\S\S\	4S j5       rSrU =r$ )GraphPickler   zV
GraphPickler is a Pickler which helps pickling fx graph - in particular
GraphModule.
filereturnNc                 p   > [         TU ]  U5        [        [        5       5      U l        [        SS9U l        g )NF)	copy_data)super__init___UnpickleStateTokenobject_unpickle_stater   _meta_tensor_describer)selfr   	__class__s     O/var/www/auris/envauris/lib/python3.13/site-packages/torch/fx/_graph_pickler.pyr    GraphPickler.__init__    s0      368< ':E&J#    obj.c                 X   [        U[        5      (       a  [        R                  X5      $ [        U[        R
                  R                  5      (       a  [        R                  X5      $ [        U[        R                  R                  [        R                  R                  45      (       a  [        R                  X5      $ [        U[        5      (       a  [        R                  X5      $ [        U[        R                  5      (       a  [        R                  X5      $ [        U[        R                   R"                  5      (       a  [$        R                  X5      $ [        U[        R
                  R&                  5      (       a   e[(        R                  X5      =n(       a  U$ [*        $ N)
isinstancer   _TensorPickleDatareduce_helpertorchfxGraphModule_GraphModulePickleData_opsOperatorBaseOpOverloadPacket_OpPickleDatar   _ShapeEnvPickleDataSymInt_SymNodePickleData_guardsr   _TracingContextPickleDataNode_TorchNumpyPickleDataNotImplemented)r%   r*   reduces      r'   reducer_overrideGraphPickler.reducer_override-   s)   * c:&&$224==UXX1122)77BBejj55uzz7R7RSTT ..t99X&&&44T??U\\**%33D>>U]]99::,::4EE "#uxx}}5555.<<TGGvG "!r)   c                 "    XR                   L a  gg )Nunpickle_stater#   )r%   r*   s     r'   persistent_idGraphPickler.persistent_idX   s    &&&#r)   c                     [         R                  " 5        nU " U5      nUR                  U5        UR                  5       sSSS5        $ ! , (       d  f       g= f)z
Pickle an object.
N)ioBytesIOdumpgetvalue)clsr*   streampicklers       r'   dumpsGraphPickler.dumps_   s8    
 ZZ\V&kGLL??$ \\s   )A		
Adata	fake_modec                     [        U5      n[        R                  " U 5       n[        X25      nUR	                  5       sSSS5        $ ! , (       d  f       g= f)z
Unpickle an object.
N)_UnpickleStaterI   rJ   _GraphUnpicklerload)rR   rS   staterN   	unpicklers        r'   loadsGraphPickler.loadsi   s<    
 y)ZZ'6I>># s   A
A)r$   r#   )__name__
__module____qualname____firstlineno____doc__rI   rJ   r    r
   r"   tupler   r   rA   r   strrF   classmethodbytesrP   staticmethodr   rZ   __static_attributes____classcell__r&   s   @r'   r   r      s    
KRZZ KD K ("("	xS!5c?2	3(" ("T  HSM   % %5 % % $E $n $ $ $r)   r   c                   &    \ rS rSrS\SS4S jrSrg)rU   t   rS   r   Nc                 .    Xl         [        5       U l        g r,   )rS   r   meta_converter)r%   rS   s     r'   r    _UnpickleState.__init__u   s    "9Fr)   )rS   rl   )r\   r]   r^   r_   r   r    rf    r)   r'   rU   rU   t   s    I. IT Ir)   rU   r!   c                   h   ^  \ rS rSrS\R
                  S\SS4U 4S jjr\S\	S\	4S j5       r
S	rU =r$ )
rV      rN   rD   r   Nc                 0   > [         TU ]  U5        X l        g r,   )r   r    r#   )r%   rN   rD   r&   s      r'   r    _GraphUnpickler.__init__   s     -r)   pidc                 R    US:X  a  U R                   $ [        R                  " S5      e)NrD   zInvalid persistent ID)r#   pickleUnpicklingError)r%   rs   s     r'   persistent_load_GraphUnpickler.persistent_load   s)    ""'''(()@AAr)   rE   )r\   r]   r^   r_   rI   rJ   rU   r    r
   r"   rw   rf   rg   rh   s   @r'   rV   rV      sG    .rzz .> .d . B6 Bf B Br)   rV   c                       \ rS rSr% \\\4   \S'   \S\	S\
S\\\\/\
4   \\\4   4   4S j5       rS\
SS4S	 jrS
\S\
4S jrSrg)r8      rR   rO   r*   r   c                 B    U R                   U " U5      UR                  44$ r,   unpickler#   rM   rO   r*   s      r'   r/   !_ShapeEnvPickleData.reduce_helper   s"     ||c#h(?(?@@@r)   envNc                     UR                   (       a   eUR                  R                  5       U l        U R                  S	 U R                  S	 g )Ntracked_fakesfake_tensor_cache)_translation_validation_enabled__dict__copyrR   )r%   r   s     r'   r    _ShapeEnvPickleData.__init__   s@     6666LL%%'	IIo&II)*r)   rD   c                    UR                   (       d   eUR                   R                  (       d   eU R                  R                  5        H%  u  p#[	        UR                   R                  X#5        M'     UR                   R                  $ r,   )rS   	shape_envrR   itemssetattr)r%   rD   kvs       r'   r}   _ShapeEnvPickleData.unpickle   sg    ''''''1111IIOO%DAN,,66= & ''111r)   )rR   )r\   r]   r^   r_   dictrb   r"   __annotations__rc   r   r   ra   r   r   rU   r!   r/   r    r}   rf   rn   r)   r'   r8   r8      s    
sF{
A"A)1A	$'12E$@S:S4TT
A A+H + +2~ 2( 2r)   r8   c                       \ rS rSr\S\S\S\\\	\
/\4   \\	\4   4   4S j5       rS\SS4S jrS\4S	 jrS
\
S\R"                  4S jrSrg)r:      rO   r*   r   c                     U " UR                   5      UR                  4n[        U[        R                  5      (       a  [
        R                  U4$ [        S[        U5       35      e)NzUnhandled SymNode type )	noder#   r-   r0   r9   r:   unpickle_sym_intNotImplementedErrortype)rM   rO   r*   argss       r'   r/    _SymNodePickleData.reduce_helper   sU     CHHw667c5<<((%66<<%(?S	{&KLLr)   r   Nc                     UR                   U l        UR                  U l        UR                  U l        UR                  U l        g r,   )_exprexprr   pytype_hinthint)r%   r   s     r'   r    _SymNodePickleData.__init__   s.    JJ	kkJJ	r)   c                     SSK Jn  U R                  c   eU" U R                  U R                  U R                  U R
                  5      $ )Nr   r   )torch.fx.experimental.sym_noder   r   r   r   r   )r%   r   s     r'   _to_sym_node_SymNodePickleData._to_sym_node   s6    :~~)))tyy$..$++tyyIIr)   rD   c                 J    [         R                  " U R                  5       5      $ r,   )r0   r9   r   r%   rD   s     r'   r   #_SymNodePickleData.unpickle_sym_int   s    ||D--/00r)   )r   r   r   r   )r\   r]   r^   r_   rc   r   r   ra   r   r   rU   r!   r/   r   r    r   r0   r9   r   rf   rn   r)   r'   r:   r:      s    MM M 
$'23U4AT;T5UU
	M MW  Jg J1~ 1%,, 1r)   r:   c                       \ rS rSr% \\   \S'   \S\S\S\	\
\\/\4   \	\\4   4   4S j5       rS\S\SS	4S
 jrS\S\4S jrSrg	)r.      metadatarO   r*   r   c                 X    U R                   U " UR                  U5      UR                  44$ r,   )r}   r$   r#   r~   s      r'   r/   _TensorPickleData.reduce_helper   s3     ||..4##
 
 	
r)   	describertNc           	         UR                  U5      nUR                  (       a?  [        UR                  [        R                  R
                  R                  5      (       d   e[        R                  " US S9U l	        [        R                   HC  nUS;   a  M  [        U R                  U5      c  M$   SU S[        U R                  U5       35       e   g )NrS   )rS   	view_funcz
not None: z: )describe_tensorr   r-   r0   _subclasses
meta_utils_FakeTensorViewFuncdataclassesreplacer   r   _UNSERIALIZABLEgetattr)r%   r   r   r   r   s        r'   r    _TensorPickleData.__init__   s    
 ,,Q/ %% 1 1 < < P P*
 *
 	
 
 $++HE  //A..q)1=A3b!: ;<=1	 0r)   rD   c                 D  ^ [         R                  " U R                  TR                  S9nS[        / [
        R                  4   S[        [
        R                  [        4   S[        4U4S jjnTR                  R                  UTR                  R                  US S 5      $ )Nr   make_meta_tdevicer   c                    > [        5          [        TR                  U " 5       U5      sS S S 5        $ ! , (       d  f       g = fr,   )r   r   rS   )r   r   rD   s     r'   	with_fake-_TensorPickleData.unpickle.<locals>.with_fake   s-     !",,M s	   2
A )r   r   r   rS   r   r0   r   r	   r   rb   r   rl   meta_tensorr   )r%   rD   r   r   s    `  r'   r}   _TensorPickleData.unpickle   s    &&MM$..

	!"ell"23	=B5<<QTCT=U		 ,,88$$..
 	
r)   )r   )r\   r]   r^   r_   r   r   r   rc   r   ra   r   r   rU   r!   r/   r   r   r    r}   rf   rn   r)   r'   r.   r.      s    Z((
"
)3
	$'34eDBU<U6VV

 
="5 =& =T =,
~ 
* 
r)   r.   c                       \ rS rSr\S\S\S\\\	\
\/\4   \\
\4   4      4S j5       rS\S\SS4S	 jrS
\S\	S\4   4S jr\S\S\\
   4S j5       rSrg)r>   i  rO   r*   r   c                 f    U R                  U5      =n(       a  U R                  X1R                  44$ g r,   )from_objectr}   r#   )rM   rO   r*   rR   s       r'   r/   #_TorchNumpyPickleData.reduce_helper	  s3     ??3''4'LL4)@)@"ABBr)   modnameNc                     Xl         X l        g r,   r   r   )r%   r   r   s      r'   r    _TorchNumpyPickleData.__init__  s    	r)   rD   .c                     [        [        R                  " U R                  5      U R                  5      n[
        R                  R                  R                  R                  5       U   $ r,   )
r   	importlibimport_moduler   r   r0   _dynamo	variablesmiscget_np_to_tnp_map)r%   rD   nps      r'   r}   _TorchNumpyPickleData.unpickle  sE    Y,,TXX6		B}}&&++==?CCr)   tnpc                    [        U5      (       d  g [        R                  R                  R                  R                  5       n UR                  U5      =n(       d  g  [        USS 5      =n(       d  Sn[        USS 5      =n(       d  g U[        [        R                  " U5      U5      :X  d   eU " XE5      $ ! [         a     g f = f)Nr]   numpyr\   )callabler0   r   r   r   get_tnp_to_np_mapget	TypeErrorr   r   r   )rM   r   	tnp_to_npr   r   r   s         r'   r   !_TorchNumpyPickleData.from_object  s    }}MM++00BBD		#--,,B, -
 r<666CJ555WY44S94@@@@3~  		s   B8 8
CCr   )r\   r]   r^   r_   rc   r   r"   r   ra   r   r   rU   r!   r/   rb   r    r}   r   rf   rn   r)   r'   r>   r>     s    
"
)/
	dN+V34eDBU<U6VV	


 
C s t D~ D(3;:O D f $  r)   r>   c                      \ rS rSr\S\S\R                  R                  S\	\
\\/\R                  R                  4   \	\\4   4   4S j5       rS\R                  R                  SS4S jrS	\S\R                  R                  4S
 jrSrg)r3   i4  rO   r*   r   c                 B    U R                   U " U5      UR                  44$ r,   r|   r~   s      r'   r/   $_GraphModulePickleData.reduce_helper5  s*     ||H##
 
 	
r)   gmNc                 8   [        U[        R                  R                  R                  5      (       a  UR                  5       nOUR                  5       nUR                  R                  5       U l	        U R                  S	 [        UR                  5      U l        g )N_graph)r-   r0   r1   _lazy_graph_module_LazyGraphModule_real_recompile	recompiler   r   gm_dict_GraphPickleDatar   graph)r%   r   _python_codes      r'   r    _GraphModulePickleData.__init__A  sh    b%((55FFGG--/L<<>L{{'')LL"%bii0
r)   rD   c                     [         R                  R                  R                  [         R                  R                  5      nU R                  Ul        U R                  R                  X!5      Ul        U$ r,   )	r0   r1   r2   __new__r   r   r   r}   r   )r%   rD   r   s      r'   r}   _GraphModulePickleData.unpickleK  sL    XX!!))%((*>*>?llJJ'';		r)   )r   r   )r\   r]   r^   r_   rc   r   r0   r1   r2   ra   r   r   rU   r!   r/   r    r}   rf   rn   r)   r'   r3   r3   4  s    	
"	
).)=)=	
	$')=)==>d''(	*
	
 	
1588// 1D 1~ %((:N:N r)   r3   c                       \ rS rSrS\R
                  R                  S\\R
                  R                  S 4   SS4S jrS\R
                  R                  S\S \R
                  R                  4   S\
S\R
                  R                  4S	 jrS
rg)_NodePickleDataiR  r   mappingr   Nc                   ^ [         R                  " [        R                  R                  U4S jUR
                  5      U l        [         R                  " [        R                  R                  U4S jUR                  5      U l        UR                  U l        UR                  U l        [        R                  UR                  5      U l        UR                  U l        UR                  U l        g )Nc                    > TU    $ r,   rn   nr   s    r'   <lambda>*_NodePickleData.__init__.<locals>.<lambda>V  s	    '!*r)   c                    > TU    $ r,   rn   r   s    r'   r   r   X  s	    WQZr)   )pytreetree_map_onlyr0   r1   r=   r   kwargsr   opr7   ru   targetr   meta)r%   r   r   s     `r'   r    _NodePickleData.__init__S  s     ((8LdiiX	**HHMM/
 II	''#**4;;7 II	 II	r)   r   rD   c                   ^ [         R                  " [        U4S jU R                  5      n[         R                  " [        U4S jU R                  5      nU R
                  R                  U5      n[        U5      (       d  [        U[        5      (       d   eUR                  U R                  XdXPR                  U R                  5      nU R                  Ul        U$ )Nc                    > TU    $ r,   rn   r   s    r'   r   *_NodePickleData.unpickle.<locals>.<lambda>l  s	    wqzr)   c                    > TU    $ r,   rn   r   s    r'   r   r  n  s	    wqzr)   )r   r   r   r   r   r   r}   r   r-   rb   create_noder   r   r   r   )r%   r   r   rD   r   r   r   r   s     `     r'   r}   _NodePickleData.unpicklef  s     ##O5I499U%%14;;
 %%n5:fc#:#:::  &		499UII	r)   )r   r   r   r   r   r   r   )r\   r]   r^   r_   r0   r1   r=   r   r    GraphrU   r}   rf   rn   r)   r'   r   r   R  s    HHMM,0@Q1Q,R	&xx~~ '67 '	
 
r)   r   c            
           \ rS rSr\S\S\S\\\	/\4   \\
   4   4S j5       r\S\SS 4S j5       r\S\S\\S	   \S
   4   SS 4S j5       r\S\	S\4S j5       r\S\S\4S j5       r\S\S\S\4S j5       rSrg)r7   iw  rO   r   r   c                 V    U R                  U5      nUR                  UR                  44$ r,   )ru   r}   r#   )rM   rO   r   results       r'   r/   _OpPickleData.reduce_helperx  s)     B'"9"9!;<<r)   c                    [        U[        5      (       a  [        U5      $ [        R                  R
                  R                  U5      n[        U[        R                  R                  5      (       a  U R                  U[        5      $ [        U[        R                  R                  5      (       a  U R                  U[        5      $ UR                  S5      (       a  UR                  SS5      u  p4[        X45      $ UR                  S5      (       a  UR                  SS5      u  pT[!        U5      $ [#        S[%        U5       SU SU 35      e)N)z	builtins.zmath.ztorch..   z	operator.zTARGET:  )r-   rb   _OpStrPickleDatar0   r1   r=   _pretty_print_targetr4   
OpOverload
_pickle_op_OpOverloadPickleDatar6   _OpOverloadPacketPickleData
startswithsplit_OpBuiltinPickleData_OpOperatorPickleDatar   r   )rM   r   r   rootdetail_s         r'   ru   _OpPickleData.pickle  s   b##B''xx}}11"5b%**//00>>$(=>>EJJ7788>>$(CDD__=>>::c1-LD'55__[))

3*IA(00 &b
!B4q&GHHr)   r   dataclsr  r  c                 `    U R                  S5      (       d  SSKJn  U" SU  35      eU" U 5      $ )Nztorch.ops.atenr   )BypassFxGraphCachez"Unable to pickle non-standard op: )r  torch._inductor.codecacher   )r   r  r   s      r'   r  _OpPickleData._pickle_op  s4     /00D$'I$%PQQt}r)   rD   c                     g r,   rn   r   s     r'   r}   _OpPickleData.unpickle  s    r)   c                     SU;   a2  UR                  SS5      u  p#[        5       U   nU R                  XC5      $ [        5       U   $ )z3
Like `globals()[name]` but supports dotted names.
r  r  )r  globals_getattr_by_name)rM   r   r   restr  s        r'   _lookup_global_by_name$_OpPickleData._lookup_global_by_name  sE    
 $;

3*IC9S>D''339T?"r)   r  c                 r    SU;   a'  UR                  SS5      u  p![        X5      n SU;   a  M'  [        X5      $ )z7
Like `getattr(root, name)` but supports dotted names.
r  r  )r  r   )r  r   r   s      r'   r'  _OpPickleData._getattr_by_name  s=    
 Tk

3*IC4%D Tk t""r)   rn   N)r\   r]   r^   r_   rc   r   r"   ra   r   rU   r!   r/   ru   re   rb   r	   r   r  r   r}   r)  r'  rf   rn   r)   r'   r7   r7   w  s#   ="=(.=	x(&0159L3MM	N= = I I? I I& 

()40M+NN


 

 
 ~ &   	## 	#& 	# 	# #v #S #V # #r)   r7   c                   8    \ rS rSrS\SS4S jrS\S\4S jrSrg)	r  i  r   r   Nc                     Xl         g r,   r   r%   r   s     r'   r    _OpStrPickleData.__init__      	r)   rD   c                     U R                   $ r,   r/  r   s     r'   r}   _OpStrPickleData.unpickle  s    yyr)   r/  )	r\   r]   r^   r_   rb   r    rU   r}   rf   rn   r)   r'   r  r    s(    S T ~ # r)   r  c                   `    \ rS rSrS\SS4S jrS\S\R                  R                  4S jr
Srg)	r  i  r   r   Nc                     Xl         g r,   r/  r0  s     r'   r    _OpOverloadPickleData.__init__  r2  r)   rD   c                     U R                  U R                  5      n[        U[        R                  R
                  5      (       d   eU$ r,   )r)  r   r-   r0   r4   r  r%   rD   r*   s      r'   r}   _OpOverloadPickleData.unpickle  s8    ))$))4#uzz445555
r)   r/  )r\   r]   r^   r_   rb   r    rU   r0   r4   r  r}   rf   rn   r)   r'   r  r    s2    S T ~ %**:O:O r)   r  c                   `    \ rS rSrS\SS4S jrS\S\R                  R                  4S jr
Srg)	r  i  r   r   Nc                     Xl         g r,   r/  r0  s     r'   r    $_OpOverloadPacketPickleData.__init__  r2  r)   rD   c                     U R                  U R                  5      n[        U[        R                  R
                  5      (       d   eU$ r,   )r)  r   r-   r0   r4   r6   r9  s      r'   r}   $_OpOverloadPacketPickleData.unpickle  s8    ))$))4#uzz::;;;;
r)   r/  )r\   r]   r^   r_   rb   r    rU   r0   r4   r6   r}   rf   rn   r)   r'   r  r    s2    S T ~ %**:U:U r)   r  c                   <    \ rS rSrS\S\SS4S jrS\S\4S jrS	r	g)
r  i  r  r   r   Nc                     Xl         X l        g r,   )r  r   )r%   r  r   s      r'   r    _OpBuiltinPickleData.__init__  s    		r)   rD   c                 *   U R                   S:X  a  [        R                  U R                  5      $ U R                   S:X  a  SS KnU R                  X R                  5      $ U R                   S:X  a   U R                  [        U R                  5      $ [        e)Nbuiltinsmathr   r0   )r  __builtins__r   r   rE  r'  r0   r   )r%   rD   rE  s      r'   r}   _OpBuiltinPickleData.unpickle  sp    99
"##DII..YY& ((yy99YY'!((		::%%r)   )r   r  
r\   r]   r^   r_   rb   r    rU   r"   r}   rf   rn   r)   r'   r  r    s/    S   
&~ 
&& 
&r)   r  c                   8    \ rS rSrS\SS4S jrS\S\4S jrSr	g)	r  i  r   r   Nc                     Xl         g r,   r/  r0  s     r'   r    _OpOperatorPickleData.__init__  r2  r)   rD   c                 @    SS K nU R                  X R                  5      $ )Nr   )operatorr'  r   )r%   rD   rM  s      r'   r}   _OpOperatorPickleData.unpickle  s    $$Xyy99r)   r/  rH  rn   r)   r'   r  r    s(    S T :~ :& :r)   r  c                       \ rS rSrS\R
                  R                  SS4S jrS\R
                  R                  S\	S\R
                  R                  4S jr
S	rg)
r   i  r   r   Nc                     UR                   U l        UR                  U l        0 nUR                   H  n[        X25      X#'   M     [        UR                  5       5      U l        g r,   )_tracer_cls
tracer_cls_tracer_extrastracer_extrasnodesr   ra   values)r%   r   rU  r   s       r'   r    _GraphPickleData.__init__  sN    ++"1168KKD)$6EK  5<<>*
r)   r   rD   c                     [         R                  R                  XR                  U R                  5      n0 nU R
                   H  nUR                  X4U5      XE'   M     U$ r,   )r0   r1   r  rR  rT  rU  r}   )r%   r   rD   r   rU  nds         r'   r}   _GraphPickleData.unpickle  sN     r??D4F4FG68**BE.AEI  r)   )rU  rR  rT  )r\   r]   r^   r_   r0   r1   r  r    r2   rU   r}   rf   rn   r)   r'   r   r     sL    +ehhnn + +&	((&&	8F			r)   r   c                       \ rS rSr\S\S\R                  R                  S\	\
\\/\R                  R                  4   \	\\4   4   4S j5       rS\SS4S jrS	\S\4S
 jrSrg)r<   i  rO   r*   r   c                 B    U R                   U " U5      UR                  44$ r,   r|   r~   s      r'   r/   '_TracingContextPickleData.reduce_helper  s*     LLC''
 	
r)   contextNc                 6   UR                   U l         UR                  U l        UR                  U l        UR                  U l        UR                  U l        UR
                  U l        UR                  U l        UR                  U l        UR                  U l        g r,   )	module_contextframe_summary_stackloc_in_frameaot_graph_nameparams_flatparams_flat_unwrap_subclassesparams_unwrapped_to_flat_indexoutput_strides#force_unspec_int_unbacked_size_like)r%   r^  s     r'   r    "_TracingContextPickleData.__init__#  s    %44#*#>#> #00%44"..-4-R-R*.5.T.T+%4477 	0r)   rD   c                 b   [        UR                  5      nU R                  Ul        U R                  Ul        U R                  Ul        U R
                  Ul        U R                  Ul        U R                  Ul        U R                  Ul        U R                  Ul	        U R                  Ul
        U$ r,   )r   rS   r`  ra  rb  rc  rd  re  rf  rg  rh  )r%   rD   r^  s      r'   r}   "_TracingContextPickleData.unpickle9  s     !9!9:!%!4!4&*&>&>##00!%!4!4"..040R0R-151T1T.!%!4!444 	3 r)   )	rc  rh  ra  rb  r`  rg  rd  re  rf  )r\   r]   r^   r_   rc   r   r0   r;   r   ra   r   r   rU   r!   r/   r    r}   rf   rn   r)   r'   r<   r<     s    
"
).)E)E
	$')E)EEFd''(	*

 

 
4 
,~ . r)   r<   )=r   r   rI   ru   abcr   typingr   r   r   r   r   r	   typing_extensionsr
   r   r0   torch.utils._pytreeutils_pytreer   torch._guardsr   torch._subclasses.fake_tensorr   r   r   torch._subclasses.meta_utilsr   r   r   r   r   %torch.fx.experimental.symbolic_shapesr   torch.utils._mode_utilsr   r9   SymFloatr   Picklerr   rU   r"   r!   	UnpicklerrV   r8   r:   r.   r>   r3   r   r7   r  r  r  r  r  r   r<   rn   r)   r'   <module>rz     s<     	   C C ,  $ $ ( L L 
 3 : / Ku~~>	W$6>> W$tI I 3V< 
Bf&& 
B2 2>1 1>;
 ;
|) )X <" "JA# A#H} M - &= &$:M : @3 3r)   