+
    –°³jRO  ã                   ó\  € ^ RI t ^ RIt^ RIt^ RIt^ RIt^ RIt^ RIt^ RIt^ RIt^ RI	H
t
Ht ^ RIHtHtHt ^ RI Ht ^ RIHtHtHt ^ RIt^ RIt^ RIHt ^ RIHt ^ RIHt ^R	IHt ^R
IH t H!t! ^RI"H#t# ]'       d   ^RI$H%t% ^RI&H't'H(t( ]PR                  ! ]*4      t+R R lt,] ! R R4      4       t- ! R R]P\                  4      t/ ! R R]P`                  4      t1] ! R R4      4       t2R R lt3] ! R R4      4       t4R$R R llt5] ! R R4      4       t6] ! R  R!4      4       t7R" R# lt8R# )%é    N)ÚCallableÚSequence)ÚAbstractContextManagerÚ	ExitStackÚnullcontext)Ú	dataclass)ÚAnyÚOptionalÚTYPE_CHECKING)ÚGraphRuntimeEnv)Ú_graph_device_type)Ú
SystemInfo)Úconvert_frame)Ú&BundledAOTAutogradSerializableCallableÚSerializableCallable)ÚHooks)ÚGuardManagerWrapper)ÚSerializedCodeÚ
SourceInfoc          
      ó~   € V ^8„  d   QhR\         P                  R\        R\        R\        \        \        3,          /# )é   Ú	signatureÚargsÚkwargsÚreturn)ÚinspectÚ	Signaturer	   ÚdictÚstr)Úformats   "ÚZ/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_dynamo/aot_compile.pyÚ__annotate__r"   %   s:   € ÷ %ñ %Ü× Ñ ð%Ü),ð%Ü8;ð%ä	Œ#Œsˆ(…^ñ%ó    c                 ó^   € V P                   ! V/ VB pVP                  4        VP                  # ©N)ÚbindÚapply_defaultsÚ	arguments)r   r   r   Úbound_argumentss   &*, r!   Úbind_localsr*   %   s/   € ð  —n’n dÐ5¨fÑ5€OØ×"Ñ"Ô$Ø×$Ñ$Ð$r#   c                   óx   a € ] tR t^-t o ]P
                  ! ]P                  R7      tV 3R lR lt	V 3R lt
