ó
    Eñi“J  ã                   ó>  • % S SK r S SKrS SKrS SKJr  S SKJr  S SKJrJrJ	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  S S
KJr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!  S SK"J#r#J$r$  S SK%J&r&J'r'  S SK(J)r*  S/r+\r,\r-\S\4   r.\\
R                  R^                  /\\$   4   r0\
Rb                  \
Rd                  \
Rf                  \
Rh                  \
Rj                  \
Rl                  \
Rn                  1r8\9\   \:S'    " S S\;5      r< " S S\#5      r=g)é    N)ÚCallable)Únullcontext)ÚAnyÚOptionalÚUnion)Úfx)Úenable_python_dispatcher)ÚNodeMetadata)Ú
ProxyValue)Ú_unstack_pytree)Ú
FakeTensorÚUnsupportedFakeTensorException)ÚFakeTensorMode)Ú	traceback)ÚPythonKeyTracer)Úcompute_unbacked_bindingsÚPropagateUnbackedSymInts)ÚCodeGen)ÚPassBaseÚ
PassResult)Ú_extract_tensor_metadataÚTensorMetadata)Ú_pytreeÚ!_ExportPassBaseDeprecatedDoNotUse.Ú_TORCH_SYM_OPSc                   ó   • \ rS rSrSrg)ÚExportPassBaseErroré1   © N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__static_attributes__r   ó    ÚT/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_export/pass_base.pyr   r   1   s   † Úr%   r   c                   óü  • \ rS rSrSr\S 5       r " S S\5      r " S S\	R                  5      rS,S
 jrS\S\R                  R                  R                   S\\S4   S\\\4   S\S\4S jrS\R                  R.                  S\\   4S jrS\SS	4S jrS\S\S\S\4S jrS\\S4   S\\\4   S\S\4S jrS\S\\S4   S\S\4S jrS\S\R                  R.                  S\R                  R.                  S\\   S\S\4S jrS \R                  R.                  S!\\   S"\\   S\S\4
