+
    –°³jO  ã                   ó~  a € R' t0 t ^ RIt^ RIt^ RIt^ RIHtHt ^ RIHt ^ RI	H
t
 ^ RIHt ^ RItRR/ R/ R	/ R
/ R/ R/ R/ /RR/ R/ R/ R/ R/ R/ R/ R/ //t] ^ k  ! R R]4      t]]
R3,          tR tR R lt]! RR7       ! R R 4      4       t/ t] ^k ]! 4       t] ^k / t] ^k / t] ^k R! R" ltR# tR$ tR% tR& tR# )(é    N)Ú	dataclassÚfield)ÚEnum)ÚAny)Ú
ModuleTypeÚtorchÚcondzdynamic-shapezescape-hatchÚmapzdynamic-valueÚoperatorÚmutationÚpythonÚassertÚbuiltinÚclosurezcontext-managerzcontrol-flowzdata-structurezstandard-libraryzobject-modelc                   ó"   € ] tR t^#tRt^t^ tRtR# )ÚSupportLevelzO
Indicates at what stage the feature
used in the example is handled in export.
© N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ú	SUPPORTEDÚNOT_SUPPORTED_YETÚ__static_attributes__r   ó    ÚV/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_export/db/case.pyr   r   #   s   † ñð
 €IØÔr   r   .c                 ó&  € \        V \        4      '       g   \        R \        V 4       24      h\        V\        4      '       g   \        R\        V4       24      hV F0  p\        V\
        4      '       d   K  \        R\        V4       24      h	  R# )z(Expecting args type to be a tuple, got: z)Expecting kwargs type to be a dict, got: z+Expecting kwargs keys to be a string, got: N)Ú
isinstanceÚtupleÚ
ValueErrorÚtypeÚdictÚstr)ÚargsÚkwargsÚkeys   && r   Úcheck_inputs_typer(   0   sˆ   € Ü�dœE×"Ò"ÜØ6´t¸D³z°lÐCó
ð 	
ô �fœd×#Ò#ÜØ7¼¸V»°~ÐFó
ð 	
ó ˆÜ˜#œs×#Ô#ÜØ=¼dÀ3»i¸[ÐIóð ó r   c                ó$   € V ^8„  d   QhR\         /# )é   Útag)r$   )Úformats   "r   Ú__annotate__r-   ?   s   € ÷ 	Lñ 	L”sñ 	Lr   c                 óø   € V P                  R 4      p\        pV F\  p\        V4      \        \        P                  R,           4      8:  g   \        RV 24      hW29   d   W#,          pKO  \        RV  R24      h	  R# )Ú.Ú-z!Tag contains invalid characters: zTag z! is not found in registered tags.N)ÚsplitÚ_TAGSÚsetÚstringÚascii_lowercaseÚAssertionErrorr!   )r+   ÚpartsÚtÚparts   &   r   Ú_validate_tagr:   ?   sn   € Ø�I‰I�c‹N€EÜ€AÛˆÜ�4‹yœC¤× 6Ñ 6¸Õ <Ó=Ô=Ü Ð#DÀTÀFÐ!KÓLÐLØŒ9Ø•ŠAä˜t C 5Ð(IÐJÓKÐKó r   T)Úfrozenc                   óv   a € ] tR t^Kt o ]! ]R7      tRt]! ]R7      t	]
P                  tRtR tV 3R ltRtV tR# )Ú
ExportCase)Údefault_factoryNc                óp  € \        V P                  V P                  4       V P                  e   \        V P                  / 4       V P                   F  p\        V4       K  	  \        V P                  \        4      '       d   \        V P                  4      ^ 8X  d   \        RV P                   R24      hR # )NzInvalid description: "Ú")r(   Úexample_argsÚexample_kwargsÚ
extra_argsÚtagsr:   r   Údescriptionr$   Úlenr!   )Úselfr+   s   & r   Ú__post_init__ÚExportCase.__post_init__X   s�   € Ü˜$×+Ñ+¨T×-@Ñ-@ÔAØ�?‰?Ò&Ü˜dŸo™o¨rÔ2à—9”9ˆCÜ˜#Öñ ô ˜$×*Ñ*¬C×0Ò0´C¸×8HÑ8HÓ4IÈQÔ4NÜÐ5°d×6FÑ6FÐ5GÀqÐIÓJÐJñ 5Or   c                ó  <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[P                  P                  ;R&   S[;R&   S[S[S[3,          ;R&   S[ R,          ;R&   S[S[,          ;R&   S[;R	&   S[S[S[3,          R,          ;R
&   # )r*   rA   rE   ÚmodelÚnamerB   NrC   rD   Úsupport_levelÚdynamic_shapes)	ÚArgsTyper$   r   ÚnnÚModuler#   r   r3   r   )r,   Ú__classdict__s   "€r   r-   ÚExportCase.__annotate__K   sœ   ø‡ ‚ áÑñ ñ Ññ ñ �8‰8�?‰?Ññ	 ñ
 �Iñ ñ ™™c˜•NÑ@ñ ñ ˜4•Ñ&ñ ñ ‰c�(Ñ/ñ ñ  Ñ8ñ ñ ™™c˜•N TÕ)Ñ0ò r   r   )r   r   r   r   r   r#   rB   rC   r3   rD   r   r   rM   rN   rH   Ú__annotate_func__r   Ú__classdictcell__)rR   s   @r   r=   r=   K   s=   ø‡ € ñ &+¸4Ô%@€NØ"&€Já¨3Ô/€DØ".×"8Ñ"8€MØ,0€Nò	K÷ ƒ r   r=   c                ó(   € V ^8„  d   QhR\         RR/# )r*   ÚcaseÚreturnN)r=   )r,   s   "r   r-   r-   j   s   € ÷ 
