
    qi*:                        d dl mZmZ d dlmZ d dlmZmZ d dlm	Z	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
 G d de	             Ze G d d             Ze
 G d de	             Ze
 G d de	             Ze
 G d de	             Ze G d d             Ze
 G d de	             Ze
 G d de	             Ze
 G d d e	             Ze G d! d"             Ze G d# d$             Ze
 G d% d&e	             Ze G d' d(             Ze G d) d*             Ze G d+ d,             Ze
 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
 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
 G dG dHe	             Z-e G dI dJ             Z.e G dK dL             Z/e G dM dN             Z0e G dO dP             Z1e G dQ dR             Z2e G dS dT             Z3e G dU dV             Z4e G dW dX             Z5e
 G dY dZe	             Z6e G d[ d\             Z7e G d] d^             Z8e G d_ d`             Z9e G da db             Z:e G dc dd             Z;e 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             ZAe G dq dr             ZBe G ds dt             ZCyu)v    )	dataclassfield)IntEnum)	AnnotatedOptional)_Union_union_dataclass)         c                   X    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dZdZy)
ScalarTyper   r                     r
   	   
                            N)__name__
__module____qualname__UNKNOWNBYTECHARSHORTINTLONGHALFFLOATDOUBLECOMPLEXHALFCOMPLEXFLOATCOMPLEXDOUBLEBOOLBFLOAT16UINT16FLOAT8E4M3FN
FLOAT8E5M2FLOAT8E4M3FNUZFLOAT8E5M2FNUZ     `/home/ubuntu/crypto_trading_bot/.venv/lib/python3.12/site-packages/torch/_export/serde/schema.pyr   r      sa    GDDE
CDDEFKLMDHFLJNNr6   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Stridedr5   r6   r7   r9   r9   (   s*    GIIIIIGGr6   r9   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!   r:   ContiguousFormatChannelsLastChannelsLast3dPreserveFormatr5   r6   r7   rC   rC   3   s    GLNNr6   rC   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__rL   r   intr5   r6   r7   rI   rI   ;   s)    
CG
*.E9Xc]B&'.r6   rI   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)SymExprHintr   as_intrK   as_boolr   as_floatN)r   r    r!   r   rO   rN   boolfloatr5   r6   r7   rQ   rQ   A   s/    c2gtRx  r	""r6   rQ   c                   @    e Zd ZU eedf   ed<   dZeee   df   ed<   y)SymExprr   expr_strNrK   hint)	r   r    r!   r   rM   rN   rZ   r   rQ   r5   r6   r7   rX   rX   L   s*    R  15D)H[)2-
.5r6   rX   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymIntr   as_exprrK   rR   N)r   r    r!   r   rX   rN   rO   r5   r6   r7   r\   r\   R   s     w{##c2gr6   r\   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymFloatr   r]   rK   rT   N)r   r    r!   r   rX   rN   rV   r5   r6   r7   r_   r_   X   s     w{##r	""r6   r_   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymBoolr   r]   rK   rS   N)r   r    r!   r   rX   rN   rU   r5   r6   r7   ra   ra   ^   s     w{##tRx  r6   ra   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   dtyperK   sizesr   requires_grad(   device2   strides<   storage_offsetF   layoutN)r   r    r!   r   r   rN   listr\   rU   rI   r9   r5   r6   r7   rc   rc   d   su    Z^$$T&\2%&&T2X&&fbj!!tF|R'((fbj))fbj!!r6   rc   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymIntArgumentr   as_namerK   rR   N)r   r    r!   r   rM   rN   rO   r5   r6   r7   rq   rq   u   s     sBwc2gr6   rq   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymFloatArgumentr   rr   rK   rT   N)r   r    r!   r   rM   rN   rV   r5   r6   r7   rt   rt      s     sBwr	""r6   rt   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SymBoolArgumentr   rr   rK   rS   N)r   r    r!   r   rM   rN   rU   r5   r6   r7   rv   rv      s     sBwtRx  r6   rv   c                   "    e Zd ZU eedf   ed<   y)TensorArgumentr   nameNr   r    r!   r   rM   rN   r5   r6   r7   rx   rx          
CG
r6   rx   c                   "    e Zd ZU eedf   ed<   y)TokenArgumentr   ry   Nrz   r5   r6   r7   r}   r}      r{   r6   r}   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)OptionalTensorArgumentrK   	as_tensorr   as_noneN)r   r    r!   r   rx   rN   rU   r5   r6   r7   r   r      s!    +,,tRx  r6   r   c                   2    e Zd ZU eedf   ed<   ed   ed<   y)GraphArgumentr   ry   )GraphrK   graphNrz   r5   r6   r7   r   r      s    
CG
[!!r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)CustomObjArgumentr   ry   rK   	class_fqnNrz   r5   r6   r7   r   r      s     
CG
b!!r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ComplexValuer   realrK   imagN)r   r    r!   r   rV   rN   r5   r6   r7   r   r      s     
E2I

