+
    ˜°³jbs  ã                  ó.  a € 0 t $ ^ RIHt ^ RIt^ RIt^ RIt^ RIHt ^ RIH	u H
t
 ^ RIHt ^ RIHtHtHtHtHtHtHtHtHtHtHt ^ RIHt ]'       d   ^ RIHt ]P:                  ! 4       tR]R	&   R
 R lt ]!! R<4      t"R R lt#R R lt$]!! R=4      t%]!! R>4      t&R R lt'R R lt(]!! R?4      t)]!! R@4      t*R R lt+Rt,]PZ                  ! ],4      t.R]R&   R R lt/R R lt0R R lt1R R  lt2]Pf                  R!]Ph                  R"]Pj                  R#]Pl                  R$]Pn                  R"]Pp                  R"]Pr                  R%/t:R& R' lt;R( R) lt<R* R+ lt=R, R- lt>R. R/ lt?R0 R1 lt@R2 R3 ltA]!! RA4      tBR4 R5 ltCR6 R7 ltD ! R8 R94      tE ! R: R;4      tFR# )Bé    )ÚannotationsN)ÚTYPE_CHECKING)Únative_function_manager)ÚArgumentÚBackendIndexÚBaseTyÚBaseTypeÚFunctionSchemaÚNativeFunctionsGroupÚNativeFunctionsViewGroupÚOptionalTypeÚSelfArgumentÚTensorOptionsArgumentsÚType)Úconfig)ÚSequencezlogging.LoggerÚloggerc               ó    € V ^8„  d   QhRRRR/# )é   Ú	argumentsz:Sequence[Argument | SelfArgument | TensorOptionsArguments]ÚreturnÚbool© )Úformats   "Úb/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchgen/static_runtime/generator.pyÚ__annotate__r   !   s   € ÷ 
ñ 
ØIð
à	ñ
ó    c                óv   € V  F2  p\        VR R4      pV'       g   K  \        VRR4      pV'       g   K1   R# 	  R# )Ú
annotationNÚ	alias_setTFr   )Úgetattr)r   Úargr   r    s   &   r   Ú	has_aliasr#   !   s=   € ó ˆÜ˜S ,°Ó5ˆ
ßÙÜ˜J¨°RÓ8ˆ	ß‰9Úñ ñ r   c               ó    € V ^8„  d   QhRRRR/# )r   Úgz/NativeFunctionsGroup | NativeFunctionsViewGroupr   r   r   )r   s   "r   r   r   ö   s   € ÷ 9ñ 9ÐCð 9Èñ 9r   c                ó  € R pRp\        V \        4      '       d.   V P                  P                  pV P                  P                  pMJV P
                  P                  P                  P                  P                  pV P
                  P                  p\        P                  ! V 4      '       d   \        P                  RV4       R# V\        9   d   \        P                  RV4       R# VP                  4        F9  p\        VP                  4      pV'       d   K"  \        P                  RV4        R# 	  \        V \        4      '       dP   R\         P"                  ! VP$                  RR7      P'                  4       8w  d   \        P                  RV4       R# R	# V P(                  P                  P                  4        FM  p\        VP                  4      pV'       d   K"  \        P                  RV P(                  P                  4        R# 	  V P*                  '       g_   \-        V R