%ñ 
%œ:ð 
%¨$ñ 
%r   c                ó"  € V P                   \        9   dg   V P                   \        9  d*   \        V P                   ,          .\        V P                   &   \        V P                   ,          P                  V 4       R# V \        V P                   &   R# )z9
Registers a user provided ExportCase into example bank.
N)rL   Ú_EXAMPLE_CASESÚ_EXAMPLE_CONFLICT_CASESÚappend)rW   s   &r   Úregister_db_caser]   j   s`   € ð ‡y�y”NÔ"Ø�9‰9Ô3Ô3Ü2@ÀÇÁÕ2KÐ1LÔ# D§I¡IÑ.Ü §	¡	Õ*×1Ñ1°$Ô7Ùà $„N�4—9‘9Ór   c                 ó~   € \         P                  ! R RV 4      p \         P                  ! RRV 4      P                  4       # )z(.)([A-Z][a-z]+)z\1_\2z([a-z0-9])([A-Z]))ÚreÚsubÚlower)rL   s   &r   Úto_snake_caserb   w   s2   € Ü�6Š6Ð$ h°Ó5€DÜ�6Š6Ð% x°Ó6×<Ñ<Ó>Ð>r   c           	      óü   € \        V \        P                  P                  4      '       g   \	        R 4      hRV9  d/   V P
                  f   \        RV  24      h/ VCRV P
                  /Cp\        R/ / VCRV RV/CB # )z.Export case class should be a torch.nn.Module.rE   z9Could not find description or docstring for export case: rK   rL   r   )r   r   rP   rQ   Ú	TypeErrorr   r6   r=   )ÚmrL   Úconfigss   &&&r   Ú_make_export_caserg   |   s‚   € Ü�aœŸ™Ÿ™×)Ò)ÜÐHÓIÐIà˜GÔ#à�9‰9ÒÜ ØKÈAÈ3ÐOóð ð 8�WÐ7˜m¨Q¯Y©YÑ7ˆäÑ>Ð=˜7Ð= G¨Q°¸Ñ=Ñ>Ð>r   c                 ó   a € V 3R lpV# )zC
Decorator for registering a user provided case into example bank.
c                 ó  <€ Sp\         P                  ! V 4      pV\        9   d   \        R 4      hVf   \	        R4      h\        P                  V4       VP                  P                  R4      R,          p\        WV4      p\        V4       V# )z6export_case should only be used once per example file.zmodule must not be Noner/   éÿÿÿÿ)
ÚinspectÚ	getmoduleÚ_MODULESÚRuntimeErrorr6   Úaddr   r1   rg   r]   )re   rf   ÚmoduleÚmodule_namerW   r&   s   &    €r   ÚwrapperÚexport_case.<locals>.wrapper�   s{   ø€ ØˆÜ×"Ò" 1Ó%ˆØ”XÔÜÐWÓXÐXàŠ>Ü Ð!:Ó;Ð;Ü�‰�VÔØ—o‘o×+Ñ+¨CÓ0°Õ4ˆÜ  °Ó9ˆÜ˜ÔØˆr   r   ©r&   rr   s   l r   Úexport_caseru   ‹   s   ø€ õ
ð €Nr   c                  ó   a € V 3R  lpV# )c                 ó^  <€ SpVP                  R 4      p\        V\        4      '       g   \        R\	        V4       24      hVP
                  pV\        9  d
   . \        V&   VP                  VR&   \        V \        V P                  4      V4      p\        V,          P                  V4       V# )Úparentzexpected ExportCase, got rA   )Úpopr   r=   r6   r"   rL   Ú_EXAMPLE_REWRITE_CASESrA   rg   rb   r   r\   )re   rf   rx   r'   rW   r&   s   &    €r   rr   Ú$export_rewrite_case.<locals>.wrapper¢   s–   ø€ Øˆà—‘˜XÓ&ˆÜ˜&¤*×-Ò-Ü Ð#<¼TÀ&»\¸NÐ!KÓLÐLØ�k‰kˆØÔ,Ô,Ø*,Ô" 3Ñ'à"(×"5Ñ"5ˆ�ÑÜ  ¤M°!·*±*Ó$=¸wÓGˆÜ˜sÕ#×*Ñ*¨4Ô0Øˆr   r   rt   s   l r   Úexport_rewrite_caser|   ¡   s   ø€ õð €Nr   c                óÌ  € V ^8„  d   Qh/ ^ \         9   d,   \        \        \        \        \        3,          3,          ;R&   ^\         9   d   \        \        \        3,          ;R&   ^\         9   d   \
        \        ,          ;R&   ^\         9   d&   \        \        \        \        ,          3,          ;R&   ^\         9   d&   \        \        \        \        ,          3,          ;R&   # )r*   r2   rZ   rm   r[   rz   )Ú__conditional_annotations__r#   r$   r   r=   r3   r   Úlist)r,   s   "r   r-   r-      s©   € ‡�÷ò „tŒC””cœ3�h•ÐÕ ñ ñ ÷D +Ò *””Sœ*�_Õ%Ñ *ñE ÷F "Ò !Œ#Œj�/Ñ !ñG ÷H :Ò 9œœc¤4¬
Õ#3Ð3Õ4Ñ 9ñI ÷J 9Ò 8œœS¤$¤zÕ"2Ð2Õ3Ñ 8òK r   ) r~   rk   r_   r4   Údataclassesr   r   Úenumr   Útypingr   Útypesr   r   r2   r   r    rO   r(   r:   r=   rZ   r3   rm   r[   rz   r]   rb   rg   ru   r|   r-   )r~   s   @r   Ú<module>r„      s  øðç €Û 	Û ß (Ý Ý Ý ã ð Ø�Ø˜Ø˜ØˆrØ˜Ø�BØ�Bðð Ø�"Ø�2Ø�2Ø˜2Ø˜Ø˜"Ø˜BØ˜ð	ð$€ó ô.�4ô ð ��c��?€òõ	Lñ �$Ô÷Kð Kó ðKð0 )+€Ó *Ù›E€Ó !Ø79Ð Ó 9Ø68Ð Ó 8õ
%ò?ò
?òô,r   