+
    –°³j1  ã                  ó  € ^ RI Ht ^ RIt^ RIHtHt ^ RIHt ^ RIH	t	H
t
 ^ RIt^ RIHu Ht ^ RIHtHtHtHtHtHtHtHtHt ^ RIHt ]
'       d   ^ RIHt   ! R	 R
]4      t]P>                  R R l4       t ]P>                  R R l4       t!]P>                  R R l4       t" ! R R]4      t#]P>                  R R l4       t$ ! R R]4      t% ! R R]4      t& ! R R]4      t'R R lt(R R lt)R R  lt*R! R" lt+R# R$ lt,R# )%é    )ÚannotationsN)ÚABCÚabstractmethod)Úcached_property)ÚAnyÚTYPE_CHECKING)	ÚCFunctionalizeInterpreterPtrÚCGradInterpreterPtrÚCInterpreterÚCJvpInterpreterPtrÚCVmapInterpreterPtrÚpop_dynamic_layer_stackÚpush_dynamic_layer_stackÚRandomnessTypeÚTransformType)Ú_set_fwd_grad_enabled)Ú	Generatorc                  ó€   € ] tR t^<tR R lt]R R l4       tR R ltR R ltR	 R
 lt	R R lt
R R ltR R ltRtR# )ÚFuncTorchInterpreterc               ó    € V ^8„  d   QhRRRR/# )é   Úcptrr   ÚreturnÚNone© )Úformats   "Ú]/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_functorch/pyfunctorch.pyÚ__annotate__Ú!FuncTorchInterpreter.__annotate__=   s   € ÷ ñ ˜Sð  Tñ ó    c                	ó   € Wn         R # ©N©Ú_cptr)Úselfr   s   &&r   Ú__init__ÚFuncTorchInterpreter.__init__=   s   € ØŽ
r    c               ó(   € V ^8„  d   QhRRRRRRRR/# ©r   Úopr   Úargsútuple[Any, ...]Úkwargsúdict[str, Any]r   r   )r   s   "r   r   r   C   s)   € ÷ ñ ˜#ð  _ð ¸nð ÐQTñ r    c                	ó   € R # r"   r   )r%   r*   r+   r-   s   &&&&r   ÚprocessÚFuncTorchInterpreter.processB   s   € ár    c               ó   € V ^8„  d   QhRR/# ©r   r   ú&contextlib.AbstractContextManager[Any]r   )r   s   "r   r   r   I   s   € ÷ 3ñ 3Ð=ñ 3r    c                	ó   € \        4       # r"   )Ú!temporarily_pop_interpreter_stack©r%   s   &r   ÚlowerÚFuncTorchInterpreter.lowerI   s   € Ü0Ó2Ð2r    c               ó   € V ^8„  d   QhRR/# ©r   r   Úintr   )r   s   "r   r   r   L   s   € ÷ "ñ "�sñ "r    c                	ó6   € V P                   P                  4       # r"   )r$   Úlevelr7   s   &r   r>   ÚFuncTorchInterpreter.levelL   s   € Ø�z‰z×ÑÓ!Ð!r    c               ó   € V ^8„  d   QhRR/# )r   r   r   r   )r   s   "r   r   r   O   s   € ÷  ñ  �]ñ  r    c                	ó6   € V P                   P                  4       # r"   )r$   Úkeyr7   s   &r   rB   ÚFuncTorchInterpreter.keyO   s   € Ø�z‰z�~‰~ÓÐr    c               ó   € V ^8„  d   QhRR/# ©r   r   r,   r   )r   s   "r   r   r   R   s   € ÷ "ñ "˜?ñ "r    c                	ó   € \         hr"   )ÚNotImplementedErrorr7   s   &r   Ú	get_stateÚFuncTorchInterpreter.get_stateR   s   € Ü!Ð!r    c               ó    € V ^8„  d   QhRRRR/# )r   Ústater,   r   Úboolr   )r   s   "r   r   r   U   s   € ÷ )ñ ) ð )°Tñ )r    c                	ó&   € WP                  4       8H  # r"   )rH   ©r%   rK   s   &&r   Úcheck_stateÚ FuncTorchInterpreter.check_stateU   s   € ØŸ™Ó(Ñ(Ð(r    c               ó   € V ^8„  d   QhRR/# )r   r   r.   r   )r   s   "r   r   r   X   s   € ÷ ñ ˜nñ r    c                	ó^   € V P                   P                  4       pVP                  R R4       V# )r$   N)Ú__dict__ÚcopyÚpoprN   s   & r   Ú__getstate__Ú!FuncTorchInterpreter.__getstate__X   s'   € Ø—‘×"Ñ"Ó$ˆØ�	‰	�'˜4Ô Øˆr    r#   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r&   r   r0   r8   r>   rB   rH   rO   rV   Ú__static_attributes__r   r    r   r   r   <   s>   † õð
 ôó ðõ3õ"õ õ"õ)÷ñ r    r   c               ó   € V ^8„  d   QhRR/# )r   r   úGenerator[None, None, None]r   )r   s   "r   r   r   _   s   € ÷ (ñ (Ð+Fñ (r    c               #  ób   "  €  \        4       p R x € \        V 4       R #   \        X 4       i ; i5ir"   )r   r   )Úsaveds    r   r6   r6   ^   s&   é € ð(Ü'Ó)ˆÛä  Ö'øÔ  Õ'üs   ‚/„ ’/Ÿ,¬/c               ó   € V ^8„  d   QhRR/# )r   r   z Generator[list[Any], None, None]r   )r   s   "r   r   r   h   s   € ÷ 2ñ 2Ð-Mñ 2r    c               #  óN  "  € . p  \         P                  P                  P                  4       e   V P	                  \        4       4       KE  \        V 4      x € V '       d   \        V P                  4       4       K#  R #   T '       d   \        T P                  4       4       K#  i ; i5ir"   )	ÚtorchÚ_CÚ