RtV tR# )ÚCompileArtifacts)Údefault_factoryc                ó   <€ V ^8„  d   QhRR/# ©r   r   N© )r    Ú__classdict__s   "€r!   r"   ÚCompileArtifacts.__annotate__;   s   ø€ ÷ Oñ O Tñ Or#   c                ó|   € \         P                  ! 4       pVP                  V P                  V P                  4       R # r%   )r   ÚcurrentÚcheck_compatibilityÚsystem_infoÚdevice_type)ÚselfÚcurrent_systems   & r!   r5   Ú$CompileArtifacts.check_compatibility;   s,   € Ü#×+Ò+Ó-ˆØ×*Ñ*¨4×+;Ñ+;¸T×=MÑ=MÖNr#   c                óÒ   <€ V ^8„  d   Qh/ S[ P                  ;R&   S[R,          ;R&   S[;R&   S[;R&   S[;R&   S[P                  ;R&   S[;R&   R	;R
&   S[;R&   S[;R&   S[	;R&   # )r   r   r   Úguard_managerÚguards_stateÚ
backend_idÚcompiled_fnÚoriginal_codeÚruntime_envr   Úsource_infor7   Úbackend_namer6   )
r   r   r
   Úbytesr   r   ÚtypesÚCodeTyper   r   )r    r1   s   "€r!   r"   r2   -   s¡   ø‡ ‚ á× Ñ Ñ ñ ñ Ð1Õ2Ñ2ñ ñ Ññ	 ñ
 �Oñ ñ &Ñ%ñ ñ —>‘>Ñ!ñ ñ !Ñ ñ ð Ññ ñ Ññ ñ Ññ ñ ÑSò r#   r0   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__ÚdataclassesÚfieldr   r4   r6   r5   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__©r1   s   @r!   r,   r,   -   s0   ø‡ € ð *×/Ò/À
×@RÑ@RÔS€K÷Oð O÷ ƒ r#   r,   c                   óî   a a€ ] tR t^@t oV3R lV 3R lltV3R lR lt]V3R lR l4       t]V3R lR l4       t]V3R	 lR
 l4       t	]V3R lR l4       t
]V3R lR l4       tV3R lR ltRtVtV ;t# )ÚAOTCompilePicklerc                óT   <€ V ^8„  d   QhRS[ S[S[3,          RS[P                  RR/# )r   Úexternal_dataÚbufr   N©r   r   ÚobjectÚioÚBytesIO)r    r1   s   "€r!   r"   ÚAOTCompilePickler.__annotate__A   s/   ø€ ÷ ñ ¡d©3±¨;Õ&7ð ¹b¿j¹jð ÈTñ r#   c                ó²   <€ \         SV `  V4       Wn        VP                  4        UUu/ uF  w  r4\	        V4      VbK  	  uppV n        / V n        R # u uppi r%   )ÚsuperÚ__init__rT   ÚitemsÚidÚid_mapÚerrors)r8   rT   rU   ÚkeyÚvalueÚ	__class__s   &&&  €r!   r]   ÚAOTCompilePickler.__init__A   sS   ø€ Ü‰Ñ˜ÔØ*Ôà-:×-@Ñ-@Ô-Bô'
Ù-B™z˜sŒBˆu‹I�sŠNÑ-Bò'
ˆŒð ˆŽùó'
s   ªAc                óD   <€ V ^8„  d   QhRS[ RS[S[,          R,          /# )r   Úobjr   N)rW   Úintr   )r    r1   s   "€r!   r"   rZ   I   s$   ø€ ÷ ñ ¡ð ©C±#­I¸Õ,<ñ r#   c                ó
  € \        V4      V P                  9   d   V P                  \        V4      ,          # \        V\        P                  P
                  4      '       d#   WP                  \        V4      &   \        V4      # R # r%   )r_   r`   Ú
isinstanceÚtorchÚnnÚModulera   )r8   rg   s   &&r!   Úpersistent_idÚAOTCompilePickler.persistent_idI   sV   € Üˆc‹7�d—k‘kÔ!Ø—;‘;œr #›wÕ'Ð'Ü˜œUŸX™XŸ_™_×-Ò-Ø#&�K‰Kœ˜3›Ñ Ü�c“7ˆNár#   c                ó&   <€ V ^8„  d   QhRS[ RS[ /# )r   Úvalr   ©rW   )r    r1   s   "€r!   r"   rZ   S   s   ø€ ÷  ñ  ¡ð  ©Fñ  r#   c                ón   a€ R  V3R llpVP                   f   \        R4      hVP                   ^ ,          # )c                ó$   € V ^8„  d   QhR\         /# ©r   r   rr   )r    s   "r!   r"   Ú6AOTCompilePickler._unpickle_cell.<locals>.__annotate__T   s   € ÷ 	ñ 	”6ñ 	r#   c                  ó   <€ S # r%   r0   )rq   s   €r!   Ú_Ú+AOTCompilePickler._unpickle_cell.<locals>._T   s   ø€ ØˆJr#   zclosure must not be None)Ú__closure__ÚAssertionError)Úclsrq   rx   s   &f r!   Ú_unpickle_cellÚ AOTCompilePickler._unpickle_cellR   s4   ø€ ÷	ð 	ð �=‰=Ò Ü Ð!;Ó<Ð<Ø�}‰}˜QÕÐr#   c                ó@   <€ V ^8„  d   QhRS[ RS[RS[P                  /# )r   ÚfuncÚbaser   )r   rW   rE   Ú
MethodType)r    r1   s   "€r!   r"   rZ   ]   s(   ø€ ÷ ,ñ ,©(ð ,¹&ð ,ÁU×EUÑEUñ ,r#   c                ó.   € \         P                  ! W4      # r%   )rE   r‚   )r|   r€   r�   s   &&&r!   Ú_unpickle_bound_methodÚ(AOTCompilePickler._unpickle_bound_method[   s   € ô ×Ò Ó+Ð+r#   c                ó:   <€ V ^8„  d   QhRS[ RS[P                  /# )r   Únamer   )r   rE   Ú
ModuleType)r    r1   s   "€r!   r"   rZ   a   s!   ø€ ÷ -ñ -¡Cð -©E×,<Ñ,<ñ -r#   c                ó.   € \         P                  ! V4      # r%   )Ú	importlibÚimport_module)r|   r‡   s   &&r!   Ú_unpickle_moduleÚ"AOTCompilePickler._unpickle_module`   s   € ä×&Ò& tÓ,Ð,r#   c                ó8   <€ V ^8„  d   QhRRRS[ P                  /# )r   Úserialized_coder   r   )rE   rF   )r    r1   s   "€r!   r"   rZ   e   s    ø€ ÷ >ñ >Ð-=ð >Á%Ç.Á.ñ >r#   c                ó2   € ^ RI Hp VP                  ! V4      # )r   ©r   )Útorch._dynamo.packager   Úto_code_object)r|   r�   r   s   && r!   Ú_unpickle_codeÚ AOTCompilePickler._unpickle_coded   s   € å8à×,Ò,¨_Ó=Ð=r#   c                ó¾   <€ V ^8„  d   QhRS[ P                  RS[RS[RS[S[R3,          R,          RS[S[ P
                  R3,          R,          RS[ P                  /# )	r   ÚcodeÚmoduleÚqualnameÚargdefs.NÚclosurer   )rE   rF   r   ÚtuplerW   ÚCellTypeÚFunctionType)r    r1   s   "€r!   r"   rZ   k   sz   ø€ ÷ 	Oñ 	Oá�n‰nð	Oñ ð	Oñ ð		Oñ
 ‘v˜s�{Õ# dÕ*ð	Oñ ‘u—~‘~ sÐ*Õ+¨dÕ2ð	Oñ 
×	Ñ	ñ	Or#   c                ór   € \         P                  ! V4      P                  p\        P                  ! WW4V4      # r%   )rŠ   r‹   Ú__dict__rE   rž   )r|   r—   r˜   r™   rš   r›   Ú	f_globalss   &&&&&& r!   Ú_unpickle_nested_functionÚ+AOTCompilePickler._unpickle_nested_functionj   s/   € ô ×+Ò+¨FÓ3×<Ñ<ˆ	Ü×!Ò! $°8ÀgÓNÐNr#   c                ó&   <€ V ^8„  d   QhRS[ RS[ /# )r   rg   r   ©r	   )r    r1   s   "€r!   r"   rZ   w   s   ø€ ÷ "ñ "¡Cð "©Cñ "r#   c                óê  € \        V\        R  ! ^ 4      P                  ^ ,          4      4      '       d#   \        V 4      P                  VP                  33# \
        P                  ! V4      '       d/   ^ RIHp \        V 4      P                  VP                  ! V4      33# \
        P                  ! V4      '       d#   \        V 4      P                  VP                  33# \
        P                  ! V4      '       d{    VP                  pVP                   p\#        WCP                  4      p\
        P                  ! V4      '       d   VP                  pW5Jd   \        V 4      P$                  W433#  \2        # \
        P&                  ! V4      '       d`   RVP(                  9   dO   \        V 4      P*                  VP,                  VP.                  VP(                  VP0                  VP                  33# \2        # )c                 ó   a € V 3R  l# )c                  ó   <€ S # r%   r0   ©Úxs   €r!   Ú<lambda>ÚFAOTCompilePickler.reducer_override.<locals>.<lambda>.<locals>.<lambda>x   s   ø€ ±1r#   r0   r©   s   fr!   r«   Ú4AOTCompilePickler.reducer_override.<locals>.<lambda>x   s   ø€ ­9r#   r‘   z<locals>)rj   Útyperz   r}   Úcell_contentsr   Úiscoder’   r   r”   Úfrom_code_objectÚismodulerŒ   rG   ÚismethodÚ__func__Ú__self__Úgetattrr„   Ú
isfunctionrI   r¢   Ú__code__rH   Ú__defaults__ÚNotImplemented)r8   rg   r   r€   Úmethod_selfÚ
inner_funcs   &&    r!   Úreducer_overrideÚ"AOTCompilePickler.reducer_overridew   s�  € Ü�cœ4Ò!4°aÓ 8× DÑ DÀQÕ GÓH×IÒIÜ˜“:×,Ñ,¨s×/@Ñ/@Ð.BÐBÐBÜ�^Š^˜C× Ò Ý<ä˜“:×,Ñ,¨~×/NÒ/NÈsÓ/SÐ.UÐUÐUä×Ò˜c×"Ò"Ü˜“:×.Ñ.°·±°Ð?Ð?Ü×Ò˜c×"Ò"ðð —<‘<ˆDØŸ,™,ˆKÜ  ¯m©mÓ<ˆJÜ×Ò 
×+Ò+Ø'×0Ñ0�
ØÓ%Ü˜D“z×8Ñ8¸4Ð:MÐMÐMð &ô Ðô ×Ò ×$Ò$Ø˜S×-Ñ-Ô-Ü˜D“z×;Ñ;Ø—L‘LØ—N‘NØ×$Ñ$Ø×$Ñ$Ø—O‘Oð>ð ð ô Ðr#   )ra   rT   r`   )rG   rH   rI   rJ   r]   rn   Úclassmethodr}   r„   rŒ   r”   r¢   r½   rN   rO   Ú__classcell__©rd   r1   s   @@r!   rR   rR   @   sŽ   ù‡ € ÷ó ÷ð ð ÷ ó ð ð ÷,ó ð,ð ÷-ó ð-ð ÷>ó ð>ð
 ÷	Oó ð	O÷"÷ "ð "r#   rR   c                   óP   a a€ ] tR t^œt oV3R lV 3R lltV3R lR ltRtVtV ;t# )ÚAOTCompileUnpicklerc                óV   <€ V ^8„  d   QhRS[ S[S[3,          RS[P                  RS[/# )r   rT   Úfiler   rV   )r    r1   s   "€r!   r"   Ú AOTCompileUnpickler.__annotate__�   s/   ø€ ÷ +ñ +¡d©3±¨;Õ&7ð +¹r¿z¹zð +Éfñ +r#   c                ó2   <€ \         SV `  V4       Wn        R # r%   )r\   r]   rT   )r8   rT   rÅ   rd   s   &&&€r!   r]   ÚAOTCompileUnpickler.__init__�   s   ø€ Ü‰Ñ˜ÔØ*Ör#   c                ó&   <€ V ^8„  d   QhRS[ RS[/# )r   rb   r   )r   rW   )r    r1   s   "€r!   r"   rÆ   ¡   s   ø€ ÷ 'ñ '¡3ð '©6ñ 'r#   c                ó~   € WP                   9  d   \        R V RV P                    24      hV P                   V,          # )z-Missing required external reference to data: zS. Please load AOT compiled function with `external_data=<external data dictionary>`)rT   ÚRuntimeError)r8   rb   s   &&r!   Úpersistent_loadÚ#AOTCompileUnpickler.persistent_load¡   sN   € Ø×(Ñ(Ô(ÜØ?À¸uð E=ð ×%Ñ%Ð&ð(óð ð ×!Ñ! #Õ&Ð&r#   )rT   )	rG   rH   rI   rJ   r]   rÌ   rN   rO   rÀ   rÁ   s   @@r!   rÃ   rÃ   œ   s   ù‡ € ÷+ó +÷'÷ 'ð 'r#   rÃ   c                   ó,   a € ] tR t^¬t o V 3R ltRtV tR# )ÚAOTCompileSaveResultc                ó&   <€ V ^8„  d   Qh/ S[ ;R&   # )r   Úserialized_data©rD   )r    r1   s   "€r!   r"   Ú!AOTCompileSaveResult.__annotate__¬   s   ø‡ ‚ áÑò r#   r0   N)rG   rH   rI   rJ   rM   rN   rO   rP   s   @r!   rÏ   rÏ   ¬   s   ø‡ ‡ „ r#   rÏ   c                ó0   € V ^8„  d   QhR\         R\        /# )r   Ú	file_pathÚdata)r   rD   )r    s   "r!   r"   r"   ±   s   € ÷ %ñ %¤3ð %¬eñ %r#   c                 ó¢  € \         P                  P                  V 4      ;'       g    R p\        P                  ! VRRR7      ;_uu_ 4       pVP
                  pVP                  V4       VP                  4        \         P                  ! VP                  4       4       RRR4       \         P                  ! XV 4       R#   + '       g   i     L); i)Ú.FÚwb)ÚdirÚdeleteÚmodeN)ÚosÚpathÚdirnameÚtempfileÚNamedTemporaryFiler‡   ÚwriteÚflushÚfsyncÚfilenoÚreplace)rÕ   rÖ   Údir_nameÚ	temp_fileÚ	temp_paths   &&   r!   Úatomic_write_binaryrê   ±   s�   € Ü�w‰w�‰˜yÓ)×0Ð0¨S€Hä	×	$Ò	$Ø˜U¨÷
õ 
à	Ø—N‘Nˆ	Ø�‰˜ÔØ�‰ÔÜ
�Š�×!Ñ!Ó#Ô$÷
ô ‡J‚Jˆy˜)Ö$÷
÷ 
ús   ÁAB>Â>C	c                   óö   a € ] tR t^¿t o RtRtV 3R lR ltV 3R lR ltV 3R lR ltV 3R	 lR
 lt	V 3R lR lt
