+
    –°³já  ã                   ó¨   € ^ RI t ^ RIt^ RIHt ^ RIHt ^ RIt^ RIt^ RIt^ RI	H
t
 ] P                  ! ]4      tR.tRR R lltRRR	R
R/R R llltR# )é    N)ÚIterable)ÚAny)Úlog_export_usageÚreport_exportabilityc                ó   € V ^8„  d   QhR\         P                  P                  R\        \        ,          R\
        \        R3,          R\        \        \        3,          R,          R\        \        \
        \        \        3,          3,          /# )é   ÚmodelÚtarget_submodulesÚargs.ÚkwargsNÚreturn)ÚtorchÚnnÚModuler   ÚstrÚtupler   Údict)Úformats   "ÚT/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_export/tools.pyÚ__annotate__r      sm   € ÷ *ñ *Ü�8‰8�?‰?ð*ä¤•}ð*ô ”�S��/ð*ô ””c��N˜TÕ!ð	*ô
 
Œ#Œu”Sœ#�X�Ð
Õñ*ó    c                óø  a
a€ T;'       g    / p. p/ o
V P                  4        UUu/ uF  w  rVWebK	  	  uppoV
V3R lp V P                  4        F/  w  rVWQ9   g   K  VP                  VP                  VRR7      4       K1  	  V ! V/ VB  V F  p	V	P                  4        K  	  S
# u uppi   \         d&   p\        P
                  ! RT 2^R7        Rp?LLRp?ii ; i  T F  p	T	P                  4        K  	  i ; i)aZ  
Generate inputs for targeting submodules in the given model. Note that if two submodules refer to the same obj, this
function doesn't work.

Args:
    model: root model.
    inputs: inputs to the root model.
    target_submodules: submodules that we want to generate inputs for.

Returns:
    A dict that maps from submodule name to its inputs.
c                 ó"   <€ W3SSV ,          &   R # ©N© )ÚmoduleÚmodule_argsÚmodule_kwargsÚresultsÚsubmodule_to_namess   &&&€€r   Úpre_forwardÚ4_generate_inputs_for_submodules.<locals>.pre_forward*   s   ø€ Ø/:Ð.JˆÐ" 6Õ*Ó+r   T)Úwith_kwargszDFailed to generate submodule inputs because of the following error:
)Ú
stacklevelN)Únamed_modulesÚappendÚregister_forward_pre_hookÚ	ExceptionÚwarningsÚwarnÚremove)r	   r
   r   r   ÚhandlesÚnameÚmodr!   ÚeÚhr   r    s   &&&&      @@r   Ú_generate_inputs_for_submodulesr1      sú   ù€ ð$ �\ˆ\�r€Fà€GØ€GØ5:×5HÑ5HÔ5JÔKÑ5J©	¨˜#š)Ñ5JÒKÐöKðØ×,Ñ,Ö.‰IˆDØÖ(Ø—‘Ø×1Ñ1°+È4Ð1ÓPöñ /ñ
 	ˆtÐ�vÒó ˆAØ�H‰HŽJñ à€Nùó) Løô ô 
Ü�ŠØSÐTUÐSVÐWØ÷	
ò 	
ûð
ûó ˆAØ�H‰HŽJò ús4   ¦B%¿B+ Á.B+ Â+CÂ6CÃC ÃCÃC ÃC9ÚstrictTÚpre_dispatchFc                ó  € V ^8„  d   QhR\         P                  P                  R\        \        R3,          R\
        \        \        3,          R,          R\        R\        R\
        \        \        R,          3,          /# )	r   r.   r   .r   Nr2   r3   r   )	r   r   r   r   r   r   r   Úboolr(   )r   s   "r   r   r   ?   sx   € ÷ Uñ UÜ	�‰�‰ðUä
”�S��/ðUô ””c��N˜TÕ!ðUô
 ðUô ðUô 
Œ#Œy˜4ÕÐ
Õ ñUr   c               ó,  aaaaaa€ \        RR7       T;'       g    / pV P                  4        UUu. uF  w  rVVR8w  g   K  VNK  	  ppp\        WW4      o\        4       o/ oVVVVVV3R loS! V RW4       \        4       pSP	                  4        F;  p	V	f   K	  \        V	4      P                  R4      ^ ,          p
VP                  V
4       K=  	  \        P                  R\        V4      4       V F  p\        P                  V4       K  	  S# u uppi )a^  
Report exportability issues for a module in one-shot.