_functorchÚpeek_interpreter_stackÚappendr   Úlistr   rU   )Ústacks    r   Ú#temporarily_clear_interpreter_stackrj   g   sn   é € à€Eð2Ü�h‰h×!Ñ!×8Ñ8Ó:ÒFØ�L‰LÔ0Ó2Ö3Ü�5‹kÒçÜ$ U§Y¡Y£[Ö1ñ ø�eÜ$ U§Y¡Y£[Ö1ð üs)   ‚B%†AA= ÁB%Á B%Á=B"ÂB"Â"B%c               ó    € V ^8„  d   QhRRRR/# )r   ri   zlist[Any] | Noner   r^   r   )r   s   "r   r   r   t   s   € ÷ &ñ &Øð&à ñ&r    c              #  óþ   "  € . pV f   R #  \        V 4       F  p\        V4       VP                  V4       K!  	  R x € \        V4       F  p\        4        K  	  R #   \        T4       F  p\        4        K  	  i ; i5ir"   )Úreversedr   rg   r   )ri   ÚpushedÚsÚ_s   &   r   Ú%temporarily_restore_interpreter_stackrq   s   sn   é € ð €FØ‚}Ùð	&Ü˜%–ˆAÜ$ QÔ'Ø�M‰M˜!Öñ !ó 	ä˜&Ö!ˆAô $Ö%ó "ø”˜&Ö!ˆAô $Ö%ò "üs   ‚	A=Œ2A ¾A=ÁA:Á:A=c                  óh   € ] tR t^†tR R lt]R R l4       tR R ltR R ltR	 R
 lt	R R lt
