ó
    !Eñi`
  ã                  ó–   • S SK Jr  S SKJr  S SKr\\\4   rS
S jr " S S\R                  R                  5      r " S S5      r/ S	Qrg)é    )Úannotations)ÚSelfNc                 ó>   • [         R                  R                  5       $ )zD
Return an opaque token representing the id of a graph memory pool.
)ÚtorchÚ_CÚ_mtia_graphPoolHandle© ó    ÚR/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/mtia/mtia_graph.pyÚgraph_pool_handler      s   € ô
 �8‰8×)Ñ)Ó+Ð+r
   c                  ó”   ^ • \ rS rSrSrSSU 4S jjjrSU 4S jjrSU 4S jjrSU 4S jjrSU 4S jjr	SU 4S jjr
SU 4S	 jjrS
rU =r$ )Ú	MTIAGraphé   z
Wrapper around a MTIA graph.
c                ó"   >• [         TU ]  X5      $ ©N)ÚsuperÚ__new__)ÚclsÚ
keep_graphÚ	__class__s     €r   r   ÚMTIAGraph.__new__   s   ø€ Ü‰w‰˜sÓ/Ð/r
   c                ó$   >• [         TU ]  U5        g)z
Begin capturing a MTIA graph.
N)r   Úcapture_begin)ÚselfÚpoolr   s     €r   r   ÚMTIAGraph.capture_begin   s   ø€ ô 	‰Ñ˜dÕ#r
   c                ó"   >• [         TU ]  5         g)z"
End the capture of a MTIA graph.
N)r   Úcapture_end©r   r   s    €r   r   ÚMTIAGraph.capture_end"   ó   ø€ ô 	‰ÑÕr
   c                ó"   >• [         TU ]  5         g)z&
Instantiate the captured MTIA graph.
N)r   Úinstantiater   s    €r   r#   ÚMTIAGraph.instantiate(   r!   r
   c                ó"   >• [         TU ]  5         g)z!
Replay the captured MTIA graph.
N)r   Úreplayr   s    €r   r&   ÚMTIAGraph.replay.   s   ø€ ô 	‰‰Õr
   c                ó"   >• [         TU ]  5         g)z2
Destroy the captured graph and reset the states.
N)r   Úresetr   s    €r   r)   ÚMTIAGraph.reset4   s   ø€ ô 	‰‰�r
   c                ó    >• [         TU ]  5       $ )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)r   ÚboolÚreturnr   )r   Ú_POOL_HANDLEr.   ÚNone©r.   r0   ©r.   r/   )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__r   r   r   r#   r&   r)   r   Ú__static_attributes__Ú__classcell__)r   s   @r   r   r      s5   ø† ñ÷0ñ 0÷$÷÷÷÷÷õ r
   r   c                  óT   • \ rS rSr% SrS\S'     S	     S
S jjrSS jrSS jrSr	g)ÚgraphéA   Nútorch.mtia.Stream | NoneÚdefault_capture_streamc                ó~  • U R                   R                  c-  [        R                  R	                  5       U R                   l        Uc  SOU4U l        Ub  UOU R                   R                  U l        U R                  c  [        S5      e[        R                  R                  U R                  5      U l	        Xl
        g )Nr	   zcapture_stream must not be None)r   r>   r   ÚmtiaÚcurrent_streamr   Úcapture_streamÚAssertionErrorÚstreamÚ
stream_ctxÚ
mtia_graph)r   rF   r   rD   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                ó*  • [         R                  R                  5         [         R                  R                  5         U R                  R                  5         U R                  (       a  U R                  S   OSnU R                  R                  U5        g )Nr   )r   r   )	r   r@   ÚsynchronizeÚempty_cacherE   Ú	__enter__r   rF   r   )r   Úpool_args     r   rL   Úgraph.__enter__V   sY   € Ü�
‰
×ÑÔ Ü�
‰
×ÑÔ à�‰×!Ñ!Ô#à#'§9§9�4—9‘9˜Q’<°&ˆØ�‰×%Ñ% hÕ/r
   c                ój   • U R                   R                  5         U R                  R                  " U6   g r   )rF   r   rE   Ú__exit__)r   Úargss     r   rP   Úgraph.__exit___   s$   € Ø�‰×#Ñ#Ô%Ø�‰× Ò  $Ò'r
   )rB   rF   r   rE   )NN)rF   r   r   z_POOL_HANDLE | NonerD   r=   r1   )rQ   Úobjectr.   r0   )
r3   r4   r5   r6   r>   Ú__annotations__rG   rL   rP   r8   r	   r
   r   r;   r;   A   sA   ‡ Ø7;ÐÐ4Ó;ð
 %)Ø+/ð	%àð%ð "ð%ð )õ	%ô$0÷(r
   r;   )r   r;   r   r2   )Ú
__future__r   Útyping_extensionsr   r   ÚtupleÚintr/   r   r   Ú
_MTIAGraphr   r;   Ú__all__r	   r
   r   Ú<module>r[      sK   ðå "å "ã ð �S˜#�X‰€ô,ô*�—‘×#Ñ#ô *÷Z (ñ  (òF�r
   