ó
    Eñi-H  ã                   ó„  • S SK r S SKrS SKrS SKrS SKrS SKrS SKrS SKJrJ	r	  S SK
JrJrJr  S SK Jr  S SKJrJrJr  S SK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  S	SKJrJr  S	SK J!r!  \(       a  S	SK"J#r#  S	SK$J%r%J&r&  \RN                  " \(5      r)S\RT                  S\S\S\+\,\4   4S jr-\ " S S5      5       r. " S S\R^                  5      r0 " S S\Rb                  5      r2\ " S S5      5       r3\ " S S5      5       r4 S+S\S\5\5\S 4   \+\,\4   4   S!\!S"\\Rl                  Rn                  \8\Rr                     /\4   S#\:S-  S\44S$ jjr;\ " S% S&5      5       r<\ " S' S(5      5       r=S\R|                  R~                  S)\8\<   S!\!S"\\Rl                  Rn                  \8\Rr                     /\4   S\=4
S* jr@g),é    N)ÚCallableÚSequence)ÚAbstractContextManagerÚ	ExitStackÚnullcontext)Ú	dataclass)ÚAnyÚOptionalÚTYPE_CHECKING)ÚGraphRuntimeEnv)Ú_graph_device_type)Ú
SystemInfoé   )Úconvert_frame)Ú&BundledAOTAutogradSerializableCallableÚSerializableCallable)ÚHooks)ÚGuardManagerWrapper)ÚSerializedCodeÚ
SourceInfoÚ	signatureÚargsÚkwargsÚreturnc                 ó^   • U R                   " U0 UD6nUR                  5         UR                  $ ©N)ÚbindÚapply_defaultsÚ	arguments)r   r   r   Úbound_argumentss       ÚV/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_dynamo/aot_compile.pyÚbind_localsr"   #   s/   € ð  —n’n dÐ5¨fÑ5€OØ×"Ñ"Ô$Ø×$Ñ$Ð$ó    c                   óð   • \ rS rSr% \R
                  \S'   \S   \S'   \\S'   \	\S'   \
