
    Vh8                        d dl mZmZ d dlmZ d dlmZmZ d dlm	Z	 dZ
dZ G d de      Z G d	 d
e      Z G d de      Ze G d d             Z ed       G d de	             Ze G d d             Z ed       G d de	             Z ed       G d de	             Z ed       G d de	             Ze G d d             Z ed       G d de	             Z ed       G d d e	             Z ed       G d! d"e	             Ze G d# d$             Ze G d% d&             Z ed       G d' d(e	             Ze G d) d*             Ze G d+ d,             Z ed       G d- d.e	             Z G d/ d0e      Ze G d1 d2             Z e G d3 d4             Z!e G d5 d6             Z"e G d7 d8             Z# ed       G d9 d:e	             Z$e G d; d<             Z%e G d= d>             Z&e G d? d@             Z'e G dA dB             Z(e G dC dD             Z)e G dE dF             Z* ed       G dG dHe	             Z+e G dI dJ             Z,e G dK dL             Z-e G dM dN             Z.e G dO dP             Z/e G dQ dR             Z0e G dS dT             Z1e G dU dV             Z2 ed       G dW dXe	             Z3e G dY dZ             Z4e G d[ d\             Z5e G d] d^             Z6e G d_ d`             Z7e G da db             Z8e G dc dd             Z9e G de df             Z:e G dg dh             Z;e G di dj             Z<e G dk dl             Z=e G dm dn             Z>e G do dp             Z?e G dq dr             Z@ys)t    )	dataclassfield)IntEnum)	AnnotatedOptional)_Union)         c                   P    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZy)
ScalarTyper   r                  r
   r	   	   
                     N)__name__
__module____qualname__UNKNOWNBYTECHARSHORTINTLONGHALFFLOATDOUBLECOMPLEXHALFCOMPLEXFLOATCOMPLEXDOUBLEBOOLBFLOAT16UINT16FLOAT8E4M3FN
FLOAT8E5M2     J/home/dcms/DCMS/lib/python3.12/site-packages/torch/_export/serde/schema.pyr   r      sW    GDDE
CDDEFKLMDHFLJr0   r   c                   ,    e Zd ZdZdZdZdZdZdZdZ	dZ
y	)
Layoutr   r   r   r   r   r   r   r
   N)r   r   r   Unknown	SparseCoo	SparseCsr	SparseCsc	SparseBsr	SparseBsc_mkldnnStridedr/   r0   r1   r3   r3   %   s*    GIIIIIGGr0   r3   c                        e Zd ZdZdZdZdZdZy)MemoryFormatr   r   r   r   r   N)r   r   r   r4   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr/   r0   r1   r=   r=   0   s    GLNNr0   r=   c                   @    e Zd ZU eedf   ed<   dZeee   df   ed<   y)Devicer   typeN   index)	r   r   r   r   str__annotations__rF   r   intr/   r0   r1   rC   rC   8   s)    