RV 3R lR llt]RV 3R lR ll4       t]RV 3R lR ll4       tV 3R lR ltV 3R ltRtV tR# )ÚAOTCompiledFunctionTNc                óB   <€ V ^8„  d   QhRS[ RS[ RS[S[S[ 3,          /# ©r   r   r   r   )rW   r   r   )r    r1   s   "€r!   r"   Ú AOTCompiledFunction.__annotate__Å   s+   ø€ ÷ ñ ¡fð ¹ð Á4ÉÉVÈÕCTñ r#   c                ó  € / pV P                   P                  pVP                  '       d¦   VP                  P                  '       d8   \        VP                  4      \        VP                  P                  4      8w  d   \        R 4      h\        VP                  P                  VP                  4       UUu/ uF  w  rVWVP                  bK  	  pppVP                  \        V P                   P                  .VO5/ VB 4       V# u uppi )z,closure length must match co_freevars length)Ú
_artifactsrA   r›   ÚbytecodeÚco_freevarsÚlenr{   Úzipr¯   Úupdater*   r   )r8   r   r   Úf_localsÚenvr‡   Úcells   &*,    r!   Úprepare_f_localsÚ$AOTCompiledFunction.prepare_f_localsÅ   sÕ   € Ø&(ˆØ�o‰o×)Ñ)ˆØ�;�;ˆ;Ø—<‘<×+×+Ð+¬s°3·;±;Ó/?Ä3Ø—‘×(Ñ(óDô 0ô %Ð%SÓTÐTô #& c§l¡l×&>Ñ&>ÀÇÁÔ"Lôá"L‘J�Dð ×(Ñ(Ò(Ù"Lð ñ ð 	�‰œ D§O¡O×$=Ñ$=ÐOÀÒOÈÑOÔPØˆùós   Â6Dc                ó,   <€ V ^8„  d   QhRS[ RS[ RS[/# rî   )r	   Úbool)r    r1   s   "€r!   r"   rï   Ô   s"   ø€ ÷ =ñ =¡ð =±ð =¹ñ =r#   c                ó¶   € V P                   ! V/ VB pV P                  P                  f   \        R4      hV P                  P                  P	                  V4      # )Núguard_manager must not be None)rú   rñ   r<   r{   Úcheck)r8   r   r   r÷   s   &*, r!   Úguard_checkÚAOTCompiledFunction.guard_checkÔ   sN   € Ø×(Ò(¨$Ð9°&Ñ9ˆØ�?‰?×(Ñ(Ò0Ü Ð!AÓBÐBØ�‰×,Ñ,×2Ñ2°8Ó<Ð<r#   c                ó   <€ V ^8„  d   QhRR/# r/   r0   )r    r1   s   "€r!   r"   rï   Ú   s   ø€ ÷ ñ ˜tñ r#   c                óö  € ^RI HpHp V P                  P	                  4        V P                  P
                  P                  V P                  P                  V P                  P                  V P                  R7      V n
        V P                  P                  f`   V! V P                  P                  4      pV! VV P                  P                  V P                  P                  4      V P                  n        R# R# )é   )Úload_guard_managerÚload_guards_state)Úextra_globalsN)Úpackager  r  rñ   r5   rA   Úforward_callabler>   r?   Ú_extra_globalsÚfnr<   r=   r@   Ú__globals__)r8   r  r  r=   s   &   r!   Ú__post_init__Ú!AOTCompiledFunction.__post_init__Ú   s²   € ßBà�‰×+Ñ+Ô-à—/‘/×-Ñ-×>Ñ>Ø�O‰O×&Ñ&Ø�O‰O×'Ñ'Ø×-Ñ-ð ?ó 
ˆŒð �?‰?×(Ñ(Ò0Ù,¨T¯_©_×-IÑ-IÓJˆLÙ,>ØØ—‘×-Ñ-Ø—‘×#Ñ#ó-ˆD�O‰OÖ)ñ 1r#   c                ó,   <€ V ^8„  d   QhRS[ RS[ RS[ /# rî   r¥   )r    r1   s   "€r!   r"   rï   í   s"   ø€ ÷ (ñ (™cð (©Sð (±Sñ (r#   c                ó\  € V P                   P                  f   \        R4      hV P                  '       dg   V P                  ! V/ VB '       gO   V P
                  ! V/ VB p\        V P                   P                  P                  V4      4      p\        RV 24      hV P                  ! V/ VB # )Nrÿ   z#GuardManager check failed, reason: )
rñ   r<   r{   Ú_guard_check_enabledr  rú   r   Úcheck_verboserË   r  )r8   r   r   r÷   Úreasons   &*,  r!   Ú__call__ÚAOTCompiledFunction.__call__í   sš   € Ø�?‰?×(Ñ(Ò0Ü Ð!AÓBÐBØ×$×$Ð$¨T×-=Ò-=¸tÐ-NÀv×-NÐ-NØ×,Ò,¨dÐ=°fÑ=ˆHÜ˜Ÿ™×6Ñ6×DÑDÀXÓNÓOˆFÜÐ!DÀVÀHÐMÓNÐNØ�wŠw˜Ð' Ñ'Ð'r#   c                ó   <€ V ^8„  d   QhRR/# )r   r   r   r0   )r    r1   s   "€r!   r"   rï   ö   s   ø€ ÷ +ñ +˜\ñ +r#   c                ó.   € V P                   P                  # r%   )rñ   rB   ©r8   s   &r!   rB   ÚAOTCompiledFunction.source_infoö   s   € Ø�‰×*Ñ*Ð*r#   c                óP   <€ V ^8„  d   QhRS[ RS[S[ S[3,          R,          RS[/# )r   rÞ   rT   Nr   )r   r   r	   rÏ   )r    r1   s   "€r!   r"   rï   ù   s2   ø€ ÷ ñ ÙðÙ(,©S±#¨X­¸Õ(=ðá	ñr#   c                ód   € \        V 4      P                  W4      p\        WP                  4       V# r%   )r®   Ú	serializerê   rÑ   )r8   rÞ   rT   Úresults   &&& r!   Úsave_compiled_functionÚ*AOTCompiledFunction.save_compiled_functionù   s+   € ô �d“×%Ñ% dÓ:ˆÜ˜D×"8Ñ"8Ô9Øˆr#   c                óN   <€ V ^8„  d   QhRRRS[ S[S[3,          R,          RS[/# )r   r  rì   rT   Nr   )r   r   r	   rÏ   )r    r1   s   "€r!   r"   rï     s7   ø€ ÷ Dñ DØ&ðDÙ7;¹CÁ¸Hµ~ÈÕ7LðDá	ñDr#   c                óÄ  € ^ RI Hp VP                  P                  P	                  4       pRVR&   \
        P                  ! VR,          VP                  ! VR,          P                  4      R7      VR&   VR,          p\        V4      P                  \        V4      P                  V4      3VR&   VP                  ! VR,          4      VR&   \        P                  ! 4       p\        T;'       g    / V4      pVP                  V4       VP                   '       d1   \#        R\%        VP                   P'                  4       4       R	24      h\)        VP+                  4       R
7      # )r   r‘   Nr<   rA   ©rò   r?   r@   z+Failed to serialize the following objects: zI
Please mark these as external data by using `external_data={'key': ...}`)rÑ   )r’   r   rñ   r    ÚcopyrK   ræ   r±   rò   r®   Údeserialize_compile_artifactsÚserialize_compile_artifactsrX   rY   rR   Údumpra   rË   ÚlistÚvaluesrÏ   Úgetvalue)r|   r  rT   r   Ústater?   rU   Úpicklers   &&&     r!   r  ÚAOTCompiledFunction.serialize   s/  € õ 	9à—‘×&Ñ&×+Ñ+Ó-ˆØ!%ˆˆoÑÜ*×2Ò2Ø�-Õ Ø#×4Ò4°U¸=Õ5I×5RÑ5RÓSô 
ˆˆmÑð ˜MÕ*ˆä�Ó×;Ñ;Ü�Ó×9Ñ9¸+ÓFð 
ˆˆmÑð "0×!@Ò!@ÀÀÕAWÓ!XˆˆoÑÜ�jŠj‹lˆÜ# M×$7Ð$7°R¸Ó=ˆØ�‰�UÔØ�>�>ˆ>ÜØ=¼dÀ7Ç>Á>×CXÑCXÓCZÓ>[Ð=\ð ][ð [óð ô $°C·L±L³NÔCÐCr#   c                óx   <€ V ^8„  d   QhRS[ RS[S[S[3,          R,          RS[S[S[3,          R,          RR/# )r   rÖ   r¡   NÚexternal_closure_datar   rì   )rD   r   r   rW   r	   )r    r1   s   "€r!   r"   rï     sN   ø€ ÷ 8ñ 8áð8ñ ™™V˜Õ$ tÕ+ð8ñ  $¡C© H�~°Õ4ð	8ð
 
ñ8r#   c                óv  € ^ RI Hp \        P                  ! V4      pVP	                  ^ 4       \        T;'       g    / V4      pVP                  4       pVP                  4        \        P                  ! VR,          VP                  ! VR,          P                  4      R7      VR&   VR,          w  r‰\        P                  P                  P                  RR7      ;_uu_ 4        V! V	4      VR&   RRR4       VP                  ! VR,          4      VR&   \!        R
/ VB p
V ! W¢R	7      #   + '       g   i     L@; i)r   r‘   rA   r#  r?   T)Úenable_autograd_for_aotNr@   )r  r0   )r’   r   rX   rY   ÚseekrÃ   ÚloadÚcloserK   ræ   r“   rò   rk   Ú	_inductorÚconfigÚpatchr,   )r|   rÖ   r¡   r/  r   ÚfÚ	unpicklerr+  ÚdeserializerÚcompiled_fn_stateÚ	artifactss   &&&&       r!   ÚdeserializeÚAOTCompiledFunction.deserialize  sÿ   € õ 	9ä�JŠJ�tÓˆØ	�‰ˆqŒ	Ü'Ð(=×(CÐ(CÀÀQÓGˆ	Ø—‘Ó ˆØ	�‰Œ	Ü*×2Ò2Ø�-Õ Ø#×2Ò2°5¸Õ3G×3PÑ3PÓQô 
ˆˆmÑð +0°Õ*>Ñ'ˆÜ�_‰_×#Ñ#×)Ñ)À$Ð)×GÕGÙ#/Ð0AÓ#BˆE�-Ñ ÷ Hà!/×!>Ò!>¸uÀ_Õ?UÓ!VˆˆoÑä$Ñ- uÑ-ˆ	Ù�9Ô7Ð7÷ H×Gús   Ã$D(Ä(D8	c                ó   <€ V ^8„  d   QhRR/# r/   r0   )r    r1   s   "€r!   r"   rï   6  s   ø€ ÷ *ñ * Tñ *r#   c                ó   € R V n         R# )FN)r  r  s   &r!   Údisable_guard_checkÚ'AOTCompiledFunction.disable_guard_check6  s
   € Ø$)ˆÖ!r#   c                ób   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[S[S[3,          R,          ;R&   # )r   rñ   r  Nr  )r,   rý   r   r   rW   )r    r1   s   "€r!   r"   rï   ¿   s8   ø‡ ‚ á Ñ ñ ñ Ñ%ñ ñ ™™f˜Õ%¨Õ,Ñ3ò	 r#   )r  r  r%   )NN)rG   rH   rI   rJ   r  r  rú   r  r  r  rB   r  r¿   r  r=  rA  rM   rN   rO   rP   s   @r!   rì   rì   ¿   s•   ø‡ € ð "&ÐØ/3€N÷ð ÷=ð =÷ð ÷&(ð (÷+ð +÷ò ð ÷Dñ Dó ðDð6 ÷8ñ 8ó ð8÷2*ð *÷o ƒ r#   rì   c                óL  € V ^8„  d   QhR\         R\        \        \         R3,          \        \        \         3,          3,          R\        R\
        \        P                  P                  \        \        P                  ,          .\        3,          R\        R,          R\        /# )	r   ÚmodelÚexample_inputs.ÚhooksÚbackendÚdynamicNr   )r	   rœ   r   r   r   r   rk   ÚfxÚGraphModuler(  ÚTensorr   rý   rì   )r    s   "r!   r"   r"   :  s�   € ÷ Tñ TÜðTäœ%¤ S �/¬4´´S°­>Ð9Õ:ðTô ðTô ”u—x‘x×+Ñ+¬T´%·,±,Õ-?Ð@ÔBVÐVÕWð	Tô
 �D�[ðTô ñTr#   c                 óà
  aa€ ^ RI Ho ^ RIHp ^ RIHpHp ^ RIHp ^ RI	H
p	 Vw  r«\        4       pVe   ^ RIHp V! V4      pV! 4       ;_uu_ 4        V! R4      ;_uu_ 4        \        P                  P                   P#                  R	R
7      ;_uu_ 4        T;_uu_ 4        T;_uu_ 4        \$        P&                  ! W
V4      pVP(                  pVP*                  f   \-        R4      hVP.                  '       g   ^ RIHo V3R lV3R llpVVn        \$        P4                  ! V 4      w  ppVP6                  pVf   \-        R4      hVP8                  VP:                  n        \?        VP:                  P@                  4      pVPB                  PD                  VP*                  PD                  Jd   \-        R4      hV	! VPB                  4      pVPF                  Vn#        \        PH                  PK                  V4      ;_uu_ 4        \        P                  P                   P#                  RR	RR	RR	RR	RR	/4      ;_uu_ 4        V! VP:                  VPL                  4      p\O        V\        PP                  4      '       g{   \S        VR4      '       dI   \O        VPT                  \        PV                  PX                  PZ                  P\                  4      '       g!   \S        VR4      '       d   VP^                  e   \a        V4      pRRR4       RRR4       \O        X\b        4      '       gE   \S        VR4      '       d   VPT                  pMTp\e        R\g        V4       R2RV R2,           4      h\i        4       pVPj                  '       d    VPm                  VPo                  4       4       V;_uu_ 4        VPq                  VPr                  VR	R	R7      pRRR4       XPt                  f   \-        R4      hV! \w        4       R7      pVPx                   F  pVP{                  V4       K  	  \}        \$        P~                  ! V4      VP€                  VPt                  VP8                  VVPr                  VPƒ                  4       VV\…        VRR 4      R!7
      p\‡        VVPˆ                  R"7      pRRR4       RRR4       RRR4       RRR4       RRR4       X#   + '       g   i     ELÙ; i  + '       g   i     ELå; i  + '       g   i     EL2; i  + '       g   i     Ln; i  + '       g   i     Ly; i  + '       g   i     L„; i  + '       g   i     L�; i  + '       g   i     X# ; i)#r   )ÚCheckFunctionManager)r   )Údynamo_timedÚget_metrics_context)Ú#torch_function_mode_stack_state_mgr)ÚTracingContextN)Úset_enable_dynamicÚfullgraph_captureT)Ústrict_autograd_cachezoutput_graph must not be None)ÚGuardFilterEntryc                óV   <€ V ^8„  d   QhR\         S,          R\         \        ,          /# )r   Úguard_entriesr   )r   rý   )r    rV  s   "€r!   r"   Ú+aot_compile_fullgraph.<locals>.__annotate__`  s'   ø€ ÷ ñ Ü'Ð(8Õ9ðäœ$•ñr#   c                 ó’   <€ V  Uu. uF5  pVP                   ;'       g    VP                  SP                  9   '       * NK7  	  up# u upi r%   )Ú	is_globalÚ
guard_typeÚ%UNSUPPORTED_SERIALIZATION_GUARD_TYPES)rX  ÚgrN  s   & €r!   Únew_guard_filter_fnÚ2aot_compile_fullgraph.<locals>.new_guard_filter_fn`  sY   ø€ ñ +ó	ñ +˜ð ŸK™K÷ Zð ZØ Ÿ|™|Ø3×YÑYñ Z÷ñ +ñ	ð 	ùò 	s
   †Až#Azbackend_input must not be Nonez>fake_mode.shape_env must be the same as output_graph.shape_envrU  Úbypass_autograd_cache_keyÚbundled_autograd_cacheÚ force_non_lazy_backward_loweringÚforce_autograd_cacheÚcompiler_fnr  zCompiled function type z (produced zfrom backend z*) does not implement SerializableCallable.)rG  ÚsaveÚstrict_errorzguards_state must not be None)Úinlined_sourcesÚcompiler_nameÚunknown)
r   r<   r=   r>   r?   r@   rA   rB   r7   rC   )rñ   r  )EÚtorch._dynamo.guardsrN  r’   r   Útorch._dynamo.utilsrO  rP  Ú&torch._dynamo.variables.torch_functionrQ  Útorch._guardsrR  r   Útorch._dynamo.eval_framerS  rk   Ú
_functorchr6  r7  r   rT  Úgraph_capture_outputÚoutput_graphr{   Úguard_filter_fnÚtorch._dynamo.typesrV  Úget_traced_fnÚbackend_inputr>   Úgraph_moduleÚ_backend_idr   ÚgraphÚ	fake_modeÚ	shape_envÚtensor_to_contextÚ_guardsÚtracingrF  rj   Ú_TorchCompileInductorWrapperÚhasattrre  Ú_dynamoÚbackendsÚcommonÚAotAutogradr  r   r   rË   r®   r   ÚstackÚenter_contextÚtemp_restore_stackÚbuild_guardsr¸   r=   ÚsetÚtraced_codeÚadd_coder,   Ú_get_signaturer<   Úget_runtime_envr¶   rì   r  ) rE  rF  rG  rH  rI  r   rO  rP  rQ  rR  r   r   Údynamic_ctxrS  Úcapture_outputrq  r_  r  rx   rv  r7   Útracing_contextr?   re  Úbuild_guards_ctxÚcheck_fnrB   rŠ  r<  Úaot_compiled_fnrN  rV  s    &&&&&                         @@r!   Úaot_compile_fullgraphr”  :  so  ù€ õ :Ý0ßEõõ -à!�L€Dä“-€KØÒÝ?á(¨Ó1ˆñ 	×ÕÙÐ(×)Õ)Ü×Ñ×Ñ×%Ñ%¸DÐ%×AÕAßŠß+Ò+ä&×8Ò8¸ÀfÓMˆØ-×BÑBÐØ×,Ñ,Ò4Ü Ð!@ÓAÐAà×$×$Ð$Ý<÷ó ð %8ˆEÔ!ä×+Ò+¨EÓ2‰ˆˆAà&×4Ñ4ˆØÒ Ü Ð!AÓBÐBØ1>×1IÑ1Iˆ×"Ñ"Ô.Ü(¨×)CÑ)C×)IÑ)IÓJˆà×#Ñ#×-Ñ-Ø'×4Ñ4×>Ñ>ó?ô !ØPóð ñ )¨×)@Ñ)@ÓAˆØ,9×,KÑ,KˆÔ)ä�M‰M×!Ñ! /×2Õ2Ü×Ñ×#Ñ#×)Ñ)à+¨TØ/°Ø,¨dØ6¸Ø*¨Dð÷õ ñ "Ø×*Ñ*¨M×,HÑ,HóˆKô ˜7¤E×$FÑ$F×GÒGä˜G ]×3Ò3Ü"Ø×+Ñ+¬U¯]©]×-CÑ-C×-JÑ-J×-VÑ-V÷ò ô
 ˜K¨×5Ò5Ø#×-Ñ-Ò9ô EÀ[ÓQ�÷;÷ 3ô@ ˜+Ô';×<Ò<Ü�w ×.Ò.Ø%×1Ñ1‘à%�ÜØ)¬$¨{Ó*;Ð)<¸KÐHØ! + Ð.XÐYõZóð ô %›;ÐØ.×4×4Ð4Ø×*Ñ*Ø3×FÑFÓHô÷ ÒØ+×8Ñ8Ø—‘ 5¨tÀ$ð 9ó ˆH÷ ð
 × Ñ Ò(Ü Ð!@ÓAÐAá ´³Ô7ˆØ/×;Ô;ˆKØ× Ñ  Ö-ñ <ô %Ü#×2Ò2°2Ó6Ø"×0Ñ0Ø!×.Ñ.Ø$×/Ñ/Ø#ØŸ+™+Ø,×<Ñ<Ó>Ø#Ø#Ü  ¨/¸9ÓEô
ˆ	ô .Ø °·±ô
ˆ÷i 	,÷ 	÷ 	B÷ 	*÷ 	ðx Ð÷Y÷ ð ú÷ 3×2Ð2ú÷b ×Ðú÷y 	,×+ú÷ 	�ú÷ 	B×Aú÷ 	*×)ú÷ 	Öðx Ðús¿   ÁUÁ#3U		ÂT6Â!T#Â,ATÃ;D
TÈ<S(ÉCSÌS(ÌB'TÎ3 S<ÏCTÒ*T#Ò2T6Ò:U		ÓUÓS%!Ó	S(Ó(S9Ó3	TÓ<TÔ	TÔT ÔT#Ô#T3Ô.T6Ô6UÕU		Õ	UÕUÕU-	c                   ó0   a € ] tR tRt o RtV 3R ltRtV tR# )Ú
ModelInputiÑ  a  
WIP type: represents a single model input
Which consists of a tuple of arguments and a set of contexts in which to run the model.

For each ModelInput, we'll compile one full graph of the model, and then use the guards generated
to dispatch between the compiled graphs.


c                ó„   <€ V ^8„  d   Qh/ S[ S[,          ;R&   S[S[S[3,          ;R&   S[S[S[,          ,          ;R&   # )r   r   r   Úcontexts)rœ   r	   r   r   r(  r   )r    r1   s   "€r!   r"   ÚModelInput.__annotate__Ñ  sC   ø‡ ‚ ñ ‘�*Ññ ñ ‘‘c��NÑñ ñ Ñ)©#Õ.Õ/Ñ/ò r#   r0   N)rG   rH   rI   rJ   Ú__doc__rM   rN   rO   rP   s   @r!   r–  r–  Ñ  s   ø‡ € ñ÷ ƒ r#   r–  c                   ól   a € ] tR tRt o V 3R lR ltV 3R lR lt]V 3R lR l4       tV 3R ltR	t	V t
R
# )ÚAOTCompiledModeliâ  c                ó,   <€ V ^8„  d   QhRS[ RS[ RS[ /# rî   r¥   )r    r1   s   "€r!   r"   ÚAOTCompiledModel.__annotate__é  s'   ø€ ÷ Eñ E™cð E©Sð E±Sñ Er#   c                óò   € V P                    FA  pVP                  ! V P                  .VO5/ VB '       g   K+  V! V P                  .VO5/ VB u # 	  V P                   ^ ,          ! V P                  .VO5/ VB # )r   )Úcompiled_resultsr  rE  )r8   r   r   r  s   &*, r!   r  ÚAOTCompiledModel.__call__é  sp   € Ø×+Ô+ˆFØ×!Ò! $§*¡*Ð>¨tÒ>°v×>Ò>Ù˜dŸj™jÐ:¨4Ò:°6Ñ:Ò:ñ ,ð ×$Ñ$ QÖ'¨¯
©
ÐD°TÒD¸VÑDÐDr#   c                ó    <€ V ^8„  d   QhRS[ /# ru   rÒ   )r    r1   s   "€r!   r"   rž  ð  s   ø€ ÷ "ñ "™5ñ "r#   c                ó´   € . pV P                    F1  pVP                  \        P                  V4      P                  4       K3  	  \
        P                  ! V4      # r%   )r   Úappendrì   r  rÑ   ÚpickleÚdumps)r8   rÖ   r  s   &  r!   r  ÚAOTCompiledModel.serializeð  sD   € ØˆØ×+Ô+ˆFØ�K‰KÔ+×5Ñ5°fÓ=×MÑMÖNñ ,ä�|Š|˜DÓ!Ð!r#   c                óR   <€ V ^8„  d   QhRS[ P                  P                  RS[RR/# )r   rE  rÖ   r   rœ  )rk   rl   rm   rD   )r    r1   s   "€r!   r"   rž  ÷  s+   ø€ ÷ ,ñ ,¡§¡§¡ð ,±uð ,ÐASñ ,r#   c           
     óœ  € ^ RI Hp ^ RIHpHp \
        P                  ! V4      p. pV Fr  pV! V! \        P                  ! / 4      4      4      ;_uu_ 4        V! 4       ;_uu_ 4        VP                  \        P                  V4      4       RRR4       RRR4       Kt  	  V ! W4      #   + '       g   i     L$; i  + '       g   i     K¢  ; i)r   )rP  )Úcompile_contextÚCompileContextN)rl  rP  rn  rª  r«  r¥  Úloadsr   Úget_compile_idr¤  rì   r=  )	r|   rE  rÖ   rP  rª  r«  Úresultsr   r  s	   &&&      r!   r=  ÚAOTCompiledModel.deserializeö  s•   € å;ßAä%Ÿ|š|¨DÓ1ˆØÐÛˆFá¡¬}×/KÒ/KÈBÓ/OÓ P×QÕQÙ#×%Õ%à ×'Ñ'Ô(;×(GÑ(GÈÓ(OÔP÷ &÷ RÑQñ ñ �5Ó+Ð+÷ &×%ú÷ R×QÐQús$   ÁB:Á&%B'ÂB:Â'B7Â2B:Â:Cc                ój   <€ V ^8„  d   Qh/ S[ P                  P                  ;R&   S[S[,          ;R&   # )r   rE  r   )rk   rl   rm   r(  rì   )r    r1   s   "€r!   r"   rž  â  s-   ø‡ ‚ ñ �8‰8�?‰?Ññ	 ñ
 Ñ.Õ/Ñ/ò r#   r0   N)rG   rH   rI   rJ   r  r  r¿   r=  rM   rN   rO   rP   s   @r!   rœ  rœ  â  s6   ø‡ € ÷Eð E÷"ð "ð ÷,ó ð,÷+ ƒ r#   rœ  c                ó  € V ^8„  d   QhR\         P                  P                  R\        \        ,          R\
        R\        \         P                  P                  \        \         P                  ,          .\        3,          R\        /# )r   rE  ÚinputsrG  rH  r   )rk   rl   rm   r(  r–  r   r   rJ  rK  rL  r   rœ  )r    s   "r!   r"   r"     sj   € ÷  5ñ  5Ü�8‰8�?‰?ð 5ä”Õð 5ô ð 5ô ”u—x‘x×+Ñ+¬T´%·,±,Õ-?Ð@ÔBVÐVÕWð	 5ô
 ñ 5r#   c                óÜ   a aa€ R VVV 3R llp. pV F0  p\         P                  RV4       VP                  V! V4      4       K2  	  \        V4      ^ 8X  d   \	        R4      h\        S V4      # )za
Compiles a single nn.Module with any number of inputs, and returns a compiled forward function.
c                ó0   € V ^8„  d   QhR\         R\        /# )r   Úmodel_inputr   )r–  rì   )r    s   "r!   r"   Ú(aot_compile_module.<locals>.__annotate__  s   € ÷ ñ ¬*ð Ô9Lñ r#   c           	      ó  <€ V P                   V P                  3pSP                  p\        4       ;_uu_ 4       pV P                   F  pVP                  V4       K  	  \        VVSSR 7      uuRRR4       #   + '       g   i     R# ; i))rG  rH  N)r   r   Úforwardr   r˜  r†  r”  )rµ  rF  Úorig_forwardr…  ÚctxrH  rG  rE  s   &    €€€r!   Úcompile_single_graphÚ0aot_compile_module.<locals>.compile_single_graph  si   ø€ Ø%×*Ñ*¨K×,>Ñ,>Ð?ˆØ—}‘}ˆÜ�[Œ[˜EØ"×+Ô+�Ø×#Ñ# CÖ(ñ ,ä(ØØØØô	÷ �[�[‹[ús   ¸3A6Á6B	zCompiling input %s..z%Expected at least one compiled result)ÚlogÚinfor¤  rô   r{   rœ  )rE  r²  rG  rH  r»  r   rµ  s   f&ff   r!   Úaot_compile_moduler¿    sl   ú€ ÷ò ð ÐÛˆÜ�‰Ð'¨Ô5Ø×ÑÑ 4°[Ó AÖBñ ô ÐÓ Ô!ÜÐDÓEÐEä˜EÐ#3Ó4Ð4r#   r%   )9rK   rŠ   r   rX   ÚloggingrÝ   r¥  rà   rE   Úcollections.abcr   r   Ú
contextlibr   r   r   r   Útypingr	   r
   r   rk   Útorch.fxÚtorch._dynamo.convert_framer   Útorch._dynamo.graph_utilsr   r’   r   Ú r   Úaot_compile_typesr   r   rG  r   Úguardsr   r	  r   r   Ú	getLoggerrG   r½  r*   r,   ÚPicklerrR   Ú	UnpicklerrÃ   rÏ   rê   rì   r”  r–  rœ  r¿  r0   r#   r!   Ú<module>rÍ     s+  ðÛ Û Û Û 	Û Û 	Û Û Û ß .ß EÑ EÝ !ß /Ñ /ã Û Ý 7Ý 8Ý ,å ÷õ ÷ Ý+ß3ð ×Ò˜Ó!€õ%ð ÷Oð Oó ðOô$Y˜Ÿ™ô Yôx'˜&×*Ñ*ô 'ð  ÷ð ó ðõ%ð ÷w*ð w*ó ðw*÷tTðn ÷0ð 0ó ð0ð  ÷ ,ð  ,ó ð ,÷F 5r#   