\S'   \R                  \S'   \\S	'   S
\S'   \	\S'   \	\S'   \R                  " \R"                  S9r\\S'   SS jrSrg)ÚCompileArtifactsé+   r   r   Úguard_managerÚguards_stateÚ
backend_idÚcompiled_fnÚoriginal_codeÚruntime_envr   Úsource_infoÚdevice_typeÚbackend_name)Údefault_factoryÚsystem_infoNc                 óz   • [         R                  " 5       nUR                  U R                  U R                  5        g r   )r   ÚcurrentÚcheck_compatibilityr1   r.   )ÚselfÚcurrent_systems     r!   r4   Ú$CompileArtifacts.check_compatibility9   s,   € Ü#×+Ò+Ó-ˆØ×*Ñ*¨4×+;Ñ+;¸T×=MÑ=MÕNr#   © ©r   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__ÚinspectÚ	SignatureÚ__annotations__r
   ÚbytesÚstrr   ÚtypesÚCodeTyper   ÚdataclassesÚfieldr   r3   r1   r4   Ú__static_attributes__r8   r#   r!   r%   r%   +   sn   ‡ à× Ñ Ó ØÐ1Ñ2Ó2ØÓØƒOØ%Ó%Ø—>‘>Ó!Ø Ó ØÓØÓØÓØ)×/Ò/À
×@RÑ@RÑS€K�ÓS÷Or#   r%   c                   ó¾  ^ • \ rS rSrS\\\4   S\R                  SS4U 4S jjr	S\S\
\-  S-  4S jr\S	\S\4S
 j5       r\S\S\S\R                   4S j5       r\S\S\R$                  4S j5       r\SSS\R(                  4S j5       r\S\R(                  S\S\S\\S4   S-  S\\R.                  S4   S-  S\R0                  4S j5       rS\S\4S jrSrU =r$ )ÚAOTCompilePickleré>   Úexternal_dataÚbufr   Nc                 ó²   >• [         TU ]  U5        Xl        UR                  5        VVs0 s H  u  p4[	        U5      U_M     snnU l        0 U l        g s  snnf r   )ÚsuperÚ__init__rK   ÚitemsÚidÚid_mapÚerrors)r5   rK   rL   ÚkeyÚvalueÚ	__class__s        €r!   rO   ÚAOTCompilePickler.__init__?   sS   ø€ Ü‰Ñ˜ÔØ*Ôà-:×-@Ñ-@Ô-Bô'
Ú-B™z˜sŒBˆu‹I�sŠNÑ-Bò'
ˆŒð ˆ�ùó'
s   ªAÚobjc                 óü   • [        U5      U R                  ;   a  U R                  [        U5         $ [        U[        R                  R
                  5      (       a"  XR                  [        U5      '   [        U5      $ g r   )rQ   rR   Ú
isinstanceÚtorchÚnnÚModulerS   )r5   rX   s     r!   Úpersistent_idÚAOTCompilePickler.persistent_idG   sV   € Üˆc‹7�d—k‘kÓ!Ø—;‘;œr #›wÑ'Ð'Ü˜œUŸX™XŸ_™_×-Ñ-Ø#&�K‰Kœ˜3›Ñ Ü�c“7ˆNàr#   Úvalc                 ó\   ^• S[         4U4S jjnUR                  c   eUR                  S   $ )Nr   c                  ó   >• T $ r   r8   )r`   s   €r!   Ú_Ú+AOTCompilePickler._unpickle_cell.<locals>._R   s   ø€ ØˆJr#   r   )ÚobjectÚ__closure__)Úclsr`   rc   s    ` r!   Ú_unpickle_cellÚ AOTCompilePickler._unpickle_cellP   s0   ø€ ð	”6÷ 	ð �}‰}Ñ(Ð(Ð(Ø�}‰}˜QÑÐr#   ÚfuncÚbasec                 ó.   • [         R                  " X5      $ r   )rC   Ú
MethodType)rg   rj   rk   s      r!   Ú_unpickle_bound_methodÚ(AOTCompilePickler._unpickle_bound_methodX   s   € ô ×Ò Ó+Ð+r#   Únamec                 ó.   • [         R                  " U5      $ r   )Ú	importlibÚimport_module)rg   rp   s     r!   Ú_unpickle_moduleÚ"AOTCompilePickler._unpickle_module]   s   € ä×&Ò& tÓ,Ð,r#   Úserialized_coder   c                 ó2   • SSK Jn  UR                  " U5      $ )Nr   ©r   )Útorch._dynamo.packager   Úto_code_object)rg   rv   r   s      r!   Ú_unpickle_codeÚ AOTCompilePickler._unpickle_codea   s   € å8à×,Ò,¨_Ó=Ð=r#   ÚcodeÚmoduleÚqualnameÚargdefs.Úclosurec                 ór   • [         R                  " U5      R                  n[        R                  " XX4U5      $ r   )rr   rs   Ú__dict__rC   ÚFunctionType)rg   r}   r~   r   r€   r�   Ú	f_globalss          r!   Ú_unpickle_nested_functionÚ+AOTCompilePickler._unpickle_nested_functiong   s/   € ô ×+Ò+¨FÓ3×<Ñ<ˆ	Ü×!Ò! $°8ÀgÓNÐNr#   c                 óÒ  • [        U[        S " S5      R                  S   5      5      (       a"  [        U 5      R                  UR                  44$ [
        R                  " U5      (       a.  SSKJn  [        U 5      R                  UR                  " U5      44$ [
        R                  " U5      (       a"  [        U 5      R                  UR                  44$ [
        R                  " U5      (       ax   UR                  nUR                   n[#        XCR                  5      n[
        R                  " U5      (       a  UR                  nX5La  [        U 5      R$                  X444$  [2        $ [
        R&                  " U5      (       a^  SUR(                  ;   aN  [        U 5      R*                  UR,                  UR.                  UR(                  UR0                  UR                  44$ [2        $ )Nc                 ó   ^ • U 4S j$ )Nc                  ó   >• T $ r   r8   ©Úxs   €r!   Ú<lambda>ÚFAOTCompilePickler.reducer_override.<locals>.<lambda>.<locals>.<lambda>u   s   ø€ ±1r#   r8   r‹   s   `r!   r�   Ú4AOTCompilePickler.reducer_override.<locals>.<lambda>u   s   ø€ ­9r#   r   rx   z<locals>)rZ   Útyperf   rh   Úcell_contentsr>   Úiscodery   r   r{   Úfrom_code_objectÚismodulert   r:   ÚismethodÚ__func__Ú__self__Úgetattrrn   Ú
isfunctionr<   r†   Ú__code__r;   Ú__defaults__ÚNotImplemented)r5   rX   r   rj   Úmethod_selfÚ
inner_funcs         r!   Úreducer_overrideÚ"AOTCompilePickler.reducer_overridet   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#   )rS   rK   rR   )r:   r;   r<   r=   ÚdictrB   re   ÚioÚBytesIOrO   Úintr^   Úclassmethodrh   r   rC   rm   rn   Ú
ModuleTypert   rD   r{   ÚtupleÚCellTyper„   r†   r	   rŸ   rG   Ú__classcell__©rV   s   @r!   rI   rI   >   s  ø† ð d¨3°¨;Ñ&7ð ¸b¿j¹jð ÈT÷ ð ð ¨C°#©I¸Ñ,<ô ð ð  ð  ¨Fó  ó ð ð ð,¨(ð ,¸&ð ,ÀU×EUÑEUó ,ó ð,ð ð- Cð -¨E×,<Ñ,<ó -ó ð-ð ð>Ð-=ð >À%Ç.Á.ó >ó ð>ð
 ð	Oà�n‰nð	Oð ð	Oð ð		Oð
 �v˜s�{Ñ# dÑ*ð	Oð �u—~‘~ sÐ*Ñ+¨dÑ2ð	Oð 
×	Ñ	ó	Oó ð	Oð" Cð "¨C÷ "ò "r#   rI   c                   óh   ^ • \ rS rSrS\\\4   S\R                  S\4U 4S jjr	S\S\4S jr
SrU =r$ )	ÚAOTCompileUnpickleré™   rK   Úfiler   c                 ó0   >• [         TU ]  U5        Xl        g r   )rN   rO   rK   )r5   rK   r®   rV   s      €r!   rO   ÚAOTCompileUnpickler.__init__š   s   ø€ Ü‰Ñ˜ÔØ*Õr#   rT   c                 ót   • XR                   ;  a  [        SU SU R                    35      eU R                   U   $ )Nz-Missing required external reference to data: zS. Please load AOT compiled function with `external_data=<external data dictionary>`)rK   ÚRuntimeError)r5   rT   s     r!   Úpersistent_loadÚ#AOTCompileUnpickler.persistent_loadž   sN   € Ø×(Ñ(Ó(ÜØ?À¸uð E=ð ×%Ñ%Ð&ð(óð ð ×!Ñ! #Ñ&Ð&r#   )rK   )r:   r;   r<   r=   r¡   rB   re   r¢   r£   rO   r³   rG   r©   rª   s   @r!   r¬   r¬   ™   sB   ø† ð+ d¨3°¨;Ñ&7ð +¸r¿z¹zð +Èf÷ +ð' 3ð '¨6÷ 'ò 'r#   r¬   c                   ó    • \ rS rSr% \\S'   Srg)ÚAOTCompileSaveResulté©   Úserialized_datar8   N)r:   r;   r<   r=   rA   r@   rG   r8   r#   r!   r¶   r¶   ©   s   ‡ àÖr#   r¶   c            
       ód  • \ rS rSr% \\S'   Sr\\S'   Sr\	\
\4   S-  \S'   S\S\S	\	\
\4   4S
 jrS\S\S	\4S jrSS jrS\S\S	\4S jrSS jr SS\
S\	\
\4   S-  S	\4S jjr\ SSS S\	\
\4   S-  S	\4S jj5       r\  SS\S\	\
\4   S-  S\	\
\4   S-  S	S 4S jj5       rSS jrSrg)ÚAOTCompiledFunctioné®   Ú
_artifactsTÚ_guard_check_enabledNÚ_extra_globalsr   r   r   c                 óþ  • 0 nU R                   R                  nUR                  (       a›  UR                  R                  (       a6  [        UR                  5      [        UR                  R                  5      :X  d   e[        UR                  R                  UR                  5       VVs0 s H  u  pVXVR                  _M     nnnUR                  [        U R                   R                  /UQ70 UD65        U$ s  snnf r   )r¼   r,   r�   ÚbytecodeÚco_freevarsÚlenÚzipr‘   Úupdater"   r   )r5   r   r   Úf_localsÚenvrp   Úcells          r!   Úprepare_f_localsÚ$AOTCompiledFunction.prepare_f_locals´   sÌ   € Ø&(ˆØ�o‰o×)Ñ)ˆØ�;�;Ø—<‘<×+×+´°C·K±KÓ0@ÄCØ—‘×(Ñ(óEó 1ð ð ô
 #& c§l¡l×&>Ñ&>ÀÇÁÔ"Lôâ"L‘J�Dð ×(Ñ(Ò(Ù"Lð ñ ð 	�‰œ D§O¡O×$=Ñ$=ÐOÀÒOÈÑOÔPØˆùós   Â*C9c                 ó¢   • U R                   " U0 UD6nU R                  R                  c   eU R                  R                  R                  U5      $ r   )rÈ   r¼   r'   Úcheck)r5   r   r   rÅ   s       r!   Úguard_checkÚAOTCompiledFunction.guard_checkÂ   sH   € Ø×(Ò(¨$Ð9°&Ñ9ˆØ�‰×,Ñ,Ñ8Ð8Ð8Ø�‰×,Ñ,×2Ñ2°8Ó<Ð<r#   c                 óê  • SSK JnJn  U R                  R	                  5         U R                  R
                  R                  U R                  R                  U R                  R                  U R                  S9U l
        U R                  R                  c^  U" U R                  R                  5      nU" UU R                  R                  U R                  R                  5      U R                  l        g g )Nr   )Úload_guard_managerÚload_guards_state)Úextra_globals)ÚpackagerÏ   rÐ   r¼   r4   r,   Úforward_callabler)   r*   r¾   Úfnr'   r(   r+   Ú__globals__)r5   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                 óD  • U R                   R                  c   eU R                  (       ae  U R                  " U0 UD6(       dN  U R                  " U0 UD6n[        U R                   R                  R                  U5      5      n[        SU 35      eU R                  " U0 UD6$ )Nz#GuardManager check failed, reason: )	r¼   r'   r½   rÌ   rÈ   rB   Úcheck_verboser²   rÔ   )r5   r   r   rÅ   Úreasons        r!   Ú__call__ÚAOTCompiledFunction.__call__Ú   sŽ   € Ø�‰×,Ñ,Ñ8Ð8Ð8Ø×$×$¨T×-=Ò-=¸tÐ-NÀv×-NØ×,Ò,¨dÐ=°fÑ=ˆHÜ˜Ÿ™×6Ñ6×DÑDÀXÓNÓOˆFÜÐ!DÀVÀHÐMÓNÐNØ�wŠw˜Ð' Ñ'Ð'r#   c                 ó.   • U R                   R                  $ r   )r¼   r-   ©r5   s    r!   r-   ÚAOTCompiledFunction.source_infoâ   s   € Ø�‰×*Ñ*Ð*r#   ÚpathrK   c                 ó¾   • [        US5       n[        U 5      R                  X5      nUR                  UR                  5        UsS S S 5        $ ! , (       d  f       g = f)NÚwb)Úopenr�   Ú	serializeÚwriter¸   )r5   rà   rK   ÚfÚresults        r!   Úsave_compiled_functionÚ*AOTCompiledFunction.save_compiled_functionå   sE   € ô �$˜Ô Ü˜$“Z×)Ñ)¨$Ó>ˆFØ�G‰G�F×*Ñ*Ô+Ø÷ ××ús   �7AÁ
ArÔ   c                 ó”  • SSK Jn  UR                  R                  R	                  5       nS US'   [
        R                  " US   UR                  " US   R                  5      S9US'   US   n[        U5      R                  [        U5      R                  U5      4US'   UR                  " US   5      US'   [        R                  " 5       n[        U=(       d    0 U5      nUR                  U5        UR                   (       a0  [#        S[%        UR                   R'                  5       5       S	35      e[)        UR+                  5       S
9$ )Nr   rx   r'   r,   ©rÀ   r*   r+   z+Failed to serialize the following objects: zI
Please mark these as external data by using `external_data={'key': ...}`)r¸   )ry   r   r¼   rƒ   ÚcopyrE   Úreplacer“   rÀ   r�   Údeserialize_compile_artifactsÚserialize_compile_artifactsr¢   r£   rI   ÚdumprS   r²   ÚlistÚvaluesr¶   Úgetvalue)rg   rÔ   rK   r   Ústater*   rL   Ú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°R¸Ó=ˆØ�‰�UÔØ�>�>ÜØ=¼dÀ7Ç>Á>×CXÑCXÓCZÓ>[Ð=\ð ][ð [óð ô $°C·L±L³NÑCÐCr#   Údatar…   Úexternal_closure_datac                 ó.  • SSK Jn  [        R                  " U5      nUR	                  S5        [        U=(       d    0 U5      nUR                  5       nUR                  5         [        R                  " US   UR                  " US   R                  5      S9US'   US   u  p‰[        R                  R                  R                  SS9   U" U	5      US'   S S S 5        UR                  " US   5      US'   [!        S
0 UD6n
U " X¢S	9$ ! , (       d  f       N7= f)Nr   rx   r,   rë   r*   T)Úenable_autograd_for_aotr+   )r¾   r8   )ry   r   r¢   r£   Úseekr¬   ÚloadÚcloserE   rí   rz   rÀ   r[   Ú	_inductorÚconfigÚpatchr%   )rg   r÷   r…   rø   r   ræ   Ú	unpicklerrô   ÚdeserializerÚcompiled_fn_stateÚ	artifactss              r!   ÚdeserializeÚAOTCompiledFunction.deserialize	  sù   € õ 	9ä�JŠJ�tÓˆØ	�‰ˆqŒ	Ü'Ð(=×(CÀÀQÓGˆ	Ø—‘Ó ˆØ	�‰Œ	Ü*×2Ò2Ø�-Ñ Ø#×2Ò2°5¸Ñ3G×3PÑ3PÓQñ 
ˆˆmÑð +0°Ñ*>Ñ'ˆÜ�_‰_×#Ñ#×)Ñ)À$Ð)ÒGÙ#/Ð0AÓ#BˆE�-Ñ ÷ Hà!/×!>Ò!>¸uÀ_Ñ?UÓ!VˆˆoÑä$Ñ- uÑ-ˆ	Ù�9Ñ7Ð7÷ HÕGús   Ã	DÄ
Dc                 ó   • SU l         g )NF)r½   rÞ   s    r!   Údisable_guard_checkÚ'AOTCompiledFunction.disable_guard_check#  s
   € Ø$)ˆÕ!r#   )r½   rÔ   r9   )r   r   r   )NN)r:   r;   r<   r=   r%   r@   r½   Úboolr¾   r¡   rB   re   rÈ   r	   rÌ   rÖ   rÛ   r-   r¶   rè   r¥   rä   rA   r  r  rG   r8   r#   r!   rº   rº   ®   s`  ‡ à Ó Ø!%Ð˜$Ó%Ø/3€N�D˜˜f˜Ñ%¨Ñ,Ó3ð fð ¸ð À4ÈÈVÈÑCTô ð= ð =°ð =¸ô =ô
ð&(˜cð (¨Sð (°Sô (ô+ð AEñØðØ(,¨S°#¨X©¸Ñ(=ðà	õð àOSñDØ&ðDØ7;¸CÀ¸H±~ÈÑ7LðDà	ôDó ðDð6 ð /3Ø7;ñ	8àð8ð ˜˜V˜Ñ$ tÑ+ð8ð  $ C¨ H™~°Ñ4ð	8ð
 
ô8ó ð8÷2*r#   rº   ÚmodelÚexample_inputs.ÚhooksÚbackendÚdynamicc                 ó¢  ^• SSK Jm  SSKJn  SSKJnJn  SSKJn  Uu  pš[        5       nUb  SSK
Jn  U" U5      nU" 5          U" S5         [        R                  R                  R                  SS	9   U   [         R"                  " X	U
5      nUR$                  nUR&                  c   eUR(                  (       d+  SS
KJn  S[.        U   S[.        [0           4U4S jjnUUl        [         R2                  " U 5      u  nnUR4                  nUc   eUR6                  UR8                  l        [=        UR8                  R>                  5      nUR@                  RB                  UR&                  RB                  L d   eU" UR@                  5      nURD                  Ul"        [        RF                  RI                  U5         [        R                  R                  R                  SSSSSS.5         U" UR8                  URJ                  5      n[M        U[        RN                  5      (       dX  [Q        US5      (       aR  [M        URR                  [        RT                  RV                  RX                  RZ                  5      (       a  []        U5      nS S S 5        S S S 5        [M        W[^        5      (       d?  [Q        US5      (       a  URR                  nOUn[a        S[c        U5       S3SU S3-   5      eURe                  URf                  USSS9nURh                  c   eU" [k        5       S9nURl                   H  nURo                  U5        M     [q        [         Rr                  " U5      URt                  URh                  UR6                  UURf                  URw                  5       UU[y        USS5      S9
n[{        UUR|                  S9nS S S 5        S S S 5        S S S 5        S S S 5        W$ ! , (       d  f       GNe= f! , (       d  f       GNo= f! , (       d  f       NL= f! , (       d  f       NU= f! , (       d  f       N^= f! , (       d  f       W$ = f)Nr   )ÚCheckFunctionManager)r   )Údynamo_timedÚget_metrics_context)ÚTracingContext)Úset_enable_dynamicÚfullgraph_captureT)Ústrict_autograd_cache)ÚGuardFilterEntryÚguard_entriesr   c                 ó’   >• U  Vs/ s H4  nUR                   =(       d    UR                  TR                  ;   (       + PM6     sn$ s  snf r   )Ú	is_globalÚ
guard_typeÚ%UNSUPPORTED_SERIALIZATION_GUARD_TYPES)r  Úgr  s     €r!   Únew_guard_filter_fnÚ2aot_compile_fullgraph.<locals>.new_guard_filter_fnH  sS   ø€ ñ +ó	ò +˜ð ŸK™K÷ ZØ Ÿ|™|Ø3×YÑYñ Z÷ñ +ñ	ð 	ùò 	s   †;A)r  Úbypass_autograd_cache_keyÚbundled_autograd_cacheÚ force_non_lazy_backward_loweringÚforce_autograd_cacheÚcompiler_fnzCompiled function type z (produced zfrom backend z*) does not implement SerializableCallable.)r  ÚsaveÚstrict_error)Úinlined_sourcesÚcompiler_nameÚunknown)
r   r'   r(   r)   r*   r+   r,   r-   r.   r/   )r¼   r¾   )?Útorch._dynamo.guardsr  ry   r   Útorch._dynamo.utilsr  r  Útorch._guardsr  r   Útorch._dynamo.eval_framer  r[   Ú
_functorchrÿ   r   r   r  Úgraph_capture_outputÚoutput_graphÚguard_filter_fnÚtorch._dynamo.typesr  r   r
  Úget_traced_fnÚbackend_inputr)   Úgraph_moduleÚ_backend_idr   ÚgraphÚ	fake_modeÚ	shape_envÚtensor_to_contextÚ_guardsÚtracingr  rZ   Ú_TorchCompileInductorWrapperÚhasattrr%  Ú_dynamoÚbackendsÚcommonÚAotAutogradr   r   r²   r�   Úbuild_guardsrš   r(   ÚsetÚtraced_codeÚadd_coder%   Ú_get_signaturer'   Úget_runtime_envr˜   rº   rÕ   )r  r  r  r  r  r   r  r  r  r   r   Údynamic_ctxr  Úcapture_outputr0  r  r  rÔ   rc   r5  r.   Útracing_contextr*   r%  Úcheck_fnr-   rF  r  Úaot_compiled_fnr  s                                @r!   Úaot_compile_fullgraphrO  '  sÃ  ø€ õ :Ý0ßEÝ,à!�L€Dä“-€KØÑÝ?á(¨Ó1ˆñ 	ÕÙÐ(Õ)Ü×Ñ×Ñ×%Ñ%¸DÐ%ÒAÚä&×8Ò8¸ÀfÓMˆØ-×BÑBÐØ#×0Ñ0Ñ<Ð<Ð<à×$×$Ý<ðÜ'Ð(8Ñ9ðäœ$‘÷ð %8ˆEÔ!ä×+Ò+¨EÓ2‰ˆˆAà&×4Ñ4ˆØÑ(Ð(Ð(Ø1>×1IÑ1Iˆ×"Ñ"Ô.Ü(¨×)CÑ)C×)IÑ)IÓJˆà×#Ñ#×-Ñ-Ø#×0Ñ0×:Ñ:ò;ð	
ð;ñ )¨×)@Ñ)@ÓAˆØ,9×,KÑ,KˆÔ)ä�M‰M×!Ñ! /Õ2Ü×Ñ×#Ñ#×)Ñ)à-1Ø15Ø.2Ø8<Ø,0ñõñ "Ø×*Ñ*¨M×,HÑ,HóˆKô
 ˜'¤5×#EÑ#E×FÑFÜ˜ ×/Ñ/ÜØ×'Ñ'¬¯©×)?Ñ)?×)FÑ)F×)RÑ)R÷ñ ô EÀ[ÓQ�÷+÷ 3ô0 ˜+Ô';×<Ñ<Ü�w ×.Ñ.Ø%×1Ñ1‘à%�ÜØ)¬$¨{Ó*;Ð)<¸KÐHØ! + Ð.XÐYñZóð ð
 (×4Ñ4Ø�K‰K˜u¨4¸dð 5ð 
ˆð ×$Ñ$Ñ0Ð0Ð0á ´³Ñ7ˆØ/×;Ô;ˆKØ× Ñ  Ö-ñ <ô %Ü#×2Ò2°2Ó6Ø"×0Ñ0Ø!×.Ñ.Ø$×/Ñ/Ø#ØŸ+™+Ø,×<Ñ<Ó>Ø#Ø#Ü  ¨/¸9ÓEñ
ˆ	ô .Ø °·±ñ
ˆ÷} 	÷ 	B÷ 	*÷ 	ðJ Ð÷wö ú÷ 3Ö2ú÷M 	�ú÷ 	BÕAú÷ 	*Õ)ú÷ 	ÔðJ Ðús‡   Á	P?Á(P.Á3PÁ6EP	Æ:0O:Ç*B O(Ê
O:ÊD4P	ÏPÏP.ÏP?Ï(
O7Ï2O:Ï:
P	ÐP	Ð
PÐPÐ
P+Ð'P.Ð.
P<	Ð8P?Ð?
Qc                   óT   • \ rS rSr% Sr\\   \S'   \\	\4   \S'   \
\\      \S'   Srg)Ú
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.


r   r   Úcontextsr8   N)r:   r;   r<   r=   Ú__doc__r§   r	   r@   r¡   rB   rñ   r   rG   r8   r#   r!   rQ  rQ  ¤  s2   ‡ ñð �‰*ÓØ��c�‰NÓØÐ)¨#Ñ.Ñ/Ö/r#   rQ  c                   óÄ   • \ rS rSr% \R
                  R                  \S'   \\	   \S'   S\
S\
S\
4S jrS\4S jr\S\R
                  R                  S	\SS 4S
 j5       rSrg)ÚAOTCompiledModeliµ  r  Úcompiled_resultsr   r   r   c                 óè   • U R                    H@  nUR                  " U R                  /UQ70 UD6(       d  M*  U" U R                  /UQ70 UD6s  $    U R                   S   " U R                  /UQ70 UD6$ )Nr   )rV  rÌ   r  )r5   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                 ó´   • / nU R                    H1  nUR                  [        R                  U5      R                  5        M3     [
        R                  " U5      $ r   )rV  Úappendrº   rä   r¸   ÚpickleÚdumps)r5   r÷   rç   s      r!   rä   ÚAOTCompiledModel.serializeÃ  sD   € ØˆØ×+Ô+ˆFØ�K‰KÔ+×5Ñ5°fÓ=×MÑMÖNñ ,ä�|Š|˜DÓ!Ð!r#   r÷   c                 ót  • SSK Jn  SSKJnJn  [
        R                  " U5      n/ nU Hb  nU" U" [        R                  " 0 5      5      5         U" 5          UR                  [        R                  U5      5        S S S 5        S S S 5        Md     U " X5      $ ! , (       d  f       N"= f! , (       d  f       MŽ  = f)Nr   )r  )Úcompile_contextÚCompileContext)r,  r  r-  r_  r`  r[  Úloadsr   Úget_compile_idrZ  rº   r  )	rg   r  r÷   r  r_  r`  ÚresultsrV  rç   s	            r!   r  ÚAOTCompiledModel.deserializeÉ  sŒ   € å;ßAä%Ÿ|š|¨DÓ1ˆØÐÛˆFá¡¬}×/KÒ/KÈBÓ/OÓ PÕQÙ#Õ%à ×'Ñ'Ô(;×(GÑ(GÈÓ(OÔP÷ &÷ RÑQñ ñ �5Ó+Ð+÷ &Õ%ú÷ RÖQús$   ÁB(Á%BÁ;B(Â
B%Â!B(Â(
B7	r8   N)r:   r;   r<   r=   r[   r\   r]   r@   rñ   rº   r	   rÛ   rA   rä   r¥   r  rG   r8   r#   r!   rU  rU  µ  sz   ‡ ð �8‰8�?‰?ÓØÐ.Ñ/Ó/ðE˜cð E¨Sð E°Sô Eð"˜5ô "ð ð, §¡§¡ð ,°uð ,ÐASó ,ó ó,r#   rU  Úinputsc                 óÞ   ^ ^^• S[         S[        4UUU 4S jjn/ nU H0  n[        R                  SU5        UR	                  U" U5      5        M2     [        U5      S:”  d   e[        T U5      $ )za
Compiles a single nn.Module with any number of inputs, and returns a compiled forward function.
Úmodel_inputr   c                 óö   >• U R                   U R                  4nTR                  n[        5        nU R                   H  nUR                  U5        M     [        UUTTS9sS S S 5        $ ! , (       d  f       g = f)N)r  r  )r   r   Úforwardr   rR  Úenter_contextrO  )rg  r  Úorig_forwardÚstackÚctxr  r  r  s        €€€r!   Úcompile_single_graphÚ0aot_compile_module.<locals>.compile_single_graphã  se   ø€ Ø%×*Ñ*¨K×,>Ñ,>Ð?ˆØ—}‘}ˆÜŒ[˜EØ"×+Ô+�Ø×#Ñ# CÖ(ñ ,ä(ØØØØñ	÷ �[�[ús   °0A*Á*
A8zCompiling input %s..r   )rQ  rº   ÚlogÚinforZ  rÂ   rU  )r  re  r  r  rn  rV  rg  s   ` ``   r!   Úaot_compile_modulerr  Ù  su   ú€ ð¬*ð Ô9L÷ ñ ð ÐÛˆÜ�‰Ð'¨Ô5Ø×ÑÑ 4°[Ó AÖBñ ô ÐÓ  1Ó$Ð$Ð$ä˜EÐ#3Ó4Ð4r#   r   )ArE   rr   r>   r¢   Úloggingr[  rC   Úcollections.abcr   r   Ú
contextlibr   r   r   r   Útypingr	   r
   r   r[   Útorch.fxÚtorch._dynamo.convert_framer   Útorch._dynamo.graph_utilsr   ry   r   Ú r   Úaot_compile_typesr   r   r  r   Úguardsr   rÒ   r   r   Ú	getLoggerr:   rp  r?   r¡   rB   r"   r%   ÚPicklerrI   Ú	Unpicklerr¬   r¶   rº   r§   ÚfxÚGraphModulerñ   ÚTensorr
  rO  rQ  rU  r\   r]   rr  r8   r#   r!   Ú<module>rƒ     s0  ðÛ Û Û Û 	Û Û Û ß .ß EÑ EÝ !ß /Ñ /ã Û Ý 7Ý 8Ý ,å ÷õ ö Ý+ß3ð ×Ò˜Ó!€ð%Ø× Ñ ð%Ø),ð%Ø8;ð%à	ˆ#ˆsˆ(�^ô%ð ÷Oð Oó ðOô$X˜Ÿ™ô Xôv'˜&×*Ñ*ô 'ð  ÷ð ó ðð ÷u*ð u*ó ðu*ðz  ñzØðzà˜%  S ™/¨4°°S°©>Ð9Ñ:ðzð ðzð �u—x‘x×+Ñ+¨T°%·,±,Ñ-?Ð@ÐBVÐVÑWð	zð
 �D‰[ðzð õzðz ÷0ð 0ó ð0ð  ÷ ,ð  ,ó ð ,ðF5Ø�8‰8�?‰?ð5à�Ñð5ð ð5ð �u—x‘x×+Ñ+¨T°%·,±,Ñ-?Ð@ÐBVÐVÑWð	5ð
 õ5r#   