o
    Zh8                     @   sV  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 deZG d	d
 d
eZG dd deZeG dd dZeddG dd de	ZeG dd dZeddG dd de	ZeddG dd de	ZeddG dd de	ZeG dd dZeddG dd de	ZeddG dd  d e	ZeddG d!d" d"e	ZeG d#d$ d$ZeG d%d& d&ZeddG d'd( d(e	ZeG d)d* d*ZeG d+d, d,ZeddG d-d. d.e	ZG d/d0 d0eZeG d1d2 d2Z eG d3d4 d4Z!eG d5d6 d6Z"eG d7d8 d8Z#eddG d9d: d:e	Z$eG d;d< d<Z%eG d=d> d>Z&eG d?d@ d@Z'eG dAdB dBZ(eG dCdD dDZ)eG dEdF dFZ*eddG dGdH dHe	Z+eG dIdJ dJZ,eG dKdL dLZ-eG dMdN dNZ.eG dOdP dPZ/eG dQdR dRZ0eG dSdT dTZ1eG dUdV dVZ2eddG dWdX dXe	Z3eG dYdZ dZZ4eG d[d\ d\Z5eG d]d^ d^Z6eG d_d` d`Z7eG dadb dbZ8eG dcdd ddZ9eG dedf dfZ:eG dgdh dhZ;eG didj djZ<eG dkdl dlZ=eG dmdn dnZ>eG dodp dpZ?eG dqdr drZ@dsS )t    )	dataclassfield)IntEnum)	AnnotatedOptional)_Union)         c                   @   sP   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S )
ScalarTyper   r
                  r	   r   	   
                     N)__name__
__module____qualname__UNKNOWNZBYTEZCHARZSHORTINTLONGZHALFFLOATDOUBLEZCOMPLEXHALFZCOMPLEXFLOATZCOMPLEXDOUBLEZBOOLZBFLOAT16ZUINT16ZFLOAT8E4M3FNZ
FLOAT8E5M2 r!   r!   I/var/www/auris/lib/python3.10/site-packages/torch/_export/serde/schema.pyr      s$    r   c                   @   s,   e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )
Layoutr   r
   r   r   r   r   r   r	   N)r   r   r   UnknownZ	SparseCooZ	SparseCsrZ	SparseCscZ	SparseBsrZ	SparseBscZ_mkldnnZStridedr!   r!   r!   r"   r#   %   s    r#   c                   @   s    e Zd ZdZdZdZdZdZdS )MemoryFormatr   r
   r   r   r   N)r   r   r   r$   ZContiguousFormatZChannelsLastZChannelsLast3dZPreserveFormatr!   r!   r!   r"   r%   0   s    r%   c                   @   6   e Zd ZU eedf ed< dZeee df ed< dS )Devicer   typeN   index)	r   r   r   r   str__annotations__r*   r   intr!   r!   r!   r"   r'   8      
 r'   F)reprc                   @   >   e Zd ZU eedf ed< eedf ed< eedf ed< dS )SymExprHintr   as_intr)   as_boolr   as_floatN)r   r   r   r   r-   r,   boolfloatr!   r!   r!   r"   r1   >      
 r1   c                   @   r&   )SymExprr   Zexpr_strNr)   hint)	r   r   r   r   r+   r,   r9   r   r1   r!   r!   r!   r"   r8   I   r.   r8   c                   @   .   e Zd ZU eedf ed< eedf ed< dS )SymIntr   as_exprr)   r2   N)r   r   r   r   r8   r,   r-   r!   r!   r!   r"   r;   O      
 r;   c                   @   r:   )SymFloatr   r<   r)   r4   N)r   r   r   r   r8   r,   r6   r!   r!   r!   r"   r>   T   r=   r>   c                   @   r:   )SymBoolr   r<   r)   r3   N)r   r   r   r   r8   r,   r5   r!   r!   r!   r"   r?   Z   r=   r?   c                   @   s   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< dS )
TensorMetar   Zdtyper)   sizesr   Zrequires_grad(   Zdevice2   strides<   Zstorage_offsetF   ZlayoutN)r   r   r   r   r   r,   listr;   r5   r'   r#   r!   r!   r!   r"   r@   `   s   
 r@   c                   @   r:   )SymIntArgumentr   as_namer)   r2   N)r   r   r   r   r+   r,   r-   r!   r!   r!   r"   rH   q   r=   rH   c                   @   r:   )SymFloatArgumentr   rI   r)   r4   N)r   r   r   r   r+   r,   r6   r!   r!   r!   r"   rJ   |   r=   rJ   c                   @   r:   )SymBoolArgumentr   rI   r)   r3   N)r   r   r   r   r+   r,   r5   r!   r!   r!   r"   rK      r=   rK   c                   @      e Zd ZU eedf ed< dS )TensorArgumentr   nameNr   r   r   r   r+   r,   r!   r!   r!   r"   rM         
 rM   c                   @   rL   )TokenArgumentr   rN   NrO   r!   r!   r!   r"   rQ      rP   rQ   c                   @   r:   )OptionalTensorArgumentr)   	as_tensorr   as_noneN)r   r   r   r   rM   r,   r5   r!   r!   r!   r"   rR      r=   rR   c                   @   s*   e Zd ZU eedf ed< ed ed< dS )GraphArgumentr   rN   )Graphr)   graphNrO   r!   r!   r!   r"   rU      s   
 rU   c                   @   .   e Zd ZU eedf ed< eedf ed< dS )CustomObjArgumentr   rN   r)   Z	class_fqnNrO   r!   r!   r!   r"   rY      r=   rY   c                   @   s  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< d3S )4Argumentr   rT   r)   rS   r   Z
as_tensorsrC   r2   rF   Zas_intsP   r4   Z   Z	as_floatsd   	as_stringe   Z
as_stringsn   Z
as_sym_intx   Zas_sym_ints   Zas_scalar_type   Zas_memory_format   Z	as_layout   Z	as_device   r3      Zas_bools   Zas_sym_bool   Zas_sym_bools   Zas_graph   Zas_optional_tensors   Zas_custom_obj   Zas_operator   Zas_sym_float   Zas_sym_floatsN)r   r   r   r   r5   r,   rM   rG   r-   r6   r+   rH   r   r%   r#   r'   rK   rU   rR   rY   rJ   r!   r!   r!   r"   rZ      s4   
 rZ   c                   @   s   e Zd ZdZdZdZdS )ArgumentKindr   r
   r   N)r   r   r   r   Z
POSITIONALZKEYWORDr!   r!   r!   r"   rp      s    rp   c                   @   sF   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< dS )NamedArgumentr   rN   r)   argNr   kind)
r   r   r   r   r+   r,   rZ   rs   r   rp   r!   r!   r!   r"   rq      s   
 rq   c                   @   sv   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< d	S )Noder   targetr)   inputsr   outputsrB   metadataNrC   is_hop_single_tensor_return)r   r   r   r   r+   r,   rG   rq   rZ   dictry   r   r5   r!   r!   r!   r"   rt      s   
 rt   c                   @   s   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dZeee	ef df ed< eedZeee	ef df ed< dS )rV   r   rv   r)   rw   r   nodesrB   Ztensor_valuesrC   Zsym_int_valuesrE   Zsym_bool_valuesFrF   is_single_tensor_returndefault_factoryr[   custom_obj_valuesr\   sym_float_valuesN)r   r   r   r   rG   rZ   r,   rt   rz   r+   r@   r;   r?   r|   r5   r   r   rY   r   r>   r!   r!   r!   r"   rV      s   
 "&rV   c                   @   rL   )UserInputSpecr   rr   Nr   r   r   r   rZ   r,   r!   r!   r!   r"   r      s   
 r   c                   @   s^   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
< dS )ConstantValuer   rT   r)   r2   r   r4   rB   r^   rC   r3   N)	r   r   r   r   r5   r,   r-   r6   r+   r!   r!   r!   r"   r      s   
 r   c                   @   r:   )InputToConstantInputSpecr   rN   r)   valueN)r   r   r   r   r+   r,   r   r!   r!   r!   r"   r     r=   r   c                   @   r:   )InputToParameterSpecr   rr   r)   parameter_nameNr   r   r   r   rM   r,   r+   r!   r!   r!   r"   r     r=   r   c                   @   r0   )InputToBufferSpecr   rr   r)   buffer_namer   
persistentN)r   r   r   r   rM   r,   r+   r5   r!   r!   r!   r"   r     r7   r   c                   @   r:   )InputToTensorConstantSpecr   rr   r)   Ztensor_constant_nameNr   r!   r!   r!   r"   r     r=   r   c                   @   r:   )InputToCustomObjSpecr   rr   r)   Zcustom_obj_nameN)r   r   r   r   rY   r,   r+   r!   r!   r!   r"   r     r=   r   c                   @   rL   )InputTokenSpecr   rr   Nr   r   r   r   rQ   r,   r!   r!   r!   r"   r   !  rP   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< dS )	InputSpecr   Z
user_inputr)   Z	parameterr   bufferrB   Ztensor_constantrC   Z
custom_objrF   tokenrE   Zconstant_inputN)r   r   r   r   r   r,   r   r   r   r   r   r   r!   r!   r!   r"   r   &     
 r   c                   @   rL   )UserOutputSpecr   rr   Nr   r!   r!   r!   r"   r   1  rP   r   c                   @   rL   )LossOutputSpecr   rr   N)r   r   r   r   rM   r,   r!   r!   r!   r"   r   6  rP   r   c                   @   r:   )BufferMutationSpecr   rr   r)   r   Nr   r!   r!   r!   r"   r   ;  r=   r   c                   @   r:   )GradientToParameterSpecr   rr   r)   r   Nr   r!   r!   r!   r"   r   A  r=   r   c                   @   r:   )GradientToUserInputSpecr   rr   r)   user_input_nameNr   r!   r!   r!   r"   r   G  r=   r   c                   @   r:   )UserInputMutationSpecr   rr   r)   r   Nr   r!   r!   r!   r"   r   M  r=   r   c                   @   rL   )OutputTokenSpecr   rr   Nr   r!   r!   r!   r"   r   S  rP   r   c                   @   r   )
OutputSpecr   Zuser_outputr)   Zloss_outputr   Zbuffer_mutationrB   Zgradient_to_parameterrC   Zgradient_to_user_inputrE   Zuser_input_mutationrF   r   N)r   r   r   r   r   r,   r   r   r   r   r   r   r!   r!   r!   r"   r   X  r   r   c                   @   s6   e Zd ZU eee df ed< eee df ed< dS )GraphSignaturer   Zinput_specsr)   Zoutput_specsN)r   r   r   r   rG   r   r,   r   r!   r!   r!   r"   r   c     
 r   c                   @   s6   e Zd ZU eee df ed< eee df ed< dS )RangeConstraintr   Zmin_valr)   Zmax_valN)r   r   r   r   r   r-   r,   r!   r!   r!   r"   r   i  r   r   c                   @   sr   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< d	S )ModuleCallSignaturer   rv   r)   rw   r   Zin_specrB   Zout_specNrC   forward_arg_names)
r   r   r   r   rG   rZ   r,   r+   r   r   r!   r!   r!   r"   r   o  s   
  r   c                   @   r&   )ModuleCallEntryr   ZfqnNr   	signature)	r   r   r   r   r+   r,   r   r   r   r!   r!   r!   r"   r   ~  r.   r   c                   @   "   e Zd ZU eee df ed< dS )NamedTupleDefr   field_namesN)r   r   r   r   rG   r+   r,   r!   r!   r!   r"   r        
 r   c                   @   s   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Zee
eef df ed	< e	e
dZee
eef d
f ed< dS )GraphModuler   rW   rC   r   rE   Zmodule_call_graphr}   rB   rx   rF   treespec_namedtuple_fieldsN)r   r   r   r   rV   r,   r   rG   r   r   rz   rx   r+   r   r   r!   r!   r!   r"   r     s   
 "&r   c                   @   rX   )SchemaVersionr   majorr)   minorN)r   r   r   r   r-   r,   r!   r!   r!   r"   r     r=   r   c                   @   s   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d	Zeee d
f ed< dZeedf ed< dS )ExportedProgramr   Zgraph_moduler)   Zopset_versionr   Zrange_constraintsrE   Zschema_versionr}   rF   	verifiersz<=2.4r[   torch_versionN)r   r   r   r   r   r,   rz   r+   r-   r   r   r   rG   r   r   r!   r!   r!   r"   r     s   
 r   c                   @   s&   e Zd ZU eeeef df ed< dS )Programrj   methodsN)r   r   r   r   rz   r+   r   r,   r!   r!   r!   r"   r     s   
 r   c                   @   s   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< dS )Modelr   rN   r)   ZtensorPathsrB   programrC   Z	delegatesrE   ZdeviceAllocationMaprF   ZconstantPathsN)r   r   r   r   r+   r,   rz   r   r!   r!   r!   r"   r     s   
 r   c                   @   s   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< dS )AOTInductorModelPickleDatar
   Zlibrary_basenamer   Zinput_namesr   Zoutput_namesNr   floating_point_input_dtyper   floating_point_output_dtyper   aot_inductor_model_is_cpu)r   r   r   r   r+   r,   rG   r   r   r-   r   r   r5   r!   r!   r!   r"   r     s   
 r   c                   @   r:   )ExternKernelNoder   rN   r)   nodeN)r   r   r   r   r+   r,   rt   r!   r!   r!   r"   r     s   
 r   c                   @   r   )ExternKernelNodesr   r{   N)r   r   r   r   rG   r   r,   r!   r!   r!   r"   r     r   r   N)Adataclassesr   r   enumr   typingr   r   Ztorch._export.serde.unionr   ZSCHEMA_VERSIONZTREESPEC_VERSIONr   r#   r%   r'   r1   r8   r;   r>   r?   r@   rH   rJ   rK   rM   rQ   rR   rU   rY   rZ   rp   rq   rt   rV   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"   <module>   s   