E2I
r6   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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<   eed3f   ed4<   eed5f   ed6<   eeee      d7f   ed8<   eee
d f   d9f   ed:<   y;)<Argumentr   r   rK   r   r   
as_tensorsri   rR   rm   as_intsP   rT   Z   	as_floatsd   	as_stringe   
as_stringsn   
as_sym_intx   as_sym_ints   as_scalar_type   as_memory_format   	as_layout   	as_device   rS      as_bools   as_sym_bool   as_sym_bools   as_graph   as_optional_tensors   as_custom_obj   as_operator   as_sym_float   as_sym_floats   as_optional_tensori  
as_complexi  as_int_listsi"  as_string_to_argumentN)r   r    r!   r   rU   rN   rx   ro   rO   rV   rM   rq   r   rC   r9   rI   rv   r   r   r   rt   r   dictr5   r6   r7   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9!"8#"=>>,+,,DcOS011$T#z/%:C%?@@r6   r   c                       e Zd ZdZdZdZy)ArgumentKindr   r   r   N)r   r    r!   r"   
POSITIONALKEYWORDr5   r6   r7   r   r      s    GJGr6   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   ry   rK   argNr   kind)
r   r    r!   r   rM   rN   r   r   r   r   r5   r6   r7   r   r      s;     CG
	8R<	  26D)H\*B.
