+
    ˜°³j`
  ã                  ó¢   € ^ RI Ht ^ RIHt ^ RIt]]]3,          tR R lt ! R R]P                  P                  4      t ! R R	4      t. R
OtR# )é    )Úannotations)ÚSelfNc               ó   € V ^8„  d   QhRR/# ©é   ÚreturnÚ_POOL_HANDLE© )Úformats   "ÚV/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/mtia/mtia_graph.pyÚ__annotate__r      s   € ÷ ,ñ ,˜<ñ ,ó    c                 ó>   € \         P                  P                  4       # )zD
Return an opaque token representing the id of a graph memory pool.
)ÚtorchÚ_CÚ_mtia_graphPoolHandler
   r   r   Úgraph_pool_handler      s   € ô
 �8‰8×)Ñ)Ó+Ð+r   c                  ó¢   a € ] tR t^tRtRR V 3R llltR V 3R lltR V 3R lltR V 3R	 lltR
 V 3R llt	R V 3R llt
R V 3R lltRtV ;t# )Ú	MTIAGraphz
Wrapper around a MTIA graph.
c               ó    € V ^8„  d   QhRRRR/# )r   Ú
keep_graphÚboolr   r   r
   )r   s   "r   r   ÚMTIAGraph.__annotate__   s   € ÷ 0ñ 0 ð 0°$ñ 0r   c                	ó"   <€ \         SV `  W4      # ©N)ÚsuperÚ__new__)Úclsr   Ú	__class__s   &&€r   r   ÚMTIAGraph.__new__   s   ø€ Ü‰w‰˜sÓ/Ð/r   c               ó    € V ^8„  d   QhRRRR/# )r   Úpoolr	   r   ÚNoner
   )r   s   "r   r   r      s   € ÷ $ñ $ ,ð $°4ñ $r   c                ó&   <€ \         SV `  V4       R# )z
Begin capturing a MTIA graph.
N)r   Úcapture_begin)Úselfr"   r   s   &&€r   r%   ÚMTIAGraph.capture_begin   s   ø€ ô 	‰Ñ˜dÖ#r   c               ó   € V ^8„  d   QhRR/# ©r   r   r#   r
   )r   s   "r   r   r   "   ó   € ÷ ñ ˜Tñ r   c                ó$   <€ \         SV `  4        R# )z"
End the capture of a MTIA graph.
N)r   Úcapture_end©r&   r   s   &€r   r,   ÚMTIAGraph.capture_end"   ó   ø€ ô 	‰ÑÖr   c               ó   € V ^8„  d   QhRR/# r)   r
   )r   s   "r   r   r   (   r*   r   c                ó$   <€ \         SV `  4        R# )z&
Instantiate the captured MTIA graph.
N)r   Úinstantiater-   s   &€r   r2   ÚMTIAGraph.instantiate(   r/   r   c               ó   € V ^8„  d   QhRR/# r)   r
   )r   s   "r   r   r   .   s   € ÷ ñ ˜ñ r   c                ó$   <€ \         SV `  4        R# )z!
Replay the captured MTIA graph.
N)r   Úreplayr-   s   &€r   r6   ÚMTIAGraph.replay.   s   ø€ ô 	‰‰Ör   c               ó   € V ^8„  d   QhRR/# r)   r
   )r   s   "r   r   r   4   s   € ÷ ñ �tñ r   c                ó$   <€ \         SV `  4        R# )z2
Destroy the captured graph and reset the states.
N)r   Úresetr-   s   &€r   r:   ÚMTIAGraph.reset4   s   ø€ ô 	‰‰Žr   c               ó   € V ^8„  d   QhRR/# r   r
   )r   s   "r   r   r   :   s   € ÷ ñ �lñ r   c                ó    <€ \         SV `  4       # )zH
Return an opaque token representing the id of this graph's memory pool
)r   r"   r-   s   &€r   r"   ÚMTIAGraph.pool:   s   ø€ ô ‰w‰|‹~Ðr   r
   )F)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__r   r%   r,   r2   r6   r:   r"   Ú__static_attributes__Ú__classcell__)r   s   @r   r   r      sN   ø† ñ÷0ò 0÷$ð $÷ð ÷ð ÷ð ÷ð ÷ö r   r   c                  óN   € ] tR t^At$ RtR]R&   RR R lltR R ltR R	 ltR
t	R# )ÚgraphNútorch.mtia.Stream | NoneÚdefault_capture_streamc               ó$   € V ^8„  d   QhRRRRRR/# )r   Ú
mtia_graphr   r"   z_POOL_HANDLE | NoneÚstreamrH   r
   )r   s   "r   r   Úgraph.__annotate__D   s(   € ÷ %ñ %àð%ð "ð%ð )ñ	%r   c                	óˆ  € V P                   P                  f.   \        P                  P	                  4       V P                   n        Vf   RMV3V n        Ve   TMV P                   P                  V n        V P                  f   \        R4      h\        P                  P                  V P                  4      V n	        Wn
        R # )Nzcapture_stream must not be Noner
   )r   rI   r   ÚmtiaÚcurrent_streamr"   Úcapture_streamÚAssertionErrorrL   Ú
stream_ctxrK   )r&   rK   r"   rL   s   &&&&r   Ú__init__Úgraph.__init__D   s“   € ð �>‰>×0Ñ0Ò8Ü49·J±J×4MÑ4MÓ4OˆD�N‰NÔ1à;?º<±RÈdÈWˆŒ	àÒ(‰F¨d¯n©n×.SÑ.Sð 	Ôð ×ÑÒ&Ü Ð!BÓCÐCÜŸ*™*×+Ñ+¨D×,?Ñ,?Ó@ˆŒØ$Žr   c               ó   € V ^8„  d   QhRR/# r)   r
   )r   s   "r   r   rM   V   s   € ÷ 0ñ 0˜4ñ 0r   c                	ó6  € \         P                  P                  4        \         P                  P                  4        V P                  P                  4        V P                  '       d   V P                  ^ ,          MRpV P                  P                  V4       R# )r   N)r   r   )	r   rO   ÚsynchronizeÚempty_cacherS   Ú	__enter__r"   rK   r%   )r&   Úpool_args   & r   rZ   Úgraph.__enter__V   s[   € Ü�
‰
×ÑÔ Ü�
‰
×ÑÔ à�‰×!Ñ!Ô#à#'§9§9 9�4—9‘9˜Q–<°&ˆØ�‰×%Ñ% hÖ/r   c               ó    € V ^8„  d   QhRRRR/# )r   ÚargsÚobjectr   r#   r
   )r   s   "r   r   rM   _   s   € ÷ (ñ (˜fð (¨ñ (r   c                	ón   € V P                   P                  4        V P                  P                  ! V!   R # r   )rK   r,   rS   Ú__exit__)r&   r^   s   &*r   ra   Úgraph.__exit___   s$   € Ø�‰×#Ñ#Ô%Ø�‰× Ò  $Ô'r   )rQ   rK   r"   rS   )NN)
r?   r@   rA   rB   rI   Ú__annotations__rT   rZ   ra   rD   r
   r   r   rG   rG   A   s"   ‡ Ø7;ÐÐ4Ó;÷%õ$0÷(ñ (r   rG   )r   rG   r   )Ú
__future__r   Útyping_extensionsr   r   ÚtupleÚintr	   r   r   Ú
_MTIAGraphr   rG   Ú__all__r
   r   r   Ú<module>rj      sK   ðå "å "ã ð �S˜#�X�€õ,ô*�—‘×#Ñ#ô *÷Z (ñ  (òF‚r   