RtR# )ÚVmapInterpreterc               ó    € V ^8„  d   QhRRRR/# ©r   Úcdatar   r   r   r   )r   s   "r   r   ÚVmapInterpreter.__annotate__‡   s   € ÷ ñ ˜lð ¨tñ r    c                	ó�   € VP                  4       \        P                  8w  d   \        R VP                  4        24      hWn        R# )z!expected TransformType.Vmap, got N)rB   r   ÚVmapÚAssertionErrorÚ_cdata©r%   rv   s   &&r   r&   ÚVmapInterpreter.__init__‡   s6   € Ø�9‰9‹;œ-×,Ñ,Ô,Ü Ð#DÀUÇYÁYÃ[ÀMÐ!RÓSÐSð Žr    c               ó   € V ^8„  d   QhRR/# )r   r   r   r   )r   s   "r   r   rw   ‘   ó   € ÷ 0ñ 0Ð*ñ 0r    c                	ó,   € \        V P                  4      # r"   )r   r{   r7   s   &r   r$   ÚVmapInterpreter._cptr�   ó   € ô # 4§;¡;Ó/Ð/r    c               ó(   € V ^8„  d   QhRRRRRRRR/# r)   r   )r   s   "r   r   rw   ”   ó)   € ÷ -ñ -˜#ð - _ð -¸nð -ÐQTñ -r    c                	ó\   € VP                   \        P                  ,          pV! V .VO5/ VB # r"   )Úfunctorch_tabler   ry   ©r%   r*   r+   r-   Úkernels   &&&& r   r0   ÚVmapInterpreter.process”   s-   € Ø×#Ñ#¤M×$6Ñ$6Õ7ˆÙ�dÐ,˜TÒ, VÑ,Ð,r    c               ó   € V ^8„  d   QhRR/# r;   r   )r   s   "r   r   rw   ˜   s   € ÷ &ñ &˜Cñ &r    c                	ó6   € V P                   P                  4       # r"   )r$   Ú	batchSizer7   s   &r   Ú
batch_sizeÚVmapInterpreter.batch_size˜   s   € Ø�z‰z×#Ñ#Ó%Ð%r    c               ó   € V ^8„  d   QhRR/# )r   r   Ústrr   )r   s   "r   r   rw   ›   s   € ÷ =ñ =˜Cñ =r    c                	óÜ   € V P                   P                  4       pV\        P                  8X  d   R # V\        P                  8X  d   R# V\        P
                  8X  d   R# \        RV 24      h)ÚerrorÚsameÚ	differentzUnknown RandomnessType: )r$   Ú
randomnessr   ÚErrorÚSameÚ	DifferentÚRuntimeError)r%   Útyps   & r   r•   ÚVmapInterpreter.randomness›   sZ   € Ø�j‰j×#Ñ#Ó%ˆØ”.×&Ñ&Ô&ÙØ”N×'Ñ'Ô'ÙØ”N×,Ñ,Ô,ÙÜÐ5°c°UÐ;Ó<Ð<r    c               ó   € V ^8„  d   QhRR/# rE   r   )r   s   "r   r   rw   ¥   s   € ÷ Bñ B˜?ñ Br    c                	ót   € V P                  4       P                  V P                  4       V P                  4       3# r"   )rB   Únamer>   r•   r7   s   &r   rH   ÚVmapInterpreter.get_state¥   s&   € Ø—‘“
—‘ §¡£¨t¯©Ó/@ÐAÐAr    ©r{   N)rX   rY   rZ   r[   r&   r   r$   r0   r�   r•   rH   r\   r   r    r   rs   rs   †   s6   † õð ô0ó ð0õ-õ&õ=÷Bñ Br    rs   c               ó    € V ^8„  d   QhRRRR/# )r   Úcontextsr4   r   zIGenerator[tuple[contextlib.AbstractContextManager[Any], ...], None, None]r   )r   s   "r   r   r   ª   s   € ÷ ñ Ø5ðàNñr    c               '  ó¾   "  € \         P                  ! 4       ;_uu_ 4       pV  F  pVP                  V4       K  	  V x € R R R 4       R #   + '       g   i     R # ; i5ir"   )Ú
contextlibÚ	ExitStackÚenter_context)r¢   ri   Úctxs   *  r   Únestedr¨   ©   sC   é € ô 
×	Ò	×	Ô	 5ÛˆCØ×Ñ Ö$ñ àŠ÷ 
 ×	×	Ò	üs   ‚A A	¿
AÁ	A	Á	Ac                  ó€   a € ] tR t^³tR R lt]R R l4       tR R ltR R ltR	 V 3R
 llt	R R lt