/6r6   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   targetrK   inputsr   outputsrg   metadataNri   is_hop_single_tensor_return)r   r    r!   r   rM   rN   ro   r   r   r   r   r   rU   r5   r6   r7   r   r      sk    c2gd=)2-..tH~r)**S#X*++AE8D>2+=!>Er6   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   rK   r   r   nodesrg   tensor_valuesri   sym_int_valuesrk   sym_bool_valuesFrm   is_single_tensor_returndefault_factoryr   custom_obj_valuesr   sym_float_valuesN)r   r    r!   r   ro   r   rN   r   r   rM   rc   r\   ra   r   rU   r   r   r   r   r_   r5   r6   r7   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Fyc+<&<!=r!AB  <AQU;ViS(] 3R 78Vr6   r   c                   "    e Zd ZU eedf   ed<   y)UserInputSpecr   r   Nr   r    r!   r   r   rN   r5   r6   r7   r   r     s     
8R<	  r6   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   rK   rR   r   rT   rg   r   ri   rS   N)	r   r    r!   r   rU   rN   rO   rV   rM   r5   r6   r7   r   r     sM    tRx  c2gr	""b!!tRx  r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToConstantInputSpecr   ry   rK   valueN)r   r    r!   r   rM   rN   r   r5   r6   r7   r   r     s!    
CG
]B&''r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToParameterSpecr   r   rK   parameter_nameNr   r    r!   r   rx   rN   rM   r5   r6   r7   r   r     !    	>2%	&&c2g&&r6   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   rK   buffer_namer   
persistentN)r   r    r!   r   rx   rN   rM   rU   r5   r6   r7   r   r      s0    	>2%	&&37##$(##r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToTensorConstantSpecr   r   rK   tensor_constant_nameNr   r5   r6   r7   r   r   '  s!    	>2%	&&#CG,,r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)InputToCustomObjSpecr   r   rK   custom_obj_nameN)r   r    r!   r   r   rN   rM   r5   r6   r7   r   r   -  s"    	$b(	))sBw''r6   r   c                   "    e Zd ZU eedf   ed<   y)InputTokenSpecr   r   Nr   r    r!   r   r}   rN   r5   r6   r7   r   r   3      	="$	%%r6   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_inputrK   	parameterr   bufferrg   tensor_constantri   
custom_objrm   tokenrk   constant_inputN)r   r    r!   r   r   rN   r   r   r   r   r   r   r5   r6   r7   r   r   8  sw    -+,,-r122'+,,8"<==.233^R'((6:;;r6   r   c                   "    e Zd ZU eedf   ed<   y)UserOutputSpecr   r   Nr   r5   r6   r7   r   r   C  s    	8R<	  r6   r   c                   "    e Zd ZU eedf   ed<   y)LossOutputSpecr   r   N)r   r    r!   r   rx   rN   r5   r6   r7   r   r   H  s    	>2%	&&r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)BufferMutationSpecr   r   rK   r   Nr   r5   r6   r7   r   r   M  s!    	>2%	&&37##r6   r   c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ParameterMutationSpecr   r   rK   r   Nr   r5   r6   r7   r  r  S  r   r6   r  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)GradientToParameterSpecr   r   rK   r   Nr   r5   r6   r7   r  r  Y  r   r6   r  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)GradientToUserInputSpecr   r   rK   user_input_nameNr   r5   r6   r7   r  r  _  !    	>2%	&&sBw''r6   r  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)UserInputMutationSpecr   r   rK   r  Nr   r5   r6   r7   r	  r	  e  r  r6   r	  c                   "    e Zd ZU eedf   ed<   y)OutputTokenSpecr   r   Nr   r5   r6   r7   r  r  k  r   r6   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<   eedf   ed<   y)
OutputSpecr   user_outputrK   loss_outputr   buffer_mutationrg   gradient_to_parameterri   gradient_to_user_inputrk   user_input_mutationrm   r   r   parameter_mutationN)r   r    r!   r   r   rN   r   r   r  r  r	  r  r  r5   r6   r7   r  r  p  s    >2-..>2-..12566$%<b%@AA%&=r&ABB"#8"#<==_b())!"7";<<r6   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_specsrK   output_specsN)r   r    r!   r   ro   r   rN   r  r5   r6   r7   r  r  |  s+    4	?B.//D,b011r6   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_valrK   max_valN)r   r    r!   r   r   rO   rN   r5   r6   r7   r  r    s*    x}b())x}b())r6   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   rK   r   r   in_specrg   out_specNri   forward_arg_names)
r   r    r!   r   ro   r   rN   rM   r!  r   r5   r6   r7   r  r    sk    d8nb())tH~r)** sBwR   =Ay$s)!4b!89@r6   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   rM   rN   r%  r   r  r5   r6   r7   r#  r#    s+    	37	>BIy"56:;Br6   r#  c                   (    e Zd ZU eee   df   ed<   y)NamedTupleDefr   field_namesN)r   r    r!   r   ro   rM   rN   r5   r6   r7   r'  r'    s    49b=))r6   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   ri   r%  rk   module_call_graphr   rg   r   rm   treespec_namedtuple_fieldsN)r   r    r!   r   r   rN   r  ro   r#  r   r   r   rM   r,  r'  r5   r6   r7   r*  r*    s    UBY+,, !o!6!:;;.3D.IHiS#X*+IJOK	$sM/A*BB*F G r6   r*  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)SchemaVersionr   majorrK   minorN)r   r    r!   r   rO   rN   r5   r6   r7   r.  r.    s,    R  R r6   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<    ee	      Zeee   df   ed<   y)ExportedProgramr   graph_modulerK   opset_versionr   range_constraintsrk   schema_versionr   rm   	verifiersz<=2.4r   torch_versionr   guards_codeN)r   r    r!   r   r*  rN   r   rM   rO   r  r.  r   ro   r7  r8  r9  r5   r6   r7   r2  r2    s    KO,,T#s(^R/00 c?&:!;R!?@@mR/00*/*EIycB'E(/M9S"W%/,1$,GK49b=)Gr6   r2  c                   d    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e   df   ed<   y	)
PayloadMetar   	path_namerK   is_paramr   
use_picklerg   tensor_metaN)	r   r    r!   r   rM   rN   rU   r   rc   r5   r6   r7   r;  r;    sH     b!!b!! $(##8J/344r6   r;  c                   ,    e Zd ZU eeeef   df   ed<   y)PayloadConfigr   configN)r   r    r!   r   r   rM   r;  rN   r5   r6   r7   rA  rA    s    d3+,b011r6   rA  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   rM   rN   ro   rH  r   rO   rI  rJ  rU   r5   r6   r7   rD  rD    s      Q'' 49a<(( DIqL))
 ?C	(3-*: ;B?C8C=!+;!<C ?Cy$):;Br6   rD  c                   6    e Zd ZU eedf   ed<   eedf   ed<   y)ExternKernelNoder   ry   rK   nodeN)r   r    r!   r   rM   rN   r   r5   r6   r7   rL  rL    s"     CG

D"H
r6   rL  c                   (    e Zd ZU eee   df   ed<   y)ExternKernelNodesr   r   N)r   r    r!   r   ro   rL  rN   r5   r6   r7   rO  rO    s    T*+R/00r6   rO  N)Ddataclassesr   r   enumr   typingr   r   torch._export.serde.unionr   r	   SCHEMA_VERSIONTREESPEC_VERSIONr   r9   rC   rI   rQ   rX   r\   r_   ra   rc   rq   rt   rv   rx   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*  r.  r2  r;  rA  rD  rL  rO  r5   r6   r7   <module>rV     sy   )  & >  
 ,W 7  / / /
 #& # # 6 6 6
 V  
 #v # #
 !f ! !
 " " "  V   #v # # !f ! !
       !V ! !
 " " "
 " " "
    Av A A@7  7 7 7 F F F W W W$ ! ! !
 !F ! ! ( ( (
 ' ' '
 $ $ $ - - -
 ( ( (
 & & & < < < ! ! ! ' ' ' $ $ $
 ' ' '
 ' ' '
 ( ( (
 ( ( (
 & & & = = = 2 2 2
 * * *
 A A A C C C
 * * *        H H H$ 	5 	5 	5 2 2 2 C C C*    1 1 1r6   