4      '       dK   \/        V4      P1                  R4      '       d+   \/        VP                  4      P1                  R4      '       g   R# R\         P"                  ! VP$                  RR7      P'                  4       8w  d   \        P                  RV4       R# \3        VP4                  P6                  4      '       d   \        P                  RV4       R# R	# )Ú NzHAND WRITTEN: %sFzBLOCKED: %sz!NOT SUPPORTED TYPE CONVERTING: %sz
at::Tensor)ÚsymintzNON-TENSOR RET TYPE: %sTÚoutzTensor(a!) out) -> Tensor(a!)z.outzat::Tensor &zNON_TENSOR RET TYPE: %szINPUTS ALIAS: %s)Ú
isinstancer   ÚviewÚ	root_nameÚfuncr)   ÚnameÚbaser   Úis_hand_writtenr   ÚinfoÚBLOCKED_OPSÚschema_order_argumentsÚivalue_type_conversion_methodÚtypeÚcppÚreturns_typeÚreturnsÚcpp_typeÚ
functionalÚ
structuredÚhasattrÚstrÚendswithr#   r   Únon_out)r%   Úbase_op_namer-   r"   Úmaybe_methods   &    r   Úis_supportedrB   ö   s  € Ø€LØ€DÜ�!Ô-×.Ò.Ø—v‘v×'Ñ'ˆØ�v‰v�{‰{‰à—u‘u—z‘z—‘×+Ñ+×0Ñ0ˆØ�u‰u�z‰zˆÜ×Ò˜a× Ò Ü�‰Ð&¨Ô5ÙØ”{Ô"Ü�‰�M <Ô0ÙØ×*Ñ*Ö,ˆÜ4°S·X±XÓ>ˆß‰|ä�K‰KÐ;¸TÔBÚñ -ô �!Ô-×.Ò.ð œ3×+Ò+¨D¯L©LÀÔG×PÑPÓRÔRä�K‰KÐ1°4Ô8ÙÙð �|‰|× Ñ ×7Ñ7Ö9ˆÜ4°S·X±XÓ>ˆß‰|ä�K‰KÐ;¸Q¿\¹\×=NÑ=NÔOÚñ :ð �<�<ˆ<ô
 ˜˜5×!Ò!Ü�t“9×%Ñ%Ð&E×FÒFÜ�t—y‘y“>×*Ñ*¨6×2Ò2áàœ×)Ò)¨$¯,©,¸uÔE×NÑNÓPÔPÜ�‰Ð-¨tÔ4ÙÜ�—‘×'Ñ'×(Ò(ä�‰Ð&¨Ô5ÙÙr   c               ó    € V ^8„  d   QhRRRR/# )r   Úarg_typeúBaseType | OptionalType | Typer   ztuple[bool, str] | Noner   )r   s   "r   r   r   2  s   € ÷ )ñ )Ø,ð)àñ)r   c                ó"  € \         P                  R\         P                  R\         P                  R\         P                  R\         P
                  R\         P                  R/pRp\        V \        4      '       d   V P                  pMQ\        V \        4      '       d:   \        V P                  \        4      '       g   R# V P                  P                  pMR# W!9  d   R# W,          p\        V \        4      '       d
   V^ ,          # V^,          # )a0  
Return the method call expression of `c10::ivalue' to convert its contained value to
the expected value of `arg_type` type. For example, for `arg_type` == BaseTy.Tensor,
this function returns ".toTensor()", so that it can be appended to the ivalue's
variable name to get the value of the expected type.
N))Tz
toTensor())FztoOptional<at::Tensor>()))FztoInt())FztoOptional<int64_t>()))FztoBool())FztoOptional<bool>()))Fz
toScalar())FztoOptional<at::Scalar>()))FztoScalarType())FztoOptional<at::ScalarType>()))FztoStringView())FztoOptional<c10::string_view>())Fz toOptional<::std::string_view>())r   ÚTensorÚintr   ÚScalarÚ
ScalarTyper=   r*   r	   r.   r   Úelem)rD   Útype_conversion_methodsÚbase_ty_objectÚmethodss   &   r   r4   r4   2  sÒ   € ô 	�‰ÐRÜ�
‰
ÐJÜ�‰ÐIÜ�‰ÐSÜ×Ñð 
ô 	�
‰
ð 
ðÐð  €NÜ�(œH×%Ò%Ø!Ÿ™‰Ü	�Hœl×	+Ò	+Ü˜(Ÿ-™-¬×2Ò2áØ!Ÿ™×+Ñ+‰áàÔ4ÙØ%Õ5€GÜ�(œH×%Ò%Ø�q�zÐØ�1�:Ðr   c               ó    € V ^8„  d   QhRRRR/# ©r   Úop_namer=   r   r   r   )r   s   "r   r   r   q  s   € ÷ 1ñ 1 3ð 1¨4ñ 1r   c                ó   € V \         9   # ©N)Úshould_use_int_tensor_ops_©rQ   s   &r   Úshould_use_int_tensorrV   q  s   € ØÔ0Ñ0Ð0r   c               ó    € V ^8„  d   QhRRRR/# rP   r   )r   s   "r   r   r   u  s   € ÷ 5ñ 5 sð 5¨tñ 5r   c                ó   € V \         9   # rS   )Úshould_use_complex_tensor_ops_rU   s   &r   Úshould_use_complex_tensorrZ   u  s   € ØÔ4Ñ4Ð4r   c               ó    € V ^8„  d   QhRRRR/# )r   rQ   r=   r   rH   r   )r   s   "r   r   r   ‹  s   € ÷ ñ ˜Sð  Sñ r   c                ó:   € V \         9   d   ^# V \        9   d   ^# ^# )é   )Útest_tensor_dim_ops_1_Útest_tensor_dim_ops_2_rU   s   &r   Útest_tensor_dimr`   ‹  s   € ØÔ(Ô(ÙØÔ(Ô(ÙÙr   z{"view_as_complex": "{2, 2}"}zdict[str, str]Útest_tensor_shape_jsonc               ó    € V ^8„  d   QhRRRR/# )r   rQ   r=   r   r   )r   s   "r   r   r   —  s   € ÷ ñ ˜sð  sñ r   c                ó6   € V \         9   d   \         V ,          # R # )r'   )ra   rU   s   &r   Útest_tensor_shaperd   —  s   € ØÔ(Ô(Ü% gÕ.Ð.ár   c               ó(   € V ^8„  d   QhRRRRRRRR/# )r   rD   rE   ÚindexrH   rQ   r=   r   r   )r   s   "r   r   r   ž  s,   € ÷ (ñ (Ø,ð(Ø58ð(ØCFð(àñ(r   c                óŠ  € \        V4      pVR 8X  du   V^ 8X  d   ^M^@p\        V4      p\        P                  ! V\	        V4      ,          4      pWf^,          ,          pRP                  RP                  V .V,          4      4      p\        V4      '       d   RV R2pM\        V4      '       d   RV R2pMRV R2p\        P                  V\        P                  R	\        P                  R
\        P                  R\        P                  R\        P                  R/pRp	\!        V \"        4      '       d   V P$                  p	Mc\!        V \&        4      '       d!   \!        V P(                  \"        4      '       g   \+        R\-        V 4       24      hV P(                  P$                  p	W˜9  d   \+        RV	 24      hW‰,          p
V
# )r'   z{{{}}}Ú,zat::randint(1, 100, z, at::kInt)z
at::randn(z, at::kComplexFloat)z	at::rand(Ú)Ú1ÚfalseÚ2zat::ScalarType::Floatz"floor"Nz.Expected OptionalType with BaseType elem, got zUnexpected type: )rd   r`   ÚmathÚceilÚfloatr   ÚjoinrV   rZ   r   rG   rH   r   rI   rJ   r=   r*   r	   r.   r   rK   ÚAssertionErrorr5   )rD   rf   rQ   Útensor_size_exÚnum_tensorsÚnum_dimÚsize_per_dimÚtensor_expressionÚvalue_expressionsrM   Úvalue_expressions   &&&        r   Útest_value_expressionry   ž  s�  € ô ' wÓ/€NØ˜ÔØ! QœJ‘b¨BˆÜ! 'Ó*ˆÜ—y’y ¬u°W«~Õ!=Ó>ˆØ qÕ(Õ(ˆØ!Ÿ™¨¯©°l°^Ð2EÈÕ2OÓ)PÓQˆÜ˜W×%Ò%Ø2°>Ð2BÀ+ÐNÑÜ	" 7×	+Ò	+Ø(¨Ð(8Ð8LÐMÑà'¨Ð'7°qÐ9Ðô 	�‰Ð(Ü�
‰
�CÜ�‰�WÜ�‰�sÜ×ÑÐ2Ü�
‰
�IðÐð €NÜ�(œH×%Ò%Ø!Ÿ™‰ô �x¤×.Ò.´:¸h¿m¹mÌX×3VÒ3Vä Ø@ÄÀhÃÐ@PÐQóð ð "Ÿ™×+Ñ+ˆØÔ.ÜÐ0°Ð0@ÐAÓBÐBØ(Õ8ÐØÐr   c               ó$   € V ^8„  d   QhRRRRRR/# ©r   Úschemar
   rf   rH   r   r=   r   )r   s   "r   r   r   É  s!   € ÷ 1ñ 1¨Nð 1À3ð 1È3ñ 1r   c                óÊ  € V P                  4       '       d   \        R V  24      hV P                  P                  P                  p/ pV P	                  4        F'  p\        VP                  W4      pWSVP                  &   K)  	  \        P                  ! W2V4       . pVP                  4        F  w  rxVP                  RV V RV 24       K   	  RP                  V4      R,           # )úExpected non-out function, got zauto z = ú;
    Ú;)Ú	is_out_fnrq   r.   r/   r3   ry   r5   r   Úoverride_test_valuesÚitemsÚappendrp   )	r|   rf   Úschema_nameÚarg_mapr"   Útest_value_expÚarg_populationsÚarg_nameÚ	arg_values	   &&       r   Úgenerate_test_value_definitionsr‹   É  sÊ   € Ø×Ñ×ÒÜÐ>¸v¸hÐGÓHÐHØ—+‘+×"Ñ"×'Ñ'€KØ€GØ×,Ñ,Ö.ˆÜ.¨s¯x©x¸ÓLˆØ*�—‘Óñ /ô ×Ò °eÔ<Ø€OØ&Ÿ}™}žÑˆØ×Ñ  x j°°°s¸9¸+ÐFÖGñ  /à�>‰>˜/Ó*¨SÕ0Ð0r   c               ó$   € V ^8„  d   QhRRRRRR/# r{   r   )r   s   "r   r   r   Ø  s&   € ÷ Uñ U nð U¸Sð UÀSñ Ur   c                óž   a€ V P                  4       '       d   \        R V  24      hRP                  V3R lV P                  4        4       4      # )r~   rh   c              3  óB   <"  € T F  qP                    S 2x € K  	  R # 5irS   ©r.   )Ú.0r"   rf   s   & €r   Ú	<genexpr>Ú,generate_test_value_names.<locals>.<genexpr>Û  s   øé € ÐTÑ4S¨S—x‘x�j  Õ(Ó4Sùs   ƒ)r�   rq   rp   r3   )r|   rf   s   &fr   Úgenerate_test_value_namesr“   Ø  sC   ø€ Ø×Ñ×ÒÜÐ>¸v¸hÐGÓHÐHØ�8‰8ÔT°F×4QÑ4QÔ4SÓTÓTÐTr   rG   rH   ro   r=   r   c               ó    € V ^8„  d   QhRRRR/# )r   r|   r
   r   zlist[tuple[str, str | None]]r   )r   s   "r   r   r   é  s    € ÷ Iñ IØðIà!ñIr   c                ób   € R  R lpV P                  4        Uu. uF
  q!! V4      NK  	  up# u upi )c               ó    € V ^8„  d   QhRRRR/# )r   r"   r   r   ztuple[str, str | None]r   )r   s   "r   r   Ú0generate_test_ir_arguments.<locals>.__annotate__ì  s   € ÷ *ñ *˜ð *Ð&<ñ *r   c                óp  € V P                   pR p\        V\        4      '       d   VP                  pRp\        V\        4      '       g   \        R\        V4       24      hRpVP                  \        9   d   \        VP                  ,          pV'       d   V'       d   V R2pRV P                  ,           V3# )FTzExpected BaseType, got NÚ?Ú%)r5   r*   r   rK   r	   rq   r.   Ú/generate_test_ir_arguments_base_ty_to_type_str_)r"   ÚtÚadd_optionalÚtype_strs   &   r   Úir_argumentÚ/generate_test_ir_arguments.<locals>.ir_argumentì  s’   € Ø�H‰HˆØˆÜ�aœ×&Ò&Ø—‘ˆAØˆLÜ˜!œX×&Ò&Ü Ð#:¼4À»7¸)Ð!DÓEÐEØˆØ�6‰6ÔDÔDÜFÀqÇvÁvÕNˆHßŸØ"˜ 1�~ˆHØ�c—h‘h• Ð)Ð)r   )r3   )r|   rŸ   r"   s   &  r   Úgenerate_test_ir_argumentsr¡   é  s1   € õ*ð )/×(EÑ(EÔ(GÓHÑ(G ˆK˜ÖÑ(GÑHÐHùÒHs   ™,c               ó    € V ^8„  d   QhRRRR/# )r   r|   r
   r   r=   r   )r   s   "r   r   r   þ  s   € ÷ 1ñ 1 Nð 1°sñ 1r   c                ól  € . p\        V P                  4       4       F~  w  r#\        VP                  4      pV'       g&   \	        R VP
                   RVP                   24      hVw  rVV'       d   RMRpVP                  RV RVP
                   RV RV 24       K€  	  RP                  V4      R	,           # )
zNo type conversion method for ú: Ú&r'   z
const autoÚ z = p_node->Input(z).r   r€   )Ú	enumerater3   r4   r5   rq   r.   r„   rp   )r|   rˆ   Úir"   rA   Úis_referenceÚtype_conversion_methodÚ	references   &       r   Úgenerate_arg_extractionr¬   þ  s±   € Ø€OÜ˜F×9Ñ9Ó;Ö<‰ˆÜ4°S·X±XÓ>ˆßÜ Ø0°·±°
¸"¸S¿X¹X¸JÐGóð ð 0<Ñ,ˆß'‘C¨Rˆ	Ø×ÑØ˜˜ 1 S§X¡X JÐ.?À¸sÀ"ÐE[ÐD\Ð]ö	
ñ =ð �>‰>˜/Ó*¨SÕ0Ð0r   c               ó$   € V ^8„  d   QhRRRRRR/# ©r   r%   r   Úbackend_indexr   r   r=   r   )r   s   "r   r   r     s"   € ÷ ñ Ð+ð ¸Lð ÈSñ r   c                óÐ   € VP                  V P                  4      pV P                  '       g   Vf+   \        P                  ! V P                  P
                  4      # VP                  # rS   )Ú
get_kernelr:   r;   r6   r.   r-   Úkernel©r%   r¯   r²   s   && r   Úget_kernel_namer´     sF   € Ø×%Ñ% a§l¡lÓ3€FØ‡|‡|€|�v’~Ü�xŠx˜Ÿ™×)Ñ)Ó*Ð*Ø�=‰=Ðr   c               ó$   € V ^8„  d   QhRRRRRR/# r®   r   )r   s   "r   r   r     s#   € ÷ ñ Ð/ð Àð ÐQTñ r   c                óÐ   € VP                  V P                  4      pV P                  '       g   Vf+   \        P                  ! V P                  P
                  4      # VP                  # rS   )r±   r)   r;   r6   r.   r-   r²   r³   s   && r   Úget_out_kernel_namer·     sD   € Ø×%Ñ% a§e¡eÓ,€FØ‡|‡|€|�v’~Ü�xŠx˜Ÿ™Ÿ
™
Ó#Ð#Ø�=‰=Ðr   c               ó$   € V ^8„  d   QhRRRRRR/# r®   r   )r   s   "r   r   r     s)   € ÷ 	Iñ 	IØð	IØ,8ð	Iàñ	Ir   c                ó  € V P                   P                  pVP                  4       '       d   \        R V 24      h\	        W4      pR VP                  4        4       pV P                  '       d   RMRpRV RV RRP                  V4       R2# )	r~   c              3  ó8   "  € T F  qP                   x € K  	  R # 5irS   r�   ©r�   r"   s   & r   r‘   Ú0generate_non_out_variant_call.<locals>.<genexpr>#  ó   é € ÐEÑ%D˜c—–Ó%Dùó   ‚ÚcpuÚnativeúat::ú::Ú(rh   ri   )r:   r-   r�   rq   r´   r3   r;   rp   )r%   r¯   r|   Úkernel_nameÚ	arg_namesÚnamespace_names   &&    r   Úgenerate_non_out_variant_callrÇ     s‡   € ð �\‰\×Ñ€FØ×Ñ×ÒÜÐ>¸v¸hÐGÓHÐHÜ! !Ó3€KÙE V×%BÑ%BÔ%DÓE€IØŸlŸl˜l‘U°€NØ�.Ð!  K =°°#·(±(¸9Ó2EÐ1FÀaÐHÐHr   c               ó$   € V ^8„  d   QhRRRRRR/# ©r   r%   r   r¯   r   r   r=   r   )r   s   "r   r   r   (  s)   € ÷ 
Iñ 
IØð
IØ0<ð
Iàñ
Ir   c                ó   € V P                   P                  p\        P                  ! V4      pVP	                  V P                   4      pV'       d   VP
                  pR  VP                  4        4       pRpRV RV RRP                  V4       R2# )c              3  ó8   "  € T F  qP                   x € K  	  R # 5irS   r�   r»   s   & r   r‘   Ú,generate_call_to_view_ops.<locals>.<genexpr>0  r½   r¾   rÀ   rÁ   rÂ   rÃ   rh   ri   )r+   r-   r6   r.   r±   r²   r3   rp   )r%   r¯   r|   rÄ   r²   rÅ   rÆ   s   &&     r   Úgenerate_call_to_view_opsrÍ   (  s}   € ð �V‰V�[‰[€FÜ—(’(˜6Ó"€KØ×%Ñ% a§f¡fÓ-€FßØ—m‘mˆÙE V×%BÑ%BÔ%DÓE€IØ€NØ�.Ð!  K =°°#·(±(¸9Ó2EÐ1FÀaÐHÐHr   c               ó$   € V ^8„  d   QhRRRRRR/# r®   r   )r   s   "r   r   r   5  s)   € ÷ Cñ CØðCØ,8ðCàñCr   c                óø  € V P                   P                  pVP                  4       '       g   \        R V 24      h. p\	        W4      pV P
                  '       d0   VP                  P                    Uu. uF  qUP                  NK  	  ppM. pVP                  P                   Fˆ  p\        V\        4      '       d(   VP                  VP                  P                  4       K@  \        V\        4      '       g   \        R\        V4       24      hVP                  VP                  4       KŠ  	  V P
                  '       g†   \        VP                  P                   4      ^8w  d,   \        R\        VP                  P                   4       24      hVP                  VP                  P                   ^ ,          P                  4       RP!                  V4      pV P
                  '       d   RMRpRV RV RV R	2# u upi )
zExpected out function, got zExpected Argument, got úExpected 1 out argument, got rh   r¿   rÀ   rÁ   rÂ   rÃ   ri   )r)   r-   r�   rq   r·   r;   r   r.   r?   r*   r   r„   Úargumentr   r5   Úlenrp   )	r%   r¯   r|   rÅ   rÄ   Úout_argr"   Úcpp_arg_namesrÆ   s	   &&       r   Úgenerate_out_variant_callrÕ   5  sš  € ð �U‰U�Z‰Z€FØ×Ñ×ÒÜÐ:¸6¸(ÐCÓDÐDØ€IÜ% aÓ7€KØ‡|‡|€|à17×1AÑ1A×1EÒ1EÓFÑ1E g—\”\Ñ1Eˆ	ÐFˆ	àˆ	Ø×Ñ×'Ô'ˆÜ�cœ<×(Ò(Ø×Ñ˜SŸ\™\×.Ñ.Ö/ä˜c¤8×,Ò,Ü$Ð'>¼tÀC»y¸kÐ%JÓKÐKØ×Ñ˜SŸX™XÖ&ñ (ð �<�<ˆ<Üˆv×Ñ×#Ñ#Ó$¨Ô)Ü Ø/´°F×4DÑ4D×4HÑ4HÓ0IÐ/JÐKóð ð 	×Ñ˜×)Ñ)×-Ñ-¨aÕ0×5Ñ5Ô6Ø—H‘H˜YÓ'€MØŸlŸl˜l‘U°€NØ�.Ð!  K =°°-°ÀÐBÐBùò% Gs   Á2G7c               ó    € V ^8„  d   QhRRRR/# )r   r|   r
   r   r   r   )r   s   "r   r   r   f  s   € ÷ <ñ < ð <°4ñ <r   c                óR   € \        V 4      pVR VP                  R4       pV\        9  # )NrÃ   )r=   ÚfindÚno_memory_resize_ops)r|   Ú
schema_strÚtype_variant_op_names   &  r   Úshould_check_resizerÜ   f  s.   € Ü�V“€JØ%Ð&<¨
¯©¸Ó(<Ð=ÐØÔ';Ñ;Ð;r   c               ó    € V ^8„  d   QhRRRR/# ©r   r%   r   r   r=   r   )r   s   "r   r   r   l  s   € ÷ ,ñ ,Ð.ð ,°3ñ ,r   c                ój   € V P                   P                  P                  P                  P                  # rS   )r:   r-   r.   r/   )r%   s   &r   Úop_name_from_grouprà   l  s$   € Ø�<‰<×Ñ×!Ñ!×&Ñ&×+Ñ+Ð+r   c                  óF   € ] tR tRtR R ltR R ltR R ltR R	 ltR
tR# )ÚGenOpDispatcherip  c               ó$   € V ^8„  d   QhRRRRRR/# )r   ÚgroupsúSequence[NativeFunctionsGroup]r¯   r   r   r=   r   )r   s   "r   r   ÚGenOpDispatcher.__annotate__q  s$   € ÷ ñ Ø4ðØEQðà	ñr   c           	     	óÊ  € V'       g   R # . pV F�  p\        V4      ;_uu_ 4        \        V4      '       g   \        RV 24      h\        V\        4      '       g   \        R\        V4       24      hV P                  WB4      pVP                  V4       RRR4       K�  	  \        V^ ,          4      pRP                  V4      pRV RV RV R2pV#   + '       g   i     KÓ  ; i)	r'   úUnsupported function group: ú#Expected NativeFunctionsGroup, got NÚ
z&
REGISTER_OPERATOR_FUNCTOR(
    aten::ú,
    aten_ú(,
    [](Node* n) -> SROperator {
      z9
      LogAndDumpSchema(n);
      return nullptr;
    })
)
r   rB   rq   r*   r   r5   Úout_variant_op_generatorr„   rà   rp   ©	Úselfrä   r¯   Úgenerated_type_variantsr%   Úgenerated_type_variantrQ   ÚbodyÚ	generateds	   &&&      r   Úout_variantÚGenOpDispatcher.out_variantq  sñ   € ÷ ÙØ"$ÐÛˆAÜ(¨×+Õ+Ü# A—’Ü(Ð+GÈÀsÐ)KÓLÐLÜ! !Ô%9×:Ò:Ü(Ø=¼dÀ1»g¸YÐGóð ð *.×)FÑ)FÀqÓ)XÐ&Ø'×.Ñ.Ð/EÔF÷ ,Ñ+ñ ô % V¨A¥YÓ/ˆØ�y‰yÐ0Ó1ˆðàˆ)ð 
Ø
ˆð à€fð ð	ˆ	ð Ð÷+ ,×+Ð+ús   ¥A/CÃC"c               ó$   € V ^8„  d   QhRRRRRR/# )r   rä   ú"Sequence[NativeFunctionsViewGroup]r¯   r   r   r=   r   )r   s   "r   r   ræ   �  s$   € ÷ ñ Ø8ðØIUðà	ñr   c           	     	óà  € V'       g   R # . pV F�  p\        V4      ;_uu_ 4        \        V4      '       g   \        RV 24      h\        V\        4      '       g   \        R\        V4       24      hV P                  WB4      pVP                  V4       RRR4       K�  	  \        P                  ! V^ ,          4      pRP                  V4      pRV RV RV R2pV#   + '       g   i     KÞ  ; i)	r'   úUnsupported view group: ú'Expected NativeFunctionsViewGroup, got Nrê   z-
REGISTER_NATIVE_OPERATOR_FUNCTOR(
    aten::rë   rì   z:
      LogAndDumpSchema(n);
      return nullptr;
    });
)r   rB   rq   r*   r   r5   Úview_op_generatorr„   r   Úfunc_name_base_strrp   rî   s	   &&&      r   r+   ÚGenOpDispatcher.view�  s÷   € ÷ ÙØ"$ÐÛˆAÜ(¨×+Õ+Ü# A—’Ü(Ð+CÀAÀ3Ð)GÓHÐHÜ! !Ô%=×>Ò>Ü(ØAÄ$ÀqÃ'ÀÐKóð ð *.×)?Ñ)?ÀÓ)QÐ&Ø'×.Ñ.Ð/EÔF÷ ,Ñ+ñ ô ×+Ò+¨F°1­IÓ6ˆØ�y‰yÐ0Ó1ˆðàˆ)ð 
Ø
ˆð à€fð ð	ˆ	ð Ð÷+ ,×+Ð+ús   ¥A/CÃC-c               ó$   € V ^8„  d   QhRRRRRR/# r®   r   )r   s   "r   r   ræ   ­  s$   € ÷ ñ Ø%ðØ6Bðà	ñr   c                	óR  € VP                   p\        VP                  4      p\        VP                   P                  4      p\	        W4      p\        VP                  P                  P                  P                  4      ^8w  d@   \        R\        VP                  P                  P                  P                  4       24      h\        VP                  P                  P                  P                  ^ ,          P                  4      p\        W4      pRV RV RV RV RV RV R2p	V	# )	r]   rÐ   ú+
      if (n->matches(torch::schema("aten::ú<"))) {
        return [](ProcessedNode* p_node) {
          zM
          if (p_node->Output(0).isNone()) {
            p_node->Output(0) = z2;
            return;
          }
          auto& z< = p_node->Output(0).toTensor();
          fastResizeToZero(z);
          ú;
        };
      })r:   r=   r-   r¬   rÇ   rÒ   r)   r   rq   r.   rÕ   )
rï   r%   r¯   r:   r|   Úpopulated_argumentÚfunctional_variant_callÚout_variable_nameÚout_variant_callró   s
   &&&       r   rí   Ú(GenOpDispatcher.out_variant_op_generator­  s  € ð —\‘\ˆ
Ü�Z—_‘_Ó%ˆÜ4°Q·\±\×5FÑ5FÓGÐÜ"?ÀÓ"QÐÜˆq�u‰u�z‰z×#Ñ#×'Ñ'Ó(¨AÔ-Ü Ø/´°A·E±E·J±J×4HÑ4H×4LÑ4LÓ0MÐ/NÐOóð ô   §¡§
¡
× 4Ñ 4× 8Ñ 8¸Õ ;× @Ñ @ÓAÐÜ4°QÓFÐð+Ø+1¨(ð 3àÐ
ð !à!8Ð 9ð :ð #Ð#ð $Ø-Ð.ð /ØÐ
ð 	ðˆ	ð Ðr   c               ó$   € V ^8„  d   QhRRRRRR/# rÉ   r   )r   s   "r   r   ræ   É  s$   € ÷ ñ Ø)ðØ:Fðà	ñr   c                	ó°   € \        VP                  P                  4      p\        VP                  P                  4      p\	        W4      pR V RV RV R2pV# )r   r  z!
            p_node->Output(0) = r  )r=   r+   r-   r¬   rÍ   )rï   r%   r¯   r|   r  r  ró   s   &&&    r   rû   Ú!GenOpDispatcher.view_op_generatorÉ  sh   € ô �Q—V‘V—[‘[Ó!ˆÜ4°Q·V±V·[±[ÓAÐÜ";¸AÓ"MÐð+Ø+1¨(ð 3àÐ
ð !Ø!8Ð 9ð :	ð	ˆ	ð Ðr   r   N)	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__rô   r+   rí   rû   Ú__static_attributes__r   r   r   râ   râ   p  s   † õõ<õ<÷8ñ r   râ   c                  óF   € ] tR tRtR R ltR R ltR R ltR R	 ltR
tR# )ÚGenOpTestCaseiÙ  c               ó    € V ^8„  d   QhRRRR/# )r   rä   rå   r   r=   r   )r   s   "r   r   ÚGenOpTestCase.__annotate__Ú  s   € ÷ 2ñ 2Ð"@ð 2ÀSñ 2r   c           	     	óŠ  € V'       g   R # . pV F�  p\        V4      ;_uu_ 4        \        V4      '       g   \        RV 24      h\        V\        4      '       g   \        R\        V4       24      hV P                  V4      pVP                  V4       RRR4       K�  	  RP                  V4      #   + '       g   i     K³  ; i)r'   rè   ré   Nrê   )	r   rB   rq   r*   r   r5   Ú"out_variant_op_test_case_generatorr„   rp   ©rï   rä   rð   r%   rñ   s   &&   r   rô   ÚGenOpTestCase.out_variantÚ  s±   € ßÙØ"$ÐÛˆAÜ(¨×+Õ+Ü# A—’Ü(Ð+GÈÀsÐ)KÓLÐLÜ! !Ô%9×:Ò:Ü(Ø=¼dÀ1»g¸YÐGóð ð *.×)PÑ)PÐQRÓ)SÐ&Ø'×.Ñ.Ð/EÔF÷ ,Ñ+ñ ð �y‰yÐ0Ó1Ð1÷ ,×+Ð+úó   ¥A/B1Â1Cc               ó    € V ^8„  d   QhRRRR/# )r   rä   r÷   r   r=   r   )r   s   "r   r   r  ê  s   € ÷ 2ñ 2Ð=ð 2À#ñ 2r   c           	     	óŠ  € V'       g   R # . pV F�  p\        V4      ;_uu_ 4        \        V4      '       g   \        RV 24      h\        V\        4      '       g   \        R\        V4       24      hV P                  V4      pVP                  V4       RRR4       K�  	  RP                  V4      #   + '       g   i     K³  ; i)r'   rù   rú   Nrê   )	r   rB   rq   r*   r   r5   Úview_op_test_case_generatorr„   rp   r  s   &&   r   r+   ÚGenOpTestCase.viewê  s°   € ßÙØ"$ÐÛˆAÜ(¨×+Õ+Ü# A—’Ü(Ð+CÀAÀ3Ð)GÓHÐHÜ! !Ô%=×>Ò>Ü(ØAÄ$ÀqÃ'ÀÐKóð ð *.×)IÑ)IÈ!Ó)LÐ&Ø'×.Ñ.Ð/EÔF÷ ,Ñ+ñ ð �y‰yÐ0Ó1Ð1÷ ,×+Ð+úr  c               ó    € V ^8„  d   QhRRRR/# rÞ   r   )r   s   "r   r   r  ú  s   € ÷ 3ñ 3Ð4Hð 3ÈSñ 3r   c                	ó¸  € VP                   P                  p\        V4      pVP                  R 4      ^ 8:  d   \	        RV 24      hVRVP                  R 4       P                  RR4      p\        V4      pVP                  V4      '       g   \	        RV RV 24      h\        V4      pRP                  R V 4       4      pRP                  R	 V 4       4      p\        VP                  4      ^8X  dk   \        VP                  ^ ,          P                  \        4      '       d:   VP                  ^ ,          P                  P                  \         P"                  J g   \	        R
VP                   24      h\%        V^ 4      p	\'        V^ 4      p
\%        V^4      p\'        V^4      p\)        V4      '       d   RMRpRV RV RV R V RV	 RV
 RV RV RV RV R2pV# )rÃ   úInvalid schema string: NÚ.Ú_úType variant op name ú doesn't start with ú, c              3  ó@   "  € T F  w  rVf   TMV RV 2x € K  	  R # 5i©Nr¤   r   ©r�   r‰   rD   s   &  r   r‘   ÚCGenOpTestCase.out_variant_op_test_case_generator.<locals>.<genexpr>  ó/   é € ð á*3Ñ&�Hð %Ò,‘°X°J¸bÀÀ
Ð2KÔKÛ*3ùó   ‚c              3  ó*   "  € T F	  w  rVx € K  	  R # 5irS   r   ©r�   r‰   r!  s   &  r   r‘   r(    ó   é € ÐE¹9©K¨H�x»9ùó   ‚ú#Expected single Tensor return, got Útruerk   ú
TEST(StaticRuntime, autogen_ú1) {
  const std::string script = R"IR(
    graph(ú?):
        %bias: None = prim::Constant()
        %ret = aten::úQ)
        %cloned = aten::clone(%ret, %bias)
        return (%cloned)
  )IR";

  ú
  std::vector<IValue> args{zj};
  testStaticRuntime(script, args, {}, /*use_allclose=*/false, /*use_equalnan=*/false, /*check_resize=*/z);

  z
  std::vector<IValue> args2{zm};
  testStaticRuntime(script, args, args2, /*use_allclose=*/false, /*use_equalnan=*/false, /*check_resize=*/z);

}
)r:   r-   r=   rØ   rq   Úreplacerà   Ú
startswithr¡   rp   rÒ   r8   r*   r5   r	   r.   r   rG   r‹   r“   rÜ   )rï   r%   r|   rÚ   rÛ   rQ   Ú	arg_typesÚarg_declarationsrÅ   Útest_value_definitionsÚtest_value_namesÚtest_value_definitions2Útest_value_names2Úcheck_resizeró   s   &&             r   r  Ú0GenOpTestCase.out_variant_op_test_case_generatorú  s
  € Ø—‘×"Ñ"ˆÜ˜“[ˆ
Ø�?‰?˜3Ó 1Ô$Ü Ð#:¸:¸,Ð!GÓHÐHØ)Ð*@¨J¯O©O¸CÓ,@ÐA×IÑIÈ#ÈsÓSÐÜ$ QÓ'ˆØ#×.Ñ.¨w×7Ò7Ü Ø'Ð(<Ð'=Ð=QÐRYÐQZÐ[óð ô /¨vÓ6ˆ	ØŸ9™9ñá*3óó
Ðð —I‘IÑE¹9ÓEÓFˆ	ä�—‘Ó 1Ô$Ü˜6Ÿ>™>¨!Õ,×1Ñ1´8×<Ò<Ø—‘˜qÕ!×&Ñ&×+Ñ+¬v¯}©}Ó<ä Ð#FÀvÇ~Á~ÐFVÐ!WÓXÐXÜ!@ÀÈÓ!KÐÜ4°V¸QÓ?ÐÜ"AÀ&È!Ó"LÐÜ5°f¸aÓ@ÐÜ!4°V×!<Ò!<‘vÀ'ˆðØ1Ð2ð 3àÐ
ð à�Y˜a 	˜{ð +ð
 Ðð Ø-Ð.ð /jØjvÐiwð xàÐð Ø/Ð0ð 1kØkwÐjxð yð!ˆ	ð( Ðr   c               ó    € V ^8„  d   QhRRRR/# )r   r%   r   r   r=   r   )r   s   "r   r   r  /  s   € ÷ ,ñ ,Ð-Eð ,È#ñ ,r   c                	ó\  € VP                   P                  p\        V4      pVP                  R 4      ^ 8:  d   \	        RV 24      hVRVP                  R 4       P                  RR4      pVP                   P                  pVP                  V4      '       g   \	        RV RV 24      h\        V4      pRP                  R V 4       4      pRP                  R	 V 4       4      p\        VP                  4      ^8X  dk   \        VP                  ^ ,          P                  \        4      '       d:   VP                  ^ ,          P                  P                  \         P"                  J g   \	        R
VP                   24      h\%        V^ 4      p	\'        V^ 4      p
RV RV RV R V RV	 RV
 R2pV# )rÃ   r  Nr   r!  r"  r#  r$  c              3  ó@   "  € T F  w  rVf   TMV RV 2x € K  	  R # 5ir&  r   r'  s   &  r   r‘   Ú<GenOpTestCase.view_op_test_case_generator.<locals>.<genexpr>=  r)  r*  c              3  ó*   "  € T F	  w  rVx € K  	  R # 5irS   r   r,  s   &  r   r‘   rC  B  r-  r.  r/  r1  r2  r3  r4  r5  z(};
  testStaticRuntime(script, args);
}
)r+   r-   r=   rØ   rq   r6  r,   r7  r¡   rp   rÒ   r8   r*   r5   r	   r.   r   rG   r‹   r“   )rï   r%   r|   rÚ   rÛ   rQ   r8  r9  rÅ   r:  r;  ró   s   &&          r   r  Ú)GenOpTestCase.view_op_test_case_generator/  s²  € Ø—‘—‘ˆÜ˜“[ˆ
Ø�?‰?˜3Ó 1Ô$Ü Ð#:¸:¸,Ð!GÓHÐHØ)Ð*@¨J¯O©O¸CÓ,@ÐA×IÑIÈ#ÈsÓSÐØ—&‘&×"Ñ"ˆØ#×.Ñ.¨w×7Ò7Ü Ø'Ð(<Ð'=Ð=QÐRYÐQZÐ[óð ô /¨vÓ6ˆ	ØŸ9™9ñá*3óó
Ðð —I‘IÑE¹9ÓEÓFˆ	ä�—‘Ó 1Ô$Ü˜6Ÿ>™>¨!Õ,×1Ñ1´8×<Ò<Ø—‘˜qÕ!×&Ñ&×+Ñ+¬v¯}©}Ó<ä Ð#FÀvÇ~Á~ÐFVÐ!WÓXÐXÜ!@ÀÈÓ!KÐÜ4°V¸QÓ?ÐðØ1Ð2ð 3àÐ
ð à�Y˜a 	˜{ð +ð
 Ðð Ø-Ð.ð /ðˆ	ð  Ðr   r   N)	r  r  r  r  rô   r+   r  r  r  r   r   r   r  r  Ù  s   † õ2õ 2õ 3÷j,ñ ,r   r  )¹Úsparse_sampled_addmmÚhspmmÚlinalg_svdvalsÚsspaddmmÚcoalesceÚ_indicesÚindicesÚ_valuesÚvaluesÚcrow_indicesÚcol_indicesÚfloor_divideÚgerÚconj_physicalÚbinary_cross_entropyÚarccoshÚcholeskyÚlu_solveÚlinalg_choleskyÚlinalg_householder_productÚlinalg_ldl_solveÚ_compute_linear_combinationÚ
_make_dualÚ
_fw_primalÚ_index_reduceÚ!_new_zeros_with_same_feature_metaÚ_conj_physicalÚ binary_cross_entropy_with_logitsÚbincountÚconv_tbcÚcopyÚ
_copy_fromÚ_copy_from_and_resizeÚcount_nonzeroÚcudnn_affine_grid_generatorÚ$cudnn_affine_grid_generator_backwardÚcudnn_grid_samplerÚ
diag_embedÚ	embeddingÚembedding_dense_backwardÚ_embedding_bag_dense_backwardÚ*_embedding_bag_per_sample_weights_backwardÚgrid_sampler_2dÚ_grid_sampler_2d_cpu_fallbackÚgrid_sampler_3dÚisnanÚmkldnn_linearÚmedianÚ	nanmedianÚ_sparse_sparse_matmulÚbatch_norm_backward_elemtÚ_euclidean_distÚpixel_shuffleÚpixel_unshuffleÚchannel_shuffleÚ_reshape_nested_backwardÚreluÚpreluÚceluÚslice_scatterÚselect_scatterÚdiagonal_scatterÚsumÚ_mkldnn_transposeÚ_nested_tensor_from_maskÚ_nested_from_paddedÚ_nested_tensor_sizeÚ&_nested_from_padded_and_nested_exampleÚ_standard_gamma_gradÚ_dirichlet_gradÚnative_normÚ_sparse_softmaxÚ_sparse_softmax_backward_dataÚ_sparse_log_softmaxÚ!_sparse_log_softmax_backward_dataÚzeroÚ_sparse_addmmÚsparse_maskÚ_sparse_mask_projectionÚ	_to_denseÚ	_coalesceÚ
_coalescedÚcopy_sparse_to_sparseÚ	to_sparseÚto_sparse_csrÚto_sparse_cscÚ	to_mkldnnÚquantize_per_tensor_dynamicÚquantize_per_channelÚq_per_channel_scalesÚq_per_channel_zero_pointsÚint_reprÚ"_make_per_channel_quantized_tensorÚsetÚliftÚ
lift_freshÚlift_fresh_copyÚmasked_scatterÚ_masked_softmaxÚ_masked_softmax_backwardÚputÚindex_reduceÚtraceÚ_cholesky_solve_helperÚdistÚmaxÚ_torch_cuda_cu_linker_symbol_opÚglu_jvpÚglu_backward_jvpÚhardswish_backwardÚrrelu_with_noise_backwardÚ#mkldnn_adaptive_avg_pool2d_backwardÚ_adaptive_avg_pool2d_backwardÚ_adaptive_avg_pool3d_backwardÚisinfÚlinalg_lu_solveÚlinalg_vecdotÚlinalg_matrix_expÚlinalg_eigvalshÚ_test_warn_in_autogradÚ%_test_autograd_multiple_dispatch_viewÚ*_test_autograd_multiple_dispatch_view_copyÚ_segment_reduceÚ_segment_reduce_backwardÚ_fw_primal_copyÚ_make_dual_copyÚview_as_real_copyÚview_as_complex_copyÚ
_conj_copyÚ_neg_view_copyÚdiagonal_copyÚdetach_copyÚsqueeze_copyÚt_copyÚunsqueeze_copyÚ_indices_copyÚ_values_copyÚindices_copyÚvalues_copyÚcrow_indices_copyÚcol_indices_copyÚccol_indicesÚccol_indices_copyÚrow_indicesÚrow_indices_copyÚunfold_copyÚ
alias_copyÚ_triton_multi_head_attentionÚspecial_airy_aiÚspecial_bessel_j0Úspecial_bessel_j1Úspecial_bessel_y0Úspecial_bessel_y1Úspecial_chebyshev_polynomial_tÚspecial_chebyshev_polynomial_uÚspecial_chebyshev_polynomial_vÚspecial_chebyshev_polynomial_wÚspecial_hermite_polynomial_hÚspecial_hermite_polynomial_heÚspecial_laguerre_polynomial_lÚspecial_legendre_polynomial_pÚspecial_modified_bessel_i0Úspecial_modified_bessel_i1Úspecial_modified_bessel_k0Úspecial_modified_bessel_k1Ú!special_scaled_modified_bessel_k0Ú!special_scaled_modified_bessel_k1Ú&special_shifted_chebyshev_polynomial_tÚ&special_shifted_chebyshev_polynomial_uÚ&special_shifted_chebyshev_polynomial_vÚ&special_shifted_chebyshev_polynomial_wÚspecial_spherical_bessel_j0Ú_foobarÚ_nested_tensor_stridesÚ_nested_tensor_storage_offsetsÚ_nested_get_valuesÚ_nested_get_values_copyÚ_nested_view_from_jaggedÚ_nested_view_from_jagged_copyÚ_nested_view_from_bufferÚ_nested_view_from_buffer_copyÚ_int_mmÚ_to_sparse_cscÚ_to_sparse_csrÚsegment_reduce)Úbitwise_notÚbitwise_andÚ
bitwise_orÚbitwise_xorÚbitwise_left_shiftÚbitwise_right_shiftÚgcdÚlcmÚscatterÚgatherÚ _convert_indices_from_coo_to_csrÚ _convert_indices_from_csr_to_coo)Úview_as_realÚimagÚ_conj)	ÚaddmvÚ	index_addr	  r
  Únll_loss_backwardÚdotÚvdotÚouterrR  )ÚaddmmÚmmÚnuclear_normÚdiagÚ_addmm_activationÚmatrix_Hrœ   )zisin.Scalar_Tensorr  r  r  r  ÚhistcÚl1_lossÚmulti_margin_lossÚmultilabel_margin_lossÚnll_lossÚ
nll_loss2dÚprod)GÚ__conditional_annotations__Ú
__future__r   ÚjsonÚloggingrm   Útypingr   Útorchgen.api.cppÚapir6   Útorchgen.contextr   Útorchgen.modelr   r   r   r	   r
   r   r   r   r   r   r   Útorchgen.static_runtimer   Úcollections.abcr   Ú	getLoggerr   Ú__annotations__r#   Ú	frozensetr2   rB   r4   rT   rY   rV   rZ   r^   r_   r`   Útest_tensor_shapes_stringÚloadsra   rd   ry   r‹   r“   rG   rH   ro   r=   rI   rJ   r   r›   r¡   r¬   r´   r·   rÇ   rÍ   rÕ   rÙ   rÜ   rà   râ   r  )r!  s   @r   Ú<module>r1     s§  øðß "Ð "ã Û Û Ý  ç Ð Ý 4÷÷ ÷ ñ õ +÷ Ý(ð !×*Ò*Ó,€ˆÓ ,õ
ñ ðCóE€õP9õx)ñX 'ðóÐ ñ  "+Ð+LÓ!MÐ õ1õ5ñ #ð
óÐ ñ #ØQóÐ õ
ð <Ð Ø)-¯ªÐ4MÓ)NÐ ˜Ó Nõõ(õV1õUð ‡M�M�8Ø
‡J�J�Ø
‡L�L�'Ø
‡J�J�Ø
‡M�M�5Ø
×Ñ�uØ
‡K�K�ð3Ð /õIõ*1õ õõ	Iõ
IõCñ> !ðóÐ õ$<õ,÷fñ f÷RBó Br   