R R ltRtV ;t# )ÚGradInterpreterc               ó    € V ^8„  d   QhRRRR/# ru   r   )r   s   "r   r   ÚGradInterpreter.__annotate__´   ó   € ÷ ñ ˜lð ¨tñ r    c                	ó�   € VP                  4       \        P                  8w  d   \        R VP                  4        24      hWn        R# )z!expected TransformType.Grad, got N)rB   r   ÚGradrz   r{   r|   s   &&r   r&   ÚGradInterpreter.__init__´   s4   € Ø�9‰9‹;œ-×,Ñ,Ô,Ü Ð#DÀUÇYÁYÃ[ÀMÐ!RÓSÐSàŽr    c               ó   € V ^8„  d   QhRR/# )r   r   r
   r   )r   s   "r   r   r¬   ¼   r   r    c                	ó,   € \        V P                  4      # r"   )r
   r{   r7   s   &r   r$   ÚGradInterpreter._cptrº   r‚   r    c               ó$   € V ^8„  d   QhRRRRRR/# ©r   r+   r,   r-   r.   r   z&tuple[tuple[Any, ...], dict[str, Any]]r   )r   s   "r   r   r¬   ¿   ó$   € ÷ ñ Ø#ðØ-;ðà	/ñr    c                	ó‚   € \         P                  ! \        P                  V P                  P
                  W.4      w  rW3# r"   ©ÚpytreeÚtree_map_onlyrc   ÚTensorr$   Úlift©r%   r+   r-   s   &&&r   r¼   ÚGradInterpreter.lift¿   ó5   € ô ×+Ò+Ü�L‰L˜$Ÿ*™*Ÿ/™/¨D¨>ó
‰ˆð ˆ|Ðr    c               ó(   € V ^8„  d   QhRRRRRRRR/# r)   r   )r   s   "r   r   r¬   Ç   ó)   € ÷ -ñ -˜#ð - _ð -¸nð -ÐQTñ -r    c                	ó‚   € VP                   \        P                  ,          pV P                  W#4      w  r#V! V .VO5/ VB # r"   )r†   r   r¯   r¼   r‡   s   &&&& r   r0   ÚGradInterpreter.processÇ   s=   € Ø×#Ñ#¤M×$6Ñ$6Õ7ˆØ—y‘y Ó.‰ˆÙ�dÐ,˜TÒ, VÑ,Ð,r    c               ó   € V ^8„  d   QhRR/# r3   r   )r   s   "r   r   r¬   Ï   ó   € ÷ ñ Ð=ñ r    c                	ó¦   <€ V P                  4       pV'       g,   \        \        P                  ! 4       \        SV `  4       4      # \        SV `  4       # r"   )Úprev_grad_moder¨   rc   Úno_gradÚsuperr8   )r%   rÇ   Ú	__class__s   & €r   r8   ÚGradInterpreter.lowerÏ   s9   ø€ Ø×,Ñ,Ó.ˆßÜœ%Ÿ-š-›/¬5©7©=«?Ó;Ð;Ü‰w‰}‹Ðr    c               ó   € V ^8„  d   QhRR/# ©r   r   rL   r   )r   s   "r   r   r¬   Õ   s   € ÷ )ñ ) ñ )r    c                	ó6   € V P                   P                  4       # r"   )r$   ÚprevGradModer7   s   &r   rÇ   ÚGradInterpreter.prev_grad_modeÕ   s   € Ø�z‰z×&Ñ&Ó(Ð(r    c               ó   € V ^8„  d   QhRR/# rE   r   )r   s   "r   r   r¬   Ø   s   € ÷ Fñ F˜?ñ Fr    c                	ót   € V P                  4       P                  V P                  4       V P                  4       3# r"   )rB   rž   r>   rÇ   r7   s   &r   rH   ÚGradInterpreter.get_stateØ   s(   € Ø—‘“
—‘ §¡£¨t×/BÑ/BÓ/DÐEÐEr    r    )rX   rY   rZ   r[   r&   r   r$   r¼   r0   r8   rÇ   rH   r\   Ú__classcell__©rÊ   s   @r   rª   rª   ³   sA   ø† õð ô0ó ð0õõ-÷ð õ)÷Fó Fr    rª   c                  ó€   a € ] tR t^ÜtR R lt]R R l4       tR R ltR R ltR	 V 3R
 llt	R R lt