CG
*.E9Xc]B&'.r0   rC   F)reprc                   J    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   y)SymExprHintr   as_intrE   as_boolr   as_floatN)r   r   r   r   rI   rH   boolfloatr/   r0   r1   rL   rL   >   s/    c2gtRx  r	""r0   rL   c                   @    e Zd ZU eedf   ed<   dZeee   df   ed<   y)SymExprr   expr_strNrE   hint)	r   r   r   r   rG   rH   rU   r   rL   r/   r0   r1   rS   rS   I   s*    R  15D)H[)2-
.5r0   rS   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymIntr   as_exprrE   rM   N)r   r   r   r   rS   rH   rI   r/   r0   r1   rW   rW   O   s     w{##c2gr0   rW   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymFloatr   rX   rE   rO   N)r   r   r   r   rS   rH   rQ   r/   r0   r1   rZ   rZ   T   s     w{##r	""r0   rZ   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymBoolr   rX   rE   rN   N)r   r   r   r   rS   rH   rP   r/   r0   r1   r\   r\   Z   s     w{##tRx  r0   r\   c                       e Zd ZU eedf   ed<   eee   df   ed<   eedf   ed<   ee	df   ed<   eee   d	f   ed
<   eedf   ed<   ee
df   ed<   y)
TensorMetar   dtyperE   sizesr   requires_grad(   device2   strides<   storage_offsetF   layoutN)r   r   r   r   r   rH   listrW   rP   rC   r3   r/   r0   r1   r^   r^   `   su    Z^$$T&\2%&&T2X&&fbj!!tF|R'((fbj))fbj!!r0   r^   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymIntArgumentr   as_namerE   rM   N)r   r   r   r   rG   rH   rI   r/   r0   r1   rl   rl   q   s     sBwc2gr0   rl   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymFloatArgumentr   rm   rE   rO   N)r   r   r   r   rG   rH   rQ   r/   r0   r1   ro   ro   |   s     sBwr	""r0   ro   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymBoolArgumentr   rm   rE   rN   N)r   r   r   r   rG   rH   rP   r/   r0   r1   rq   rq      s     sBwtRx  r0   rq   c                   "    e Zd ZU eedf   ed<   y)TensorArgumentr   nameNr   r   r   r   rG   rH   r/   r0   r1   rs   rs          
CG
r0   rs   c                   "    e Zd ZU eedf   ed<   y)TokenArgumentr   rt   Nru   r/   r0   r1   rx   rx      rv   r0   rx   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)OptionalTensorArgumentrE   	as_tensorr   as_noneN)r   r   r   r   rs   rH   rP   r/   r0   r1   rz   rz      s!    +,,tRx  r0   rz   c                   2    e Zd ZU eedf   ed<   ed   ed<   y)GraphArgumentr   rt   )GraphrE   graphNru   r/   r0   r1   r~   r~      s    
CG
[!!r0   r~   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)CustomObjArgumentr   rt   rE   	class_fqnNru   r/   r0   r1   r   r      s     
CG
b!!r0   r   c                   8   e Zd ZU eedf   ed<   eedf   ed<   eee   df   ed<   eedf   ed<   eee   d	f   ed
<   ee	df   ed<   eee	   df   ed<   ee
df   ed<   eee
   df   ed<   eedf   ed<   eee   df   ed<   eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed <   eee   d!f   ed"<   eed#f   ed$<   eee   d%f   ed&<   eed'f   ed(<   eee   d)f   ed*<   eed+f   ed,<   ee
d-f   ed.<   eed/f   ed0<   eee   d1f   ed2<   y3)4Argumentr   r|   rE   r{   r   
as_tensorsrd   rM   rh   as_intsP   rO   Z   	as_floatsd   	as_stringe   
as_stringsn   
as_sym_intx   as_sym_ints   as_scalar_type   as_memory_format   	as_layout   	as_device   rN      as_bools   as_sym_bool   as_sym_bools   as_graph   as_optional_tensors   as_custom_obj   as_operator   as_sym_float   as_sym_floatsN)r   r   r   r   rP   rH   rs   rj   rI   rQ   rG   rl   r   r=   r3   rC   rq   r~   rz   r   ro   r/   r0   r1   r   r      s   tRx  +,,$~.233c2gtCy"}%%r	""eb))c""$s)S.)).#-..4/455j#o..c 122%%%%tSy!!T
C((?C/00D13677s*++"4(>#?#DEE.34438$$,c122T"23S899r0   r   c                       e Zd ZdZdZdZy)ArgumentKindr   r   r   N)r   r   r   r   
POSITIONALKEYWORDr/   r0   r1   r   r      s    GJGr0   r   c                   T    e Zd ZU eedf   ed<   eedf   ed<   dZeee	   df   ed<   y)NamedArgumentr   rt   rE   argNr   kind)
r   r   r   r   rG   rH   r   r   r   r   r/   r0   r1   r   r      s;     CG
	8R<	  26D)H\*B.
/6r0   r   c                       e Zd ZU eedf   ed<   eee   df   ed<   eee   df   ed<   ee	eef   df   ed<   d	Z
eee   d
f   ed<   y	)Noder   targetrE   inputsr   outputsrb   metadataNrd   is_hop_single_tensor_return)r   r   r   r   rG   rH   rj   r   r   dictr   r   rP   r/   r0   r1   r   r      sk    c2gd=)2-..tH~r)**S#X*++AE8D>2+=!>Er0   r   c                   .   e Zd ZU eee   df   ed<   eee   df   ed<   eee   df   ed<   eee	e
f   df   ed<   eee	ef   d	f   ed
<   eee	ef   df   ed<   dZeedf   ed<    ee      Zeee	ef   df   ed<    ee      Zeee	ef   df   ed<   y)r   r   r   rE   r   r   nodesrb   tensor_valuesrd   sym_int_valuesrf   sym_bool_valuesFrh   is_single_tensor_returndefault_factoryr   custom_obj_valuesr   sym_float_valuesN)r   r   r   r   rj   r   rH   r   r   rG   r^   rW   r\   r   rP   r   r   r   r   rZ   r/   r0   r1   r   r      s    d8nb())tH~r)**T$Z^$$T#z/2B677d3;/344tCL12566
 49YtRx08EJ[_E`yc+<&<!=r!AB`;@QU;ViS(] 3R 78Vr0   r   c                   "    e Zd ZU eedf   ed<   y)UserInputSpecr   r   Nr   r   r   r   r   rH   r/   r0   r1   r   r      s     
8R<	  r0   r   c                   r    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed<   eed	f   ed
<   y)ConstantValuer   r|   rE   rM   r   rO   rb   r   rd   rN   N)	r   r   r   r   rP   rH   rI   rQ   rG   r/   r0   r1   r   r      sM    tRx  c2gr	""b!!tRx  r0   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToConstantInputSpecr   rt   rE   valueN)r   r   r   r   rG   rH   r   r/   r0   r1   r   r     s!    
CG
]B&''r0   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToParameterSpecr   r   rE   parameter_nameNr   r   r   r   rs   rH   rG   r/   r0   r1   r   r     !    	>2%	&&c2g&&r0   r   c                   J    e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   y)InputToBufferSpecr   r   rE   buffer_namer   
persistentN)r   r   r   r   rs   rH   rG   rP   r/   r0   r1   r   r     s0    	>2%	&&37##$(##r0   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToTensorConstantSpecr   r   rE   tensor_constant_nameNr   r/   r0   r1   r   r     s!    	>2%	&&#CG,,r0   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToCustomObjSpecr   r   rE   custom_obj_nameN)r   r   r   r   r   rH   rG   r/   r0   r1   r   r     s"    	$b(	))sBw''r0   r   c                   "    e Zd ZU eedf   ed<   y)InputTokenSpecr   r   Nr   r   r   r   rx   rH   r/   r0   r1   r   r   !      	="$	%%r0   r   c                       e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed<   ee	d	f   ed
<   ee
df   ed<   eedf   ed<   y)	InputSpecr   
user_inputrE   	parameterr   bufferrb   tensor_constantrd   
custom_objrh   tokenrf   constant_inputN)r   r   r   r   r   rH   r   r   r   r   r   r   r/   r0   r1   r   r   &  sw    -+,,-r122'+,,8"<==.233^R'((6:;;r0   r   c                   "    e Zd ZU eedf   ed<   y)UserOutputSpecr   r   Nr   r/   r0   r1   r   r   1  s    	8R<	  r0   r   c                   "    e Zd ZU eedf   ed<   y)LossOutputSpecr   r   N)r   r   r   r   rs   rH   r/   r0   r1   r   r   6  s    	>2%	&&r0   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)BufferMutationSpecr   r   rE   r   Nr   r/   r0   r1   r   r   ;  s!    	>2%	&&37##r0   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)GradientToParameterSpecr   r   rE   r   Nr   r/   r0   r1   r   r   A  r   r0   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)GradientToUserInputSpecr   r   rE   user_input_nameNr   r/   r0   r1   r   r   G  !    	>2%	&&sBw''r0   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)UserInputMutationSpecr   r   rE   r   Nr   r/   r0   r1   r   r   M  r   r0   r   c                   "    e Zd ZU eedf   ed<   y)OutputTokenSpecr   r   Nr   r/   r0   r1   r   r   S  r   r0   r   c                       e Zd ZU eedf   ed<   eedf   ed<   eedf   ed<   eedf   ed<   ee	d	f   ed
<   ee
df   ed<   eedf   ed<   y)
OutputSpecr   user_outputrE   loss_outputr   buffer_mutationrb   gradient_to_parameterrd   gradient_to_user_inputrf   user_input_mutationrh   r   N)r   r   r   r   r   rH   r   r   r   r   r   r   r/   r0   r1   r   r   X  sv    >2-..>2-..12566$%<b%@AA%&=r&ABB"#8"#<==_b())r0   r   c                   B    e Zd ZU eee   df   ed<   eee   df   ed<   y)GraphSignaturer   input_specsrE   output_specsN)r   r   r   r   rj   r   rH   r   r/   r0   r1   r  r  c  s+    4	?B.//D,b011r0   r  c                   B    e Zd ZU eee   df   ed<   eee   df   ed<   y)RangeConstraintr   min_valrE   max_valN)r   r   r   r   r   rI   rH   r/   r0   r1   r	  r	  i  s*    x}b())x}b())r0   r	  c                       e Zd ZU eee   df   ed<   eee   df   ed<   eedf   ed<   eedf   ed<   d	Zee	ee      d
f   ed<   y	)ModuleCallSignaturer   r   rE   r   r   in_specrb   out_specNrd   forward_arg_names)
r   r   r   r   rj   r   rH   rG   r  r   r/   r0   r1   r  r  o  sk    d8nb())tH~r)** sBwR   =Ay$s)!4b!89@r0   r  c                   @    e Zd ZU eedf   ed<   dZeee   df   ed<   y)ModuleCallEntryr   fqnNr   	signature)	r   r   r   r   rG   rH   r  r   r  r/   r0   r1   r  r  ~  s+    	37	>BIy"56:;Br0   r  c                   (    e Zd ZU eee   df   ed<   y)NamedTupleDefr   field_namesN)r   r   r   r   rj   rG   rH   r/   r0   r1   r  r    s    49b=))r0   r  c                       e Zd ZU eedf   ed<   eedf   ed<   eee   df   ed<    e	e
      Zee
eef   df   ed	<    e	e
      Zee
eef   d
f   ed<   y)GraphModuler   r   rd   r  rf   module_call_graphr   rb   r   rh   treespec_namedtuple_fieldsN)r   r   r   r   r   rH   r  rj   r  r   r   r   rG   r  r  r/   r0   r1   r  r    s~    UBY+,, !o!6!:;;.3D.IHiS#X*+IJO`dJe	$sM/A*BB*F Ger0   r  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SchemaVersionr   majorrE   minorN)r   r   r   r   rI   rH   r/   r0   r1   r  r    s     S"WS"Wr0   r  c                       e Zd ZU eedf   ed<   eeeef   df   ed<   eeee	f   df   ed<   ee
df   ed<    ee	      Zeee   d
f   ed<   dZeedf   ed<   y)ExportedProgramr   graph_modulerE   opset_versionr   range_constraintsrf   schema_versionr   rh   	verifiersz<=2.4r   torch_versionN)r   r   r   r   r  rH   r   rG   rI   r	  r  r   rj   r&  r'  r/   r0   r1   r!  r!    s    KO,,T#s(^R/00 c?&:!;R!?@@mR/00*/*EIycB'E(/M9S"W%/r0   r!  c                   ,    e Zd ZU eeeef   df   ed<   y)Programr   methodsN)r   r   r   r   r   rG   r!  rH   r/   r0   r1   r)  r)    s    tC013677r0   r)  c                       e Zd ZU eedf   ed<   eeeef   df   ed<   eedf   ed<   eeeef   df   ed<   eeeef   d	f   ed
<   eeeef   df   ed<   y)Modelr   rt   rE   tensorPathsrb   programrd   	delegatesrf   deviceAllocationMaprh   constantPathsN)r   r   r   r   rG   rH   r   r)  r/   r0   r1   r,  r,    s     CG
 4S>2-..w{## c7l+R/00"4S>2#566 T#s(^R/00r0   r,  c                       e Zd ZU eedf   ed<   eee   df   ed<   eee   df   ed<   dZeee	   df   ed	<   dZ
eee	   d
f   ed<   dZeee   df   ed<   y)AOTInductorModelPickleDatar   library_basenamer   input_namesr   output_namesNr   floating_point_input_dtyper   floating_point_output_dtyper   aot_inductor_model_is_cpu)r   r   r   r   rG   rH   rj   r7  r   rI   r8  r9  rP   r/   r0   r1   r3  r3    s      Q'' 49a<(( DIqL))
 ?C	(3-*: ;B?C8C=!+;!<C ?Cy$):;Br0   r3  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ExternKernelNoder   rt   rE   nodeN)r   r   r   r   rG   rH   r   r/   r0   r1   r;  r;    s"     CG

D"H
r0   r;  c                   (    e Zd ZU eee   df   ed<   y)ExternKernelNodesr   r   N)r   r   r   r   rj   r;  rH   r/   r0   r1   r>  r>    s    T*+R/00r0   r>  N)Adataclassesr   r   enumr   typingr   r   torch._export.serde.unionr   SCHEMA_VERSIONTREESPEC_VERSIONr   r3   r=   rC   rL   rS   rW   rZ   r\   r^   rl   ro   rq   rs   rx   rz   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r	  r  r  r  r  r  r!  r)  r,  r3  r;  r>  r/   r0   r1   <module>rE     s   )  & ,  
 (W 7  / / /
 #& # # 6 6 6
 V   #v # #
 !f ! !
 " " "  V   #v # # !f ! !
       !V ! !
 " " "
 " " " :v : :87  7 7 7 F F F W W W ! ! !
 !F ! ! ( ( (
 ' ' '
 $ $ $ - - -
 ( ( (
 & & & < < < ! ! ! ' ' ' $ $ $
 ' ' '
 ( ( (
 ( ( (
 & & & * * * 2 2 2
 * * *
 A A A C C C
 * * * 	f 	f 	f   
 0 0 0 8 8 8 1 1 1. C C C(   
 1 1 1r0   