Args:
    mod: root module.
    args: args to the root module.
    kwargs: kwargs to the root module.
Returns:
    A dict that maps from submodule name to the exception that was raised when trying to export it.
    `None` means the module is exportable without issue.
Sample output:
    {
        '': UnsupportedOperatorException(func=<OpOverload(op='testlib.op_missing_meta', overload='default')>),
        'submod_1': UnsupportedOperatorException(func=<OpOverload(op='testlib.op_missing_meta', overload='default')>),
        'submod_2': None
    }
zexport.report_exportability)ÚeventÚ c                 ó<  <€ \        V 4      S9   d   R # SP                  \        V 4      4       Vf   VeM    \        P                  P                  P                  V VVSSR7       R SV&   \        P                  RV4       R # V P                  4        F2  w  rgVR8X  d   TMV RV 2pSP                  VR4      w  ršS! WxWš4       K4  	  R #   \         dG   p\        T4      P                  R4      ^ ,          p\        P                  RY4       TST&    R p?L”R p?ii ; i)N)r2   r3   zSuccessfully exported `%s`Ú
z(Failed exporting `%s` with exception: %sr8   Ú.)NN)ÚtypeÚaddr   ÚexportÚ_traceÚ_exportÚlogÚinfor(   ÚreprÚsplitÚwarningÚnamed_childrenÚget)r   Úmodule_namer   r   r/   Ú	short_msgr-   ÚsubmodÚsub_module_nameÚsubmod_argsÚsubmod_kwargsr3   Úreportr2   Úsubmod_inputsÚtried_module_typesÚ
try_exports   &&&&       €€€€€€r   rQ   Ú(report_exportability.<locals>.try_exportc   s   ø€ ô �‹<Ð-Ô-ÙØ×Ñœt F›|Ô,àÒ˜vÒ1ð(Ü—‘×#Ñ#×+Ñ+ØØØØ!Ø!-ð ,ô ð '+��{Ñ#Ü—‘Ð5°{ÔCÙð #×1Ñ1Ö3‰LˆDØ&1°RÔ&7™dÀ¸}ÈAÈdÈVÐ=TˆOà)6×):Ñ):Ø ó*Ñ&ˆKñ �v°ÖKñ 4ñ 	øô! ô (Ü  ›GŸM™M¨$Ó/°Õ2�	Ü—‘Ø>Àôð '(��{Ö#ûð(ús   ·A	C
 Ã
DÃ<DÄDz\nzFound %d export issues:)r   r%   r1   ÚsetÚvaluesrC   rD   r=   rA   rE   Úlen)r.   r   r   r2   r3   r-   Ú_Úall_submod_namesÚunique_issuesÚ	exceptionÚkeyÚissuerN   rO   rP   rQ   s   &&&dd       @@@@r   r   r   ?   só   ý€ ô4 Ð8Õ9à�\ˆ\�r€Fà,/×,=Ñ,=Ô,?ÔNÑ,?¡ À4È2Á:Ÿ˜Ñ,?ÐÑNÜ3°CÈ4ÓX€Mä›ÐØ*,€F÷#ò #ñJ ˆs�B˜Ô%ä“E€MØ—]‘]–_ˆ	ØÔ Ü�y“/×'Ñ'¨Ó.¨qÕ1ˆCØ×Ñ˜cÖ"ñ %ô
 ‡K�KÐ)¬3¨}Ó+=Ô>ÛˆÜ�‰�EÖñ ð €Mùóo Os   ²DÁDr   )Úloggingr)   Úcollections.abcr   Útypingr   r   Útorch.exportÚtorch.export._traceÚtorch._utils_internalr   Ú	getLoggerÚ__name__rA   Ú__all__r1   r   r   r   r   Ú<module>re      s_   ðã Û Ý $Ý ã Û Û Ý 2ð ×Ò˜Ó!€à!Ð
"€÷*ñZUð
 ðUð ÷Uò Ur   