R R ltRtV ;t# )ÚJvpInterpreterc               ó    € V ^8„  d   QhRRRR/# ru   r   )r   s   "r   r   ÚJvpInterpreter.__annotate__Ý   r­   r    c                	ó�   € VP                  4       \        P                  8w  d   \        R VP                  4        24      hWn        R# )z expected TransformType.Jvp, got N)rB   r   ÚJvprz   r{   r|   s   &&r   r&   ÚJvpInterpreter.__init__Ý   s4   € Ø�9‰9‹;œ-×+Ñ+Ô+Ü Ð#CÀEÇIÁIÃKÀ=Ð!QÓRÐRàŽr    c               ó   € V ^8„  d   QhRR/# )r   r   r   r   )r   s   "r   r   rÙ   å   s   € ÷ /ñ /Ð)ñ /r    c                	ó,   € \        V P                  4      # r"   )r   r{   r7   s   &r   r$   ÚJvpInterpreter._cptrã   s   € ô " $§+¡+Ó.Ð.r    c               ó$   € V ^8„  d   QhRRRRRR/# rµ   r   )r   s   "r   r   rÙ   è   r¶   r    c                	ó‚   € \         P                  ! \        P                  V P                  P
                  W.4      w  rW3# r"   r¸   r½   s   &&&r   r¼   ÚJvpInterpreter.liftè   r¿   r    c               ó(   € V ^8„  d   QhRRRRRRRR/# r)   r   )r   s   "r   r   rÙ   ð   rÁ   r    c                	ó‚   € VP                   \        P                  ,          pV P                  W#4      w  r#V! V .VO5/ VB # r"   )r†   r   rÛ   r¼   r‡   s   &&&& r   r0   ÚJvpInterpreter.processð   s=   € Ø×#Ñ#¤M×$5Ñ$5Õ6ˆØ—y‘y Ó.‰ˆÙ�dÐ,˜TÒ, VÑ,Ð,r    c               ó   € V ^8„  d   QhRR/# r3   r   )r   s   "r   r   rÙ   ø   rÅ   r    c                	ó’   <€ V P                  4       pV'       g"   \        \        R 4      \        SV `  4       4      # \        SV `  4       # )F)Úprev_fwd_grad_moder¨   r   rÉ   r8   )r%   rè   rÊ   s   & €r   r8   ÚJvpInterpreter.lowerø   s:   ø€ Ø!×4Ñ4Ó6Ðß!ÜÔ/°Ó6¼¹¹»ÓHÐHÜ‰w‰}‹Ðr    c               ó   € V ^8„  d   QhRR/# rÍ   r   )r   s   "r   r   rÙ   þ   s   € ÷ ,ñ , Dñ ,r    c                	ó6   € V P                   P                  4       # r"   )r$   ÚprevFwdGradModer7   s   &r   rè   Ú!JvpInterpreter.prev_fwd_grad_modeþ   s   € Ø�z‰z×)Ñ)Ó+Ð+r    c               ó   € V ^8„  d   QhRR/# rE   r   )r   s   "r   r   rÙ     s   € ÷ Jñ J˜?ñ Jr    c                	ót   € V P                  4       P                  V P                  4       V P                  4       3# r"   )rB   rž   r>   rè   r7   s   &r   rH   ÚJvpInterpreter.get_state  s(   € Ø—‘“
—‘ §¡£¨t×/FÑ/FÓ/HÐIÐIr    r    )rX   rY   rZ   r[   r&   r   r$   r¼   r0   r8   rè   rH   r\   rÔ   rÕ   s   @r   r×   r×   Ü   sA   ø† õð ô/ó ð/õõ-÷ð õ,÷Jó Jr    r×   c                  ó\   € ] tR tRtR R lt]R R l4       tR R ltR R	 ltR
 R lt	Rt
R# )ÚFunctionalizeInterpreteri  c               ó    € V ^8„  d   QhRRRR/# ru   r   )r   s   "r   r   Ú%FunctionalizeInterpreter.__annotate__  s   € ÷ ñ ˜lð ¨tñ r    c                	ó�   € VP                  4       \        P                  8w  d   \        R VP                  4        24      hWn        R# )z*expected TransformType.Functionalize, got N)rB   r   ÚFunctionalizerz   r{   r|   s   &&r   r&   Ú!FunctionalizeInterpreter.__init__  s:   € Ø�9‰9‹;œ-×5Ñ5Ô5Ü Ø<¸U¿Y¹Y»[¸MÐJóð ð Žr    c               ó   € V ^8„  d   QhRR/# )r   r   r	   r   )r   s   "r   r   rô     s   € ÷ 9ñ 9Ð3ñ 9r    c                	ó,   € \        V P                  4      # r"   )r	   r{   r7   s   &r   r$   ÚFunctionalizeInterpreter._cptr  s   € ô ,¨D¯K©KÓ8Ð8r    c               ó(   € V ^8„  d   QhRRRRRRRR/# r)   r   )r   s   "r   r   rô     r„   r    c                	ó\   € VP                   \        P                  ,          pV! V .VO5/ VB # r"   )r†   r   rö   r‡   s   &&&& r   r0   Ú FunctionalizeInterpreter.process  s-   € Ø×#Ñ#¤M×$?Ñ$?Õ@ˆÙ�dÐ,˜TÒ, VÑ,Ð,r    c               ó   € V ^8„  d   QhRR/# rÍ   r   )r   s   "r   r   rô     s   € ÷ 6ñ 6¨dñ 6r    c                	ó6   € V P                   P                  4       # r"   )r$   ÚfunctionalizeAddBackViewsr7   s   &r   Úfunctionalize_add_back_viewsÚ5FunctionalizeInterpreter.functionalize_add_back_views  s   € Ø�z‰z×3Ñ3Ó5Ð5r    c               ó   € V ^8„  d   QhRR/# rE   r   )r   s   "r   r   rô     s   € ÷ /ñ /˜?ñ /r    c                	óV   € V P                  4       P                  V P                  4       3# r"   )rB   rž   r>   r7   s   &r   rH   Ú"FunctionalizeInterpreter.get_state  s   € Ø—‘“
—‘ §¡£Ð.Ð.r    r    N)rX   rY   rZ   r[   r&   r   r$   r0   r  rH   r\   r   r    r   rò   rò     s/   † õð ô9ó ð9õ-õ6÷/ñ /r    rò   c               ó    € V ^8„  d   QhRRRR/# )r   Úcinterpreterr   r   r   r   )r   s   "r   r   r     s   € ÷ 
Sñ 
S lð 
SÐ7Kñ 
Sr    c                ó>  € V P                  4       pV\        P                  8X  d   \        V 4      # V\        P                  8X  d   \        V 4      # V\        P                  8X  d   \        V 4      # V\        P                  8X  d   \        V 4      # \        R V 24      h)z2NYI: PyDispatcher has not implemented support for )rB   r   r¯   rª   ry   rs   rÛ   r×   rö   rò   r™   )r  rB   s   & r   Úcoerce_cinterpreterr	    s‡   € Ø
×
Ñ
Ó
€CØ
Œm× Ñ Ô Ü˜|Ó,Ð,Ø
Œm× Ñ Ô Ü˜|Ó,Ð,Ø
Œm×ÑÔÜ˜lÓ+Ð+Ø
Œm×)Ñ)Ô)Ü'¨Ó5Ð5Ü
ÐKÈCÈ5ÐQÓ
RÐRr    c               ó   € V ^8„  d   QhRR/# )r   r   r   r   )r   s   "r   r   r   *  s   € ÷ ,ñ ,Ð0Dñ ,r    c                 ó†   € \         P                  P                  P                  4       p V f   \	        R4      h\        V 4      # )Nzinterpreter must not be None)rc   rd   re   rf   rz   r	  )Úinterpreters    r   Ú&retrieve_current_functorch_interpreterr  *  s7   € Ü—(‘(×%Ñ%×<Ñ<Ó>€KØÒÜÐ;Ó<Ð<Ü˜{Ó+Ð+r    c               ó   € V ^8„  d   QhRR/# )r   r   zlist[FuncTorchInterpreter]r   )r   s   "r   r   r   1  s   € ÷ 3ñ 3Ð-Gñ 3r    c                 óž   € \         P                  P                  P                  4       p V f   . # V  Uu. uF  p\	        V4      NK  	  up# u upi r"   )rc   rd   re   Úget_interpreter_stackr	  )ÚcisÚcis     r   Ú#retrieve_all_functorch_interpretersr  1  sB   € Ü
�(‰(×
Ñ
×
3Ñ
3Ó
5€CØ
‚{Øˆ	Ù.1Ó2©c¨Ô Ö#©cÑ2Ð2ùÒ2s   ³A
c               ó    € V ^8„  d   QhRRRR/# )r   Ústateszlist[tuple[Any, ...]]r   rL   r   )r   s   "r   r   r   8  s   € ÷ ñ Ð$9ð ¸dñ r    c                ó„  € \         P                  P                  P                  4       pVf   \	        V 4      ^ 8w  g   Ve   \	        V 4      ^ 8X  d   R# \        4       p\	        V4      \	        V 4      8H  ;'       dG    \        ;QJ d%    R \        W 4       4       F  '       d   K   R# 	  R# ! R \        W 4       4       4      # )NFc              3  óH   "  € T F  w  rVP                  V4      x € K  	  R # 5ir"   )rO   )Ú.0r  rK   s   &  r   Ú	<genexpr>Ú*compare_functorch_state.<locals>.<genexpr>C  s"   é € ð +Ù/?¡) "ˆ�‰�u×ÐÓ/?ùs   ‚ "T)rc   rd   re   rf   Úlenr  ÚallÚzip)r  Úpeekr  s   &  r   Úcompare_functorch_stater  8  s¦   € ô �8‰8×Ñ×5Ñ5Ó7€DØŠœ˜V›¨Ô)¨tÒ/?ÄCÈÃKÐSTÔDTÙä
-Ó
/€CÜˆs‹8”s˜6“{Ñ"÷ ð §s£sñ +Ü/2°3Ô/?ó+§s¤sð ¢sð  sñ +Ü/2°3Ô/?ó+ó (ð r    c               ó(   € V ^8„  d   QhRRRRRRRR/# r)   r   )r   s   "r   r   r   H  s)   € ÷ 
1ñ 
1˜3ð 
1 oð 
1¸~ð 
1ÐRUñ 
1r    c                óÐ   € \        4       p\        P                  ! \        P                  \        P
                  P                  P                  W34      w  rVP                  WV4      # r"   )	r  r¹   rº   rc   r»   rd   re   Úunwrap_if_deadr0   )r*   r+   r-   r  s   &&& r   Údispatch_functorchr#  H  sP   € Ü8Ó:€Kô ×'Ò'Ü�‰”e—h‘h×)Ñ)×8Ñ8¸4¸.ó�L€Dð ×Ñ˜r¨Ó0Ð0r    )-Ú
__future__r   r¤   Úabcr   r   Ú	functoolsr   Útypingr   r   rc   Útorch.utils._pytreeÚutilsÚ_pytreer¹   Útorch._C._functorchr	   r
   r   r   r   r   r   r   r   Útorch.autograd.forward_adr   Úcollections.abcr   r   Úcontextmanagerr6   rj   rq   rs   r¨   rª   r×   rò   r	  r  r  r  r#  r   r    r   Ú<module>r/     s  ðÝ "ã ß #Ý %ß %ã ß $Ð $÷
÷ 
õ 
õ <÷ Ý)ðô@˜3ô ðD ×Ñô(ó ð(ð ×Ñô2ó ð2ð ×Ñô&ó ð&ô$ BÐ*ô  BðF ×Ñôó ðô&FÐ*ô &FôR&JÐ)ô &JôR/Ð3ô /õ0
Sõ,õ3õ÷ 
1r    