S# jr S$\S%\!S\S\4S& jr"S'\\   S\S\4S( jr#S\	R.                  S\\S4   S\$4S) jr%S\	R.                  S\$4S* jr&S+r'g	)-r   é5   z`
Interpreter-based pass class to help users maintain the IR spec while writing
transformations.
c            	      ó^   • [        SSR                  [        R                  " SS95      05      $ )NÚstack_traceÚ é   )Úlimit)r
   Újoinr   Úformat_stackr   r%   r&   Ú_create_dummy_node_metadataÚ=_ExportPassBaseDeprecatedDoNotUse._create_dummy_node_metadata;   s'   € ä˜]¨B¯G©G´I×4JÒ4JÐQRÑ4SÓ,TÐUÓVÐVr%   c                   óÀ   ^ • \ rS rSrSSS\SS4U 4S jjrSS jrS	\S\R                  R                  4U 4S
 jjrS\R                  R                  S\SS4S jrSrU =r$ )Ú._ExportPassBaseDeprecatedDoNotUse.ExportTraceré?   Úcallbackr   ÚcodegenÚreturnNc                 ó  >• [         TU ]  5         Xl        [        R                  R                  5       U l        [        R                  R                  5       U l	        U R                  R                  U5        0 U l        S U l        0 U l        g ©N)ÚsuperÚ__init__r5   ÚtorchÚnnÚModuleÚrootr   ÚGraphÚgraphÚset_codegenÚtensor_attrsÚfake_tensor_modeÚ
submodules)Úselfr5   r6   Ú	__class__s      €r&   r;   Ú7_ExportPassBaseDeprecatedDoNotUse.ExportTracer.__init__@   sb   ø€ ô ‰GÑÔØ$ŒMÜŸ™Ÿ™Ó)ˆDŒIÜŸ™Ÿ™Ó)ˆDŒJØ�J‰J×"Ñ" 7Ô+Ø9;ˆDÔØ>BˆDÔ!Ø:<ˆD�Or%   c                 ó   • [        S5      e)Nz%ExportTracer doesn't support trace().©r   ©rF   s    r&   ÚtraceÚ4_ExportPassBaseDeprecatedDoNotUse.ExportTracer.traceL   s   € Ü%Ð&MÓNÐNr%   Úac                 óÐ  >• [        U[        R                  R                  5      (       aQ  XR                  ;  aA  S[        U R                  5       3nU R                  R                  X!5        X R                  U'   ON[        U[        5      (       a9  [        US5      (       a  UR                  c  [        SU S35      eUR                  n[        TU ]5  U5      n[        U[        R                  5      (       an  [        U[        R                  R                   5      (       aE  UR"                  S:X  a5  U R%                  X15        U R&                  R)                  [+        X5      5        U$ )NÚ
submodule_ÚconstantzCannot add z
 to graph.Úget_attr)Ú
isinstancer<   r=   r>   rE   Úlenr?   Ú
add_moduler   ÚhasattrrQ   r   r:   Ú
create_argÚTensorr   ÚNodeÚopÚset_metadatar5   Úon_attrr   )rF   rN   Úname_submoduleÚnoderG   s       €r&   rW   Ú9_ExportPassBaseDeprecatedDoNotUse.ExportTracer.create_argO   s  ø€ Ü˜!œUŸX™XŸ_™_×-Ñ-ØŸO™OÓ+Ø'1´#°d·o±oÓ2FÐ1GÐ%H�NØ—I‘I×(Ñ(¨Ô;Ø)7—O‘O AÑ&øÜ˜Aœz×*Ñ*Ü˜q *×-Ñ-°·±Ñ1CÜ-°¸A¸3¸jÐ.IÓJÐJØ—J‘J�Ü‘7Ñ% aÓ(ˆDä˜1œeŸl™l×+Ñ+Ü˜t¤U§X¡X§]¡]×3Ñ3Ø—G‘G˜zÓ)à×!Ñ! $Ô*Ø—‘×%Ñ%¤j°Ó&9Ô:ØˆKr%   r^   Úvaluec                 ó„  ^ • S[         S[        [        [        R                  [        R
                  [        R                  [        [        [        [        S 4	   4U 4S jjn[        R                  " X25      UR                  S'   S[         S[        [           4U 4S jjn[        R                  " XB5      UR                  S'   g )NÚxr7   c           
      óœ  >• [        U [        5      (       a  U $ [        U [        R                  5      (       a£  U R                  (       a  [        R
                  " U 5      n  TR                  c  [        S5      e[        U [        R                  R                  5      (       a  TR                  R                  U SS9nU$ TR                  R                  U 5      n U$ [        U [        R                  [        R                  [        R                  [         ["        [$        [&        45      (       a  U $ g ! [         a    [        S5        S n U$ f = f)Nú!fake_tensor_mode must not be NoneT)Ústatic_shapeszvFakeifying a Tensor subclass is not supported                             right now. Instead a TensorMetadata is used.)rS   r   r<   rX   Úis_quantizedÚ
dequantizerD   ÚAssertionErrorr=   Ú	ParameterÚfrom_tensorr   ÚprintÚSymIntÚSymFloatÚSymBoolÚintÚfloatÚboolÚstr)rb   Úfake_tensorrF   s     €r&   Úmake_valÚM_ExportPassBaseDeprecatedDoNotUse.ExportTracer.set_metadata.<locals>.make_vali   s,  ø€ ô ˜a¤×,Ñ,Ø�HÜ ¤5§<¡<×0Ñ0Ø—~—~ä!×,Ò,¨QÓ/˜ð+Ø×0Ñ0Ñ8Ü"0Ð1TÓ"UÐUô & a¬¯©×);Ñ);×<Ñ<Ø*.×*?Ñ*?×*KÑ*KØ !°ð +Lð +˜Kð 'Ð&ð +/×*?Ñ*?×*KÑ*KÈAÓ*N™Kð 'Ð&ÜØäŸ™ÜŸ™ÜŸ™ÜÜÜÜð÷ñ ð �Hàøô/ :ó +ô ðJôð '+™Ø&Ð&ð+ús   Á AD0 Â=D0 Ä0EÅ
EÚvalc                 ód  >• [        U [        5      (       d~  [        U [        R                  5      (       a_  U R                  (       a  [        R
                  " U 5      n  TR                  c  [        S5      eTR                  R                  U 5      nS nU$ g ! [         a    [        U 5      n U$ f = f)Nrd   )rS   r   r<   rX   rf   rg   rD   rh   rj   r   r   )rb   Ú_Útensor_metarF   s      €r&   Úmake_tensor_metaÚU_ExportPassBaseDeprecatedDoNotUse.ExportTracer.set_metadata.<locals>.make_tensor_meta¤   s    ø€ Ü! !¤Z×0Ñ0´ZÀÄ5Ç<Á<×5PÑ5PØ—~—~ä!×,Ò,¨QÓ/˜ðBØ×0Ñ0Ñ8Ü"0Ð1TÓ"UÐUØ ×1Ñ1×=Ñ=¸aÓ@˜Ø&*˜ð
 'Ð&àøô :ó Bô '?¸qÓ&A™Ø&Ð&ð	Bús   Á5B ÂB/Â.B/ry   )ÚArgumentr   r   r<   rl   rm   rn   ro   rp   rq   rr   ÚpytreeÚtree_mapÚmetar   r   )rF   r^   r`   rt   rz   s   `    r&   r[   Ú;_ExportPassBaseDeprecatedDoNotUse.ExportTracer.set_metadatac   s•   ø€ ð6 Üð6 äÜÜ—‘Ü—‘Ü—‘ÜÜÜÜØðñ
÷6 ôp  &Ÿš¨xÓ?ˆD�I‰I�eÑð ¤Hð  ´¼.Ñ1I÷  ô& (.§¢Ð7GÓ'OˆD�I‰I�mÒ$r%   )r5   rD   rA   r?   rE   rC   ©r7   N)r    r!   r"   r#   r   r;   rL   r|   r<   r   rY   rW   r[   r$   Ú__classcell__©rG   s   @r&   ÚExportTracerr3   ?   sy   ø† ð
	=Ø?ð
	=ØJQð
	=à÷
	=ô	Oð	 ð 	¨U¯X©X¯]©]÷ 	ð(T	Pà—(‘(—-‘-ðT	Pð ðT	Pð ÷	T	Pò T	Pr%   r„   c                   ó`  ^ • \ rS rSrSSS\R
                  SS4U 4S jjrS\S	\\	S
4   S\
\\	4   S\4U 4S jjrS\R                  R                  R                  S	\\	S
4   S\
\\	4   S\4S jrS\R                  R                  R                  S	\\	S
4   S\
\\	4   S\4S jrS\S	\\	S
4   S\
\\	4   S\	4U 4S jjrS\R                  R                  R                  S	\\	S
4   S\
\\	4   SS4S jrS\S	\\	S
4   S\
\\	4   SS4S jrS\R                  R*                  S\	4U 4S jjrSrU =r$ )Ú3_ExportPassBaseDeprecatedDoNotUse.ExportInterpreteré¹   r5   r   Úgmr7   Nc                 óŠ   >• [         TU ]  U5        Xl        [        [	        UR
                  R                  5      5      U l        g r9   )r:   r;   r5   ÚnextÚiterrA   Únodesr^   )rF   r5   rˆ   rG   s      €r&   r;   Ú<_ExportPassBaseDeprecatedDoNotUse.ExportInterpreter.__init__º   s0   ø€ ô ‰GÑ˜RÔ Ø$ŒMÜ'+¬D°·±·±Ó,@Ó'AˆD�Ir%   ÚtargetÚargs.Úkwargsc                 ó–   >• [         TU ]  XU5      nU R                  R                  X[        U R                  R
                  5      5      $ r9   )r:   Úplaceholderr5   r
   r^   r   )rF   rŽ   r�   r�   ÚargrG   s        €r&   r’   Ú?_ExportPassBaseDeprecatedDoNotUse.ExportInterpreter.placeholderÂ   s:   ø€ ô ‘'Ñ% f°FÓ;ˆCØ—=‘=×,Ñ,¨V¼,ÀtÇyÁyÇ~Á~Ó:VÓWÐWr%   c                 óŽ   • U R                   R                  US   [        U R                  R                  5      5      R
                  $ )Nr   )r5   Úoutputr
   r^   r   Údata©rF   rŽ   r�   r�   s       r&   r–   Ú:_ExportPassBaseDeprecatedDoNotUse.ExportInterpreter.outputË   s2   € ð —=‘=×'Ñ'¨¨Q©´¸d¿i¹i¿n¹nÓ1MÓN×SÑSÐSr%   c                 óT  • [        U R                  R                  5      nU[        R                  L a   Uu  pVU R
                  R                  XVU5      $ [        USS 5      S;   a:  [        U5      (       d  [        SU 35      eU R
                  R                  XU5      $ U[        ;   a:  [        U5      (       d  [        SU 35      eU R
                  R                  XU5      $ [        U[        R                  R                  [        R                  R                   45      (       a  U R
                  R#                  UUUU5      $ U[        R$                  R&                  R(                  L a"  Uu  pxpšU R
                  R+                  XxXšU5      $ U[        R$                  R&                  R,                  L a!  Uu  p¼nU R
                  R/                  X¼XÔ5      $ [        U[        R                  R0                  5      (       a  U R
                  R3                  SUUUU5      $ [5        SU 35      e)Nr!   >   ÚmathÚbuiltinsÚ	_operatorzexpected callable target, got Úcall_functionzUnsupported target type: )r
   r^   r   ÚoperatorÚgetitemr5   Úcall_getitemÚgetattrÚcallablerh   Úcall_symr   rS   r<   Ú_opsÚ
OpOverloadÚOpOverloadPacketÚcall_operatorÚopsÚhigher_orderÚcondÚ	call_condÚmap_implÚcall_mapÚHigherOrderOperatorÚ_fxr   )rF   rŽ   r�   r�   r   r`   ÚkeyÚpredÚtrue_fnÚfalse_fnÚinputsÚfÚmapped_argsÚoperandss                 r&   rž   ÚA_ExportPassBaseDeprecatedDoNotUse.ExportInterpreter.call_functionÓ   så  € ô   §	¡	§¡Ó/ˆDàœ×)Ñ)Ò)Ø!‘
�Ø—}‘}×1Ñ1°%¸dÓCÐCÜ˜ ¨tÓ4ð 9ó ô
   ×'Ñ'Ü(Ð+IÈ&ÈÐ)RÓSÐSØ—}‘}×-Ñ-¨f¸DÓAÐAØœ>Ó)Ü ×'Ñ'Ü(Ð+IÈ&ÈÐ)RÓSÐSØ—}‘}×-Ñ-¨f¸DÓAÐAÜØœŸ™×.Ñ.´·
±
×0KÑ0KÐL÷ñ ð —}‘}×2Ñ2ØØØØó	ð ð œ5Ÿ9™9×1Ñ1×6Ñ6Ò6Ø26Ñ/�˜xØ—}‘}×.Ñ.¨t¸hÐPTÓUÐUØœ5Ÿ9™9×1Ñ1×:Ñ:Ò:Ø+/Ñ(� Ø—}‘}×-Ñ-¨a¸hÓMÐMä˜F¤E§J¡J×$BÑ$B×CÑCØ—}‘}×(Ñ(Ø#ØØØØóð ô *Ð,EÀfÀXÐ*NÓOÐOr%   c                 ó$   >• [         TU ]  XU5      $ r9   )r:   rR   )rF   rŽ   r�   r�   rG   s       €r&   rR   Ú<_ExportPassBaseDeprecatedDoNotUse.ExportInterpreter.get_attr  s   ø€ ô ‘7Ñ# F°&Ó9Ð9r%   c                 ó   • [        S5      e)Nzcall_module is not supported.rJ   r˜   s       r&   Úcall_moduleÚ?_ExportPassBaseDeprecatedDoNotUse.ExportInterpreter.call_module  ó   € ô &Ð&EÓFÐFr%   c                 ó   • [        S5      e)Nzcall_method is not supported.rJ   r˜   s       r&   Úcall_methodÚ?_ExportPassBaseDeprecatedDoNotUse.ExportInterpreter.call_method  r¿   r%   Únc                 ól   >• Xl         UR                  5       U R                  l        [        TU ]  U5      $ r9   )r^   Úformat_noder5   Únode_debug_strr:   Úrun_node)rF   rÃ   rG   s     €r&   rÇ   Ú<_ExportPassBaseDeprecatedDoNotUse.ExportInterpreter.run_node  s*   ø€ ØŒIØ+,¯=©=«?ˆD�M‰MÔ(Ü‘7Ñ# AÓ&Ð&r%   )r5   r^   )r    r!   r"   r#   r   ÚGraphModuler;   rr   Útupler|   Údictr   r’   r<   r^   ÚTargetr–   rž   rR   r½   rÁ   rY   rÇ   r$   r‚   rƒ   s   @r&   ÚExportInterpreterr†   ¹   sú  ø† ð	BØ?ð	BØEGÇ^Á^ð	Bà÷	Bð	Xàð	Xð ˜ #˜Ñ&ð	Xð ˜˜h˜Ñ'ð		Xð
 ÷	Xð	Tà—H‘H—M‘M×(Ñ(ð	Tð ˜ #˜Ñ&ð	Tð ˜˜h˜Ñ'ð		Tð
 ô	Tð0	Pà—H‘H—M‘M×(Ñ(ð0	Pð ˜ #˜Ñ&ð0	Pð ˜˜h˜Ñ'ð	0	Pð
 ô0	Pðd	:àð	:ð ˜ #˜Ñ&ð	:ð ˜˜h˜Ñ'ð		:ð
 ÷	:ð	Gà—H‘H—M‘M×(Ñ(ð	Gð ˜ #˜Ñ&ð	Gð ˜˜h˜Ñ'ð		Gð
 ô	Gð	Gàð	Gð ˜ #˜Ñ&ð	Gð ˜˜h˜Ñ'ð		Gð
 ô	Gð	'˜eŸh™hŸm™mð 	'°÷ 	'õ 	'r%   rÍ   r7   Nc                 ó8  • [        [        R                  R                  [        R                  R                  5       [        R                  R                  5       5      5      U l        U R                  U [        5       5      U l
        S U l        SU l        S U l        g )NT)r   r<   r   rÉ   r=   r>   r@   Úinterpreterr„   r   ÚtracerrD   Ú_initializedrÆ   rK   s    r&   r;   Ú*_ExportPassBaseDeprecatedDoNotUse.__init__"  si   € Ü3Ü�H‰H× Ñ ¤§¡§¡Ó!2´E·H±H·N±NÓ4DÓEó
ˆÔð ×'Ñ'¨¬g«iÓ8ˆŒØ:>ˆÔØ ˆÔØ48ˆÕr%   ÚkindrŽ   r�   .r�   r   c                 ó*  • [         R                  " [        S X445      u  pg[        U R                  U5      " X&U5      n[         R                  " [        S X445      u  pšS n[        U[        R                  R                  5      (       a9  U R                  R                  R                  UR                  R                  5      nU R                  R                  XXšUS9nUR                  R                   R#                  UR$                  5        U R&                  (       aG  U R&                  R(                  =n(       a*  [+        XØ5      =n(       a  XìR                  R                   S'   U R                  R-                  UR                  U5        [        XŒ5      $ )Nc                 ó   • U R                   $ r9   ©r—   ©rb   s    r&   Ú<lambda>Ú7_ExportPassBaseDeprecatedDoNotUse._fx.<locals>.<lambda>4  s   €  !§&¢&r%   c                 ó   • U R                   $ r9   )Úproxyr×   s    r&   rØ   rÙ   8  s   €  !§'¢'r%   )ÚnameÚunbacked_bindings)r}   Útree_map_onlyr   r¢   rÏ   rS   r<   r¥   r¦   rÐ   rA   Ú_target_to_strÚoverloadpacketr    Úcreate_proxyr^   r   Úupdater—   rD   Ú	shape_envr   r[   )rF   rÓ   rŽ   r�   r�   r   Ú	args_dataÚkwargs_dataÚres_dataÚ
args_proxyÚkwargs_proxyrÜ   Ú	res_proxyrã   Úsymbol_to_paths                  r&   r°   Ú%_ExportPassBaseDeprecatedDoNotUse._fx+  s@  € ô "(×!5Ò!5ÜÑ(¨4¨.ó"
Ñˆ	ô ˜4×+Ñ+¨TÔ2°6ÀkÓRˆÜ#)×#7Ò#7ÜÑ)¨D¨>ó$
Ñ ˆ
ð ˆÜ�fœeŸj™j×3Ñ3×4Ñ4Ø—;‘;×$Ñ$×3Ñ3°F×4IÑ4I×4RÑ4RÓSˆDà—K‘K×,Ñ,Ø˜*¸ð -ð 
ˆ	ð 	�‰×Ñ×"Ñ" 4§9¡9Ô-Ø× × °4×3HÑ3H×3RÑ3RÐ&R iÕ&RÜ!:¸9Ó!OÐOˆ~ÕOØ;I—‘×#Ñ#Ð$7Ñ8Ø�‰× Ñ  §¡°Ô:Ü˜(Ó.Ð.r%   Úgraph_modulec                 óB  ^ ^• TR                   R                  SS 5      =nb  [        U5      $ S[        R                  R
                  S[        [           4UU 4S jjnTR                  R                   Vs/ s H  nUR                  S:X  d  M  U" U5      PM     sn$ s  snf )Nr�   r^   r7   c                 ó2  >• SU R                   ;   a;  U R                   S   n[        US5      (       a  UR                  b  UR                  $ U$ U R                   R                  S5      =n(       a�  TR                  c  [        S5      e[        TR                  [        R                  " UR                  UR                  SUR                  UR                  S9[        R                  " S5      5      $ [        U R                  5      S:X  a  g [!        S	T S
35      e)Nrv   rQ   ry   rd   r   )ÚdtypeÚdeviceÚrequires_gradÚmemory_formatÚcpur   z,Cannot construct an input for graph module: Ú.)r   rV   rQ   ÚgetrD   rh   r   r<   ÚemptyÚshaperï   rñ   rò   rð   rT   Úusersr   )r^   Úfakery   rì   rF   s      €€r&   Úextract_inputÚ?_ExportPassBaseDeprecatedDoNotUse.inputs.<locals>.extract_inputO  s÷   ø€ Ø˜Ÿ	™	Ó!Ø—y‘y Ñ'�Ü˜4 ×,Ñ,°·±Ñ1JØŸ=™=Ð(Ø�Ø $§	¡	§¡¨mÓ <Ð<�Õ<Ø×(Ñ(Ñ0Ü(Ð)LÓMÐMÜ!Ø×)Ñ)Ü—K’KØ#×)Ñ)Ø)×/Ñ/Ø%Ø&1×&?Ñ&?Ø&1×&?Ñ&?ñô —L’L Ó'ó
ð 
ô �T—Z‘Z“ AÓ%ØÜ%Ø>¸|¸nÈAÐNóð r%   r’   )r   rõ   Úlistr<   r   rY   r   r   rA   rŒ   rZ   )rF   rì   r�   rú   r^   s   ``   r&   rµ   Ú(_ExportPassBaseDeprecatedDoNotUse.inputsI  s˜   ù€ ð !×%Ñ%×)Ñ)¨&°$Ó7Ð7ˆDÑDÜ˜“:Ðð	¤§¡§¡ð 	´(¼:Ñ2F÷ 	ð 	ð8 %×*Ñ*×0Ò0ó
â0�Ø�w‰w˜-Ñ'ó  ‰M˜$ÖÙ0ñ
ð 	
ùò 
s   Á5BÂBÚattrc                 ó   • g r9   r   )rF   rþ   s     r&   r\   Ú)_ExportPassBaseDeprecatedDoNotUse.on_attro  s   € Ør%   rÜ   r“   c                 óÖ   • U R                   R                  SUS0 5      nUR                  UR                  l        U R                   R                  UR                  U5        [        X$5      $ )Nr’   r   )rÐ   rá   r—   r^   r   r[   r   )rF   rÜ   r“   r   Ú	arg_proxys        r&   r’   Ú-_ExportPassBaseDeprecatedDoNotUse.placeholderr  sO   € Ø—K‘K×,Ñ,¨]¸DÀ"ÀbÓIˆ	Ø"Ÿi™iˆ	�‰ÔØ�‰× Ñ  §¡°Ô5Ü˜#Ó)Ð)r%   c                 ó(   • U R                  SXX45      $ ©Nrž   ©r°   )rF   rZ   r�   r�   r   s        r&   r¨   Ú/_ExportPassBaseDeprecatedDoNotUse.call_operatorx  s   € ð �x‰x˜¨°6Ó@Ð@r%   c                 ó*   • U R                  SX0 U5      $ r  r  )rF   rŽ   r�   r   s       r&   r¤   Ú*_ExportPassBaseDeprecatedDoNotUse.call_sym�  s   € ð �x‰x˜¨°r¸4Ó@Ð@r%   r²   r³   r´   rµ   c           
      óT  • U R                  U[        U5      5      nU R                  U[        U5      5      nUc  [        S5      eUc  [        S5      eU R                  S[        R
                  R                  R                  XR                  UR                  [        U5      40 U5      $ )Nztrue_branch must not be Nonezfalse_branch must not be Nonerž   )
Úcall_submodulerÊ   rh   r°   r<   r©   rª   r«   rì   rü   )rF   r²   r³   r´   rµ   r   Útrue_branchÚfalse_branchs           r&   r¬   Ú+_ExportPassBaseDeprecatedDoNotUse.call_cond‰  s›   € ð ×)Ñ)¨'´5¸³=ÓAˆØ×*Ñ*¨8´U¸6³]ÓCˆØÑÜ Ð!?Ó@Ð@ØÑÜ Ð!@ÓAÐAØ�x‰xØÜ�I‰I×"Ñ"×'Ñ'Ø×+Ñ+¨\×-FÑ-FÌÈVËÐUØØó
ð 	
r%   r¶   r·   r¸   c           
      óv  • [        U Vs/ s H  oUR                  PM     sn5      S   nU R                  U[        Xc Vs/ s H  oUR                  PM     sn-   5      5      nUc  [	        S5      eU R                  S[        R                  R                  R                  UR                  X#40 U5      $ s  snf s  snf )Nr   zf_branch must not be Nonerž   )r   r—   r  rÊ   rh   r°   r<   r©   rª   r­   rì   )rF   r¶   r·   r¸   r   r“   ÚxsÚf_branchs           r&   r®   Ú*_ExportPassBaseDeprecatedDoNotUse.call_mapŸ  s¦   € ô ±+Ó>²+¨3Ÿhœh±+Ñ>Ó?ÀÑBˆØ×&Ñ& q¬%°È(Ó5SÊ(À3·h´hÉ(Ñ5SÑ0SÓ*TÓUˆØÑÜ Ð!<Ó=Ð=Ø�x‰xØÜ�I‰I×"Ñ"×+Ñ+Ø×"Ñ" KÐ:ØØó
ð 	
ùò	 ?ùÚ5Ss
   ŠB1¾B6r`   r±   c                 óJ   • U R                  S[        R                  X40 U5      $ r  )r°   rŸ   r    )rF   r`   r±   r   s       r&   r¡   Ú._ExportPassBaseDeprecatedDoNotUse.call_getitem²  s#   € ð �x‰x˜¬×)9Ñ)9¸E¸<ÈÈTÓRÐRr%   Úresultsc                 ó.   • U R                  SSU40 U5      $ )Nr–   r  )rF   r  r   s      r&   r–   Ú(_ExportPassBaseDeprecatedDoNotUse.output·  s   € Ø�x‰x˜ (¨W¨J¸¸DÓAÐAr%   c                 óH  • U R                   U R                  XR                  R                  5      so0l         UR                  U R                   l        U R                  X5      nU R                  [        R                  R                  [        R                  R                  [        R                  R                  5       [        R                  R                  5       5      5      soPl        [        R                  " [         S U5      n["        R$                  " 5          UR&                  " U6   S S S 5        [        R                  R                  U R                   R(                  U R                   R                  5      nX0l         XPl        [+        US5      $ ! , (       d  f       Nn= f)Nc                 ó   • U R                   $ r9   rÖ   r×   s    r&   rØ   ÚB_ExportPassBaseDeprecatedDoNotUse.call_submodule.<locals>.<lambda>Ê  s   € ÀÇÂr%   T)rÐ   r„   rA   Ú_codegenrD   rÍ   rÏ   r<   r   ÚInterpreterrÉ   r=   r>   r@   r}   rÞ   r   Úfx_tracebackÚpreserve_node_metaÚrunr?   r   )rF   rì   rµ   Úprev_tracerrÏ   Úprev_interpreterÚinputs_dataÚnew_graph_modules           r&   r  Ú0_ExportPassBaseDeprecatedDoNotUse.call_submoduleº  s,  € ð �K‰KØ×Ñ˜d×$6Ñ$6×$?Ñ$?Ó@ð 	!ˆ”[ð (3×'CÑ'Cˆ�‰Ô$Ø×,Ñ,¨TÓ@ˆð ×ÑÜ�H‰H× Ñ Ü—‘×$Ñ$¤U§X¡X§_¡_Ó%6¼¿¹¿¹Ó8HÓIóð 	+ÐÔ*ô ×*Ò*¬:Ñ7GÈÓPˆÜ×,Ò,Õ.Ø�OŠO˜[Ñ)÷ /ô !Ÿ8™8×/Ñ/°·±×0@Ñ0@À$Ç+Á+×BSÑBSÓTÐà!ŒØ+ÔÜØØó
ð 	
÷ /Õ.ús   ÄFÆ
F!c                 óv  • [        U SS5      (       d  [        S5      eU R                  U5      nS nU HB  n[        U[        5      (       d  M  Ub  X4R
                  La  [        S5      eUR
                  nMD     Uc-  [        SS9U R                  l	        [        5       n[        5       nO!SUl        X0R                  l	        [        5       nU R                  R                  U l	        U   U   U R                  U[        U5      5      nS S S 5        S S S 5        W$ ! , (       d  f       N= f! , (       d  f       W$ = f)NrÑ   Fz.ExportPass is not initialized with __init__().z#Multiple fake tensor mode detected.T)Úallow_non_fake_inputs)r¢   r   rµ   rS   r   Ú	fake_moderh   r   rÐ   rD   r   r&  r	   r  rÊ   )rF   rì   rµ   rD   ÚiÚdispatcher_modeÚresults          r&   ÚcallÚ&_ExportPassBaseDeprecatedDoNotUse.call×  s
  € Ü�t˜^¨U×3Ñ3Ü%Ø@óð ð —‘˜\Ó*ˆàÐÛˆAÜ˜!œZ×(Ó(Ø#Ñ/Ð4DÏKÉKÒ4WÜ(Ð)NÓOÐOØ#$§;¡;Ò ñ	 ð
 Ñ#Ü+9ÐPTÑ+UˆD�K‰KÔ(Ü*›}ÐÜ)›m‰Oà59ÐÔ2Ø+;�K‰KÔ(Ü6Ó8ˆOØ $§¡× <Ñ <ˆÔâšØ×(Ñ(¨´u¸V³}ÓEˆF÷  /×ð ˆ÷  /�ú×Ôð ˆús$   Ã'D)Ã*DÄD)Ä
D&	Ä"D)Ä)
D8)rÑ   rD   rÏ   rÆ   rÐ   r�   )(r    r!   r"   r#   Ú__doc__Ústaticmethodr0   r   r„   r   r  rÍ   r;   rr   r<   r^   rÌ   rÊ   r|   rË   r
   r   r°   rÉ   rü   rµ   r\   r’   r¨   ÚFnr¤   r¬   r®   ro   r¡   r–   r   r  r+  r$   r   r%   r&   r   r   5   s¥  † ñð
 ñWó ðWôxP�ô xPôtg'˜BŸN™Nô g'ôR9ð/àð/ð —‘—‘×$Ñ$ð/ð �H˜c�MÑ"ð	/ð
 �S˜(�]Ñ#ð/ð ð/ð 
ô/ð<$
 5§8¡8×#7Ñ#7ð $
¸DÀ¹Nô $
ðL˜Jð ¨4ô ð* ð *¨(ð *¸,ð *È:ô *ðAð �H˜c�MÑ"ðAð �S˜(�]Ñ#ð	Að
 ðAð 
ôAðAàðAð �H˜c�MÑ"ðAð ð	Að
 
ôAð
àð
ð —‘×%Ñ%ð
ð —(‘(×&Ñ&ð	
ð
 �X‘ð
ð ð
ð 
ô
ð,
à�8‰8×Ñð
ð ˜*Ñ%ð
ð �zÑ"ð	
ð
 ð
ð 
ô
ð&SØðSØ&)ðSØ1=ðSà	ôSð
B˜d 8™nð B°Lð BÀZô Bð
ØŸN™Nð
Ø49¸(ÀC¸-Ñ4Hð
à	ô
ð: §¡ð °J÷ r%   )>rŸ   r   ÚtypingÚcollections.abcr   Ú
contextlibr   r   r   r   r<   r   Útorch._dispatch.pythonr	   Ú&torch._export.pass_infra.node_metadatar
   Ú$torch._export.pass_infra.proxy_valuer   Útorch._higher_order_ops.mapr   Útorch._subclassesr   r   Útorch._subclasses.fake_tensorr   Útorch.fxr  Ú"torch.fx.experimental.proxy_tensorr   Ú%torch.fx.experimental.symbolic_shapesr   r   Útorch.fx.graphr   Útorch.fx.passes.infra.pass_baser   r   Útorch.fx.passes.shape_propr   r   Útorch.utilsr   r}   Ú__all__r|   ÚValuer/  rÉ   ÚPassTypeÚsym_intÚ	sym_floatÚsym_iteÚsym_maxÚsym_minÚsym_notÚsym_sqrtr   ÚsetÚ__annotations__ÚRuntimeErrorr   r   r   r%   r&   Ú<module>rM     sï   ðä Û Û Ý $Ý "ß 'Ñ 'ã Ý Ý ;Ý ?Ý ;Ý 7ß HÝ 8Ý .Ý >÷õ #ß @ß OÝ )ð /Ð
/€ð €Ø€Øˆc�3ˆhÑ€Ø�U—X‘X×)Ñ)Ð*¨H°ZÑ,@Ð@ÑA€ð 
‡M�MØ	‡O�OØ	‡M�MØ	‡M�MØ	‡M�MØ	‡M�MØ	‡N�Nð!€��H‘ó ô	˜,ô 	ô}¨õ }r%   