
    JTh^G                    |   % S SK Jr  S SKrS SKJr  S SKJr  S SKJr  S SK	J
r
Jr  S SKJr  S SKJrJrJrJrJrJrJrJrJrJrJrJr  S S	KJr  \(       a  S S
KJr  \R>                  S\R@                  S\RB                  S\RD                  S\RF                  S\RH                  S\RJ                  S\RL                  S\RN                  S\RP                  S\RR                  S\RT                  S0r+\R>                  S\R@                  S\RB                  S\RD                  S\RF                  S\RH                  S\RJ                  S\RL                  S\RN                  S\RP                  S\RR                  S\RT                  S0r,\R>                  S\R@                  S\RB                  S\RD                  S\RF                  S\RH                  S\RJ                  S\RL                  S\RN                  S\RP                  S\RR                  S \RT                  S!0r- S4       S5S" jjr.S6S# jr/S7S$ jr0S8S% jr10 q2S&\3S''           S9S( jr4      S:S) jr5      S;S* jr6            S<S+ jr7            S=S, jr8S>S- jr9              S?S. jr:\" S/S09 " S1 S25      5       r; S@               SAS3 jjr<g)B    )annotationsN)	dataclass)TYPE_CHECKING)DispatcherSignature)CppSignatureCppSignatureGroup)method_with_native_function)ArgumentBackendIndexBaseTyBaseTypeDispatchKeyFunctionSchemaListTypeNativeFunctionNativeFunctionsGroupOperatorNameOptionalTypeType)mapMaybe)SequenceAtenTensorHandleint32_tint64_tdoublezconst char*AtenGeneratorHandle
at::Tensorboolzc10::SymIntzc10::Scalarzc10::string_viewzc10::DeviceIndexzc10::Layoutzc10::MemoryFormatzc10::ScalarTypezat::Generatorresolve_tensor_dispatch_flags zstatic_cast<c10::DeviceIndex>zstatic_cast<c10::Layout>zstatic_cast<c10::MemoryFormat>zstatic_cast<c10::ScalarType>z&*generator_handle_to_generator_pointerc                   [        U [        5      (       a  U R                  [        ;   a  U R                  [        R
                  :X  a  U(       a  SU S3/nO4[        U R                     (       a  [        U R                      SU S3OU/n[        U R                     /U/[        U R                     /U4$ U R                  [        R                  :X  a  SS/XS-   /S/SU SU S	3/4$ [        S
[        U 5       35      e[        U [        5      (       GaE  [        U R                  U5      u  pEpgSn/ n	/ n
U GH  nXH   S-   XH'   UR                  S5      (       aL  U	R                  SU S35        U[!        S5      S nU
R                  SU SXX    SXXS-       S35        US-  nMo  US:X  a7  U	R                  S5        U
R                  SXX    SXXS-       S35        US-  nM  US:X  a3  U	R                  SU S35        U
R                  SXX    S35        US-  nM  U	R                  SU S35        U
R                  SU SXX    S35        US-  nGM     UUU	U
4$ [        U ["        5      (       Ga2  [        U R                  U5      u  pEpm[!        U5      S:X  d   S[        U 5      -   5       eSUS    S3US'   UR                  S5        US   nUR                  US-   5        US   n/ nUS:X  a2  U R$                  c   eUR                  S U R$                   SU S35        OfUS:X  a   U(       d  UR                  S!U SU S"35        O@US#:X  a  UR                  S$U S%U S&U SU S'3	5        OUR                  S U SU SU S"35        U Vs/ s H	  nSU S3PM     nnUUUU4$ [        S([        U 5       S)35      es  snf )*Nz!*tensor_handle_to_tensor_pointer()(r   _index_zc10::Devicez)c10::Device(static_cast<c10::DeviceType>(z!), static_cast<c10::DeviceIndex>(z	_index_))zTODO: add support for arg type r   *zc10::ArrayRef<z::std::optional<>zpointer_to_optional_list<z>(,       z::std::optional<c10::Device>zpointer_to_optional_device(r   zresolve_tensor_dispatch_flags(zpointer_to_optional<z'ListType with unsupported element type zconst r   _len_r   zpointer_to_list<z#resolve_tensor_list_dispatch_flags(z_len_)z::std::optional<at::Tensor>z
c10::List<z>(c10::ArrayRef<z%>(resolve_tensor_list_dispatch_flags(z_len_)))zArgument type z not supported!)
isinstancer   namebase_type_to_c_typer   Tensorbase_type_to_callsite_exprbase_type_to_aten_typeDeviceNotImplementedErrorreprr   convert_arg_type_and_nameelem
startswithappendlenr   size)typr-   is_writecallsite_exprc_typesnames
aten_typescallsite_exprsjnew_aten_typesnew_callsite_exprs	aten_type	base_type_atypets                   P/var/www/auris/envauris/lib/python3.13/site-packages/torchgen/gen_aoti_c_shim.pyr5   r5   N   sq   
 #x  88**xx6==(X $ETF!!L M 2#((; 2#((;<AdV1E! %SXX./'12	  XX&I&i'(?vEfgkflluv	  &(GS	{&STT	C	&	&5NHHd6
2
 #I c)GJ##$455%%(81&EF%c*:&;bA	"))/	{"UXJbSTuVWX Qm+%%&DE"))1%(2eEl^1M Ql*%%(81&EF")),J58*TU*VWQ%%(81&EF"))*9+RzC Q7 $< 	
 	
 
C	"	"(A#((D(Q%
7|q W"KdSVi"WW  gaj\+
y!QxTG^$1F? 88'''!!$4SXXJba"HIl"8!!5dV2dV6J 33!!UG#3E7:_`d_eeghlgmmuv !!$4UG2dV2dV6"RS5?@Zqc+Z
@	
 	
 tCykI
JJ As   N;c                Z    [        X5       VVs/ s H  u  p#US-   U-   PM     snn$ s  snnf )N )zip)typesr?   r;   r-   s       rJ   zip_type_and_namerO      s,    .1%.?@.?C#I.?@@@s   'c                    / n/ n/ nU  Hd  n[        UR                  UR                  UR                  5      u  pVpxUR	                  U5        UR	                  U5        UR	                  U5        Mf     [        X5      U4$ N)r5   typer-   r<   extendrO   )	flat_argumentsrN   	new_namesrA   arg	new_typesr?   rG   rD   s	            rJ   gen_argumentsrX      sz    EIN2KHHchh3
/	! 	Y01  U.>>    c           
     b   / n/ n[        U R                  5       H  u  p4UR                  SU 35        [        UR                  [
        5      (       aO  UR                  R                  [        ;   a1  UR                  [        UR                  R                     S-   5        M  [        S[        UR                  5       35      e   SS jnSnU R                  R                  5       nS H  nX;   d  M
  Sn  O   / n	[        U R                  5       H  u  p4[        U5      S:X  a  S	OS
U S3n
[        UR                  [
        5      (       d   eU" UR                  U
5      nU(       a!  U	R                  SX#    SX#    SU S35        Mx  U	R                  SX#    SU S35        M     [        X5      U	4$ )Nretr%   z"TODO: add support for return type c                    U R                   [        R                  :X  a  SU S3$ U R                   [        R                  :X  a  U S3$ U R                   [        R                  :X  a  U S3$ U$ )Nznew_tensor_handle(std::move(z));z.expect_int()z.toDouble())r-   r   r/   SymIntScalar)r;   vals     rJ   convert_return#gen_returns.<locals>.convert_return   s`    88v}}$1#c::XX&U-((XX&U+&&JrY   F)#_scaled_dot_product_flash_attention'_scaled_dot_product_efficient_attention#_scaled_dot_product_cudnn_attention0_scaled_dot_product_fused_attention_overrideableconvolution_backwardTr)   
tmp_resultz	std::get<z>(tmp_result)zif (z) { *z = z; };)r;   r   r_   strreturnri   )	enumeratereturnsr8   r,   rR   r   r-   r.   r3   r4   unambiguous_namer9   rO   )schemarN   r?   idxr[   r`   ret_pointer_can_be_nullrm   r-   rA   tmprvals               rJ   gen_returnsrs      s   EEfnn-s3%[!chh))chhmm?R.RLL,SXX]];cAB%4T#((^4DE  . ${{335 #&*# !#Nfnn-!%jAolYse=3Q#((H----chh,"!!DF5:,c$t"TU!!Aej\TF!"<= . U*N::rY   z+dict[tuple[str, str, str], tuple[str, str]]declaration_definition_cachec                   U R                   R                  5       nX1U4[        ;   a  [        X1U4   $ U R                  5       (       a<  [	        / U R
                  R                  QU R
                  R                  Q5      u  pE/ nOh[	        U R
                  R                  5      u  pEU R                   R                   R                  (       a  / / 4O
[        U 5      u  pvUR                  U5        SU SU SSR                  U5       S3nU(       a  SOSn	U(       a  SSR                  U5      -   OSn
SU S	U	 U S
[        R                  " SR                  U5      S5       S[        R                  " U
S5       S3
nUU4[        X1U4'   X4$ )NzAOTITorchError aoti_torch_rG   r#   r(   r"   zauto tmp_result = r    
z< {
    AOTI_TORCH_CONVERT_EXCEPTION_TO_ERROR_CODE({
        z(
z            z
        );z        z
    });
}
)r-   rm   rt   	is_out_fnrX   	argumentsoutflat_non_outflat_allinplacers   rS   jointextwrapindent)rn   devicebackend_call	func_nameargsrA   ret_assignmentsret_declarationsdeclarationrg   ret_assignments_str
definitions               rJ   gen_declaration_and_definitionr     s    ,,.I 	<(,HH+Y,MNN  -Cf""CV%5%5%B%BC 
 &(,V-=-=-F-FG ((00RHk&6I 	* 	$%.vha	{!DIIdOCTTUVK)8%bJ?N$?!;;TV 		\N #	>*N; < =??.
;
< =J 	G )\!BC ""rY   c                   [         R                  " UR                  5      n [        R                  " USSS9nU R
                  (       a,  UR                  R                  5       (       a  UR                  nOUR                  nUc   eU$ )NF)methodfallback_binding)	r   from_schemafuncr   from_native_functionsymint
has_symintsymint_signature	signature)sigfcpp_sigscpp_sigs       rJ   *gen_static_dispatch_backend_call_signaturer   ;  sq     
)
)!&&
1C 55	%%H zzaff''))++$$NrY   c                    [         R                  " U R                  5      n[        X 5      nSUR                  R                  5        SUR                  5        3$ )Nzat::z::)r   r   r   r   dispatch_keylowerr-   )r   backend_indexr   r   s       rJ    gen_static_dispatch_backend_callr   K  sL     
)
)!&&
1C8@G-,,2245R7GHHrY   c                >   S nX2   R                  U 5      (       dA  U R                  b:  U R                  U;   a*  X2   R                  XR                     5      (       a  X2   nU$ U(       a  U$ U[        R                     R                  U 5      (       a  U[        R                     nU$ U[        R                     R                  U 5      (       a  U[        R                     nU$ U[        R
                     R                  U 5      (       a  U[        R
                     nU$ rQ   )
has_kernelstructured_delegater   CompositeExplicitAutograd&CompositeExplicitAutogradNonFunctionalCompositeImplicitAutogradr   func_group_mappingr   backend_indicesextend_aoti_c_shimr   s         rJ   get_backend_index_for_aotir   T  s    M$//55  ,$$(::)44778
 
 (5( !   [BBCNNtTT+K,Q,QRM  >>

*T
 ,BBM  [BBCNNtTT+K,Q,QRMrY   c                |    [        XX#U5      nUc  S $ SU R                   SUR                  R                  5        S3$ )Nz#include <ATen/ops/rG   z_dispatch.h>)r   	root_namer   r   r   s         rJ   get_header_for_aotir   z  sW     /,ASM
   	 #4>>"2!M4N4N4T4T4V3WWcdrY   c                L   U R                   R                  R                  (       aP  U R                   SU R                   R                  R                   SU R                   R                  R                   3$ U R                   SU R                   R                  R                   S3$ )N.z.default)r   r-   overload_name	namespace)r   s    rJ   get_fallback_op_namer     s~     99>>'' >>
!DIINN//0$))..2N2N1OP q!4!4 5X>rY   c                    [        XX#U5      nUc  g U R                  nUR                  5       n[        U U5      n	 U(       a  [	        XxU	5      u  pSU
 S3$ [	        XxU	5      u  pU$ ! [
         a     g f = f)NzAOTI_TORCH_EXPORT rh   )r   r   r   r   r   r3   )r   r   r   r   headerr   r   rn   r   r   r   rG   r   s                rJ   
gen_c_shimr     s     /,ASM YYF!F3L
;NK (}A66:6<XMA s   A& A& &
A32A3T)frozenc                  d    \ rS rSr% S\S'   S\S'   S\S'   S\S	'   S\S
'   \    SS j5       rSrg)ShimGeneratori  (dict[OperatorName, NativeFunctionsGroup]r   r   r   dict[DispatchKey, BackendIndex]r   r   r   r   c                    [        UU R                  U R                  U R                  U R                  U R
                  5      nU$ rQ   )r   r   r   r   r   r   )selfr   results      rJ   __call__ShimGenerator.__call__  sC    
 ##  KK##
 rY    N)r   r   rj   
str | None)__name__
__module____qualname____firstlineno____annotations__r	   r   __static_attributes__r   rY   rJ   r   r     sC    @@44L  
 !rY   r   c                    SR                  [        [        [        UUUUU5      U 5      5      5      nUR	                  5       nSn	U(       a	  SU	 SU S3$ SU	 SU(       a  SOS SU S	[        U5       S
U SU 3$ )Nrv   z
// WARNING: THIS FILE IS AUTOGENERATED BY torchgen. DO NOT MODIFY BY HAND.
// See https://github.com/pytorch/pytorch/blob/7e86a7c0155295539996e0cf422883571126073e/torchgen/gen.py#L2424-L2436 for detailszl

#pragma once

#include <torch/csrc/inductor/aoti_torch/c/shim.h>

#ifdef __cplusplus
extern "C" {
#endif

z,

#ifdef __cplusplus
} // extern "C"
#endif
z5

#include <torch/csrc/inductor/aoti_torch/generated/zextend/r    c_shim_zf.h>
#include <torch/csrc/inductor/aoti_torch/utils.h>

#ifndef AT_PER_OPERATOR_HEADERS
#include <ATen/zFunctions.h>
#include <ATen/CompositeExplicitAutogradFunctions.h>
#include <ATen/CompositeExplicitAutogradNonFunctionalFunctions.h>
#include <ATen/CompositeImplicitAutogradFunctions.h>
#else
z/
#endif

using namespace torch::aot_inductor;

)r}   listr   r   r   ri   )
native_functionsr   r   r   r   r   includesbodyr   warnings
             rJ   gen_aoti_c_shimr     s     99& #& !		
D !FCG 	 

   	&	 
4ASIY[3\\cdjck l L!" #
 

 
 #
 	
rY   )F)r;   r   r-   ri   r<   r   rj   z1tuple[list[str], list[str], list[str], list[str]])rN   	list[str]r?   r   rj   r   )rT   zSequence[Argument]rj   tuple[list[str], list[str]])rn   r   rj   r   )rn   r   r   ri   r   ri   rj   ztuple[str, str])r   z"CppSignature | DispatcherSignaturer   r   rj   r   )r   r   r   r   rj   ri   )r   r   r   r   r   r   r   r   r   r   rj   zBackendIndex | None)r   r   r   r   r   r   r   r   r   r   rj   r   )r   r   rj   ri   )r   r   r   r   r   r   r   r   r   r   r   r   rj   r   )r    )r   zSequence[NativeFunction]r   r   r   r   r   r   r   r   r   r   r   ri   rj   ri   )=
__future__r   r~   dataclassesr   typingr   torchgen.api.typesr   torchgen.api.types.signaturesr   r   torchgen.contextr	   torchgen.modelr
   r   r   r   r   r   r   r   r   r   r   r   torchgen.utilsr   collections.abcr   r/   r   intr]   r^   floatri   DeviceIndexLayoutMemoryFormat
ScalarType	Generatorr.   r1   r0   r5   rO   rX   rs   rt   r   r   r   r   r   r   r   r   r   r   r   rY   rJ   <module>r      s   "  !   2 I 8    $ ( MM%
KK
JJ	
MM9
MM8
LL(
JJ
	
MM9

y
+   MM<
KK
JJ	
MM=
MM=
LL(
JJ"
*
MM=
,
(
o   MM2
KK
JJ
MM2
MM2
LL"
JJ
7
MM-
9
5
> ( tK	tK
tK tK 7	tKnA
?"-;b MO I N)#)#$')#7:)#)#X	+  III 	I#
#@# # 5	#
 # #L
@  5	
  " 
 @    5	 
      F $  < A
.A
@A
 A
 5	A

 A
 A
 A
 	A
rY   