+
    –°³jä  ã                   ó†  € ^ RI Ht ^ RIHtHt ^ RIt^ RIHtHt ^RI	H
t
 ^RIHtHt ^RIHt ^RIHtHt ^R	IHtHtHt ^R
IHtHt ^RIHt ^RIHt ]'       d   ^ RIHt ^RI H!t! ^ RI"H#t# R R lt$R R lt%]#! RR#R7      R R l4       t&]&PN                  R R l4       t(]! ]PR                  PT                  PV                  PX                  4       ]#! RR#R7      R R l4       t-]-PN                  R R l4       t(]! ]PR                  PT                  P\                  PX                  4        ! R R ]4      t/ ! R! R"]4      t0R# )$é    )Ú	ExitStack)ÚAnyÚTYPE_CHECKINGN)Úhas_side_effectÚProxy)Úcreate_call_function)Úraise_type_errorÚunimplemented©Úget_external_object_by_index)ÚGuardBuilderÚinstall_guard)Ú
AttrSourceÚCallFunctionNoArgsSourceÚImportSource)ÚMemberÚVariableTracker)ÚConstantVariable)ÚContextWrappingVariable)ÚInstructionTranslatorBase)Ú	PyCodegen)Ú	custom_opc                óX   € V ^8„  d   QhR\         R\        P                  P                  /# )é   ÚindexÚreturn)ÚintÚtorchÚcudaÚMemPool)Úformats   "Ú_/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_dynamo/variables/memory.pyÚ__annotate__r#      s$   € ÷ ñ ¤ð ¬¯©×);Ñ);ñ ó    c                 óŒ   € \        V 4      p\        V\        P                  P                  4      '       g   \        R V  24      hV# )z;use_mem_pool expected a torch.cuda.MemPool object at index )r   Ú
isinstancer   r   r    ÚRuntimeError)r   Úmempools   & r"   Ú_get_mempool_by_indexr)      sA   € Ü*¨5Ó1€GÜ�gœuŸz™z×1Ñ1×2Ò2ÜØIÈ%ÈÐQó
ð 	
ð €Nr$   c                ó$   € V ^8„  d   QhR\         /# ©r   r   )r   )r!   s   "r"   r#   r#   "   s   € ÷ ñ Ô%=ñ r$   c            	      óR   € \        \        \        \        R 4      R4      R4      4      # )r   r   Úcurrent_device)r   r   r   © r$   r"   Ú_current_cuda_device_sourcer/   "   s&   € Ü#Ü”:œl¨7Ó3°VÓ<Ð>NÓOóð r$   zmempool::begin)Úmutates_argsc                ó4   € V ^8„  d   QhR\         R\         RR/# ©r   Údevice_indexÚmempool_indexr   N©r   )r!   s   "r"   r#   r#   -   s&   € ÷ Wñ W¤ð W´Cð W¸Dñ Wr$   c                 ó‚   € \        V4      p\        P                  P                  P	                  WP
                  4       R # ©N)r)   r   r   ÚmemoryÚ&_cuda_beginAllocateCurrentThreadToPoolÚid©r3   r4   r(   s   && r"   Úbegin_mempoolr<   ,   s(   € ä# MÓ2€GÜ	‡J�J×Ñ×<Ñ<¸\Ï:É:ÖVr$   c                ó4   € V ^8„  d   QhR\         R\         RR/# r2   r5   )r!   s   "r"   r#   r#   3   ó!   € ÷ 	ñ 	”Cð 	¬ð 	°ñ 	r$   c                 ó   € R # r7   r.   ©r3   r4   s   &&r"   Ú_rA   2   ó   € ár$   zmempool::endc                ó4   € V ^8„  d   QhR\         R\         RR/# r2   r5   )r!   s   "r"   r#   r#   ;   s&   € ÷ Bñ Bœcð B´#ð B¸$ñ Br$   c                 óè   € \        V4      p\        P                  P                  P	                  WP
                  4       \        P                  P                  P                  WP
                  4       R # r7   )r)   r   r   r8   Ú_cuda_endAllocateToPoolr:   Ú_cuda_releasePoolr;   s   && r"   Úend_mempoolrG   :   sD   € ä# MÓ2€GÜ	‡J�J×Ñ×-Ñ-¨l¿J¹JÔGÜ	‡J�J×Ñ×'Ñ'¨·j±jÖAr$   c                ó4   € V ^8„  d   QhR\         R\         RR/# r2   r5   )r!   s   "r"   r#   r#   B   r>   r$   c                 ó   € R # r7   r.   r@   s   &&r"   rA   rA   A   rB   r$   c                   ó¶   a a€ ] tR t^It oRtV3R lV 3R lltV3R lR ltV3R lR ltV3R lR	 ltR
]	! ]4      /t
V3R lR ltV3R lV 3R lltRtVtV ;t# )ÚCUDAMemPoolVariablez'Represents a torch.cuda.MemPool object.c          
      ó^   <€ V ^8„  d   QhRS[ RS[P                  P                  RS[RS[RR/# )r   ÚproxyÚvalueÚuser_object_indexÚkwargsr   N)r   r   r   r    r   r   )r!   Ú__classdict__s   "€r"   r#   Ú CUDAMemPoolVariable.__annotate__L   sG   ø€ ÷ 
3ñ 
3áð
3ñ �z‰z×!Ñ!ð
3ñ ð	
3ñ
 ð
3ð 
ñ
3r$   c                óL   <€ \         SV `  ! R/ VB  Wn        W n        W0n        R # )Nr.   )ÚsuperÚ__init__rM   rN   rO   )ÚselfrM   rN   rO   rP   Ú	__class__s   &&&&,€r"   rU   ÚCUDAMemPoolVariable.__init__L   s%   ø€ ô 	‰ÒÑ"˜6Ò"ØŒ
ØŒ
Ø!2Ör$   c                ó    <€ V ^8„  d   QhRS[ /# r+   )Útype)r!   rQ   s   "€r"   r#   rR   X   s   ø€ ÷ "ñ "™Tñ "r$   c                ó6   € \         P                  P                  # r7   )r   r   r    ©rV   s   &r"   Úpython_typeÚCUDAMemPoolVariable.python_typeX   s   € Ü�z‰z×!Ñ!Ð!r$   c                ó    <€ V ^8„  d   QhRS[ /# r+   )Úobject)r!   rQ   s   "€r"   r#   rR   [   s   ø€ ÷ ñ ©fñ r$   c                ó   € V P                   # r7   )rN   r\   s   &r"   Úget_real_python_backed_valueÚ0CUDAMemPoolVariable.get_real_python_backed_value[   ó   € Ø�z‰zÐr$   c                ó$   <€ V ^8„  d   QhRRRS[ /# ©r   Útxr   r   ©r   )r!   rQ   s   "€r"   r#   rR   ^   s   ø€ ÷ 6ñ 6Ð1ð 6±oñ 6r$   c                óÞ   € V P                   '       d3   \        V P                   P                  \        P                  4      4       \
        P                  ! V P                  P                  4      # r7   )	Úsourcer   Ú
make_guardr   ÚID_MATCHr   ÚcreaterN   r:   )rV   rg   s   &&r"   Ú_idÚCUDAMemPoolVariable._id^   sA   € Ø�;�;ˆ;Ü˜$Ÿ+™+×0Ñ0´×1FÑ1FÓGÔHÜ×&Ò& t§z¡z§}¡}Ó5Ð5r$   r:   c                ó    <€ V ^8„  d   QhRS[ /# r+   )r   )r!   rQ   s   "€r"   r#   rR   g   s   ø€ ÷ ñ ™%ñ r$   c                ó   € V P                   # r7   )rM   r\   s   &r"   Úas_proxyÚCUDAMemPoolVariable.as_proxyg   rd   r$   c                ó"   <€ V ^8„  d   QhRRRR/# ©r   Úcodegenr   r   Nr.   )r!   rQ   s   "€r"   r#   rR   j   s   ø€ ÷ >ñ > ;ð >°4ñ >r$   c                ó   <a€ V P                   '       d   \        SV `	  S4      # SP                  V3R  l4       SP	                  SP                  V P                  4      4       SP                  \        ^R4      4       R# )c                  ól   <€ S P                  \        P                  P                  P                  R 4      # r   )Úload_import_fromr   Ú_dynamoÚgraph_bytecode_inputsÚ__name__)rv   s   €r"   Ú<lambda>Ú1CUDAMemPoolVariable.reconstruct.<locals>.<lambda>o   s&   ø€ �G×,Ñ,Ü—‘×3Ñ3×<Ñ<Ø.ôr$   FN)	rj   rT   ÚreconstructÚadd_push_nullÚappend_outputÚcreate_load_constrO   Úextend_outputr   )rV   rv   rW   s   &f€r"   r   ÚCUDAMemPoolVariable.reconstructj   sf   ù€ Ø�;�;ˆ;Ü‘7Ñ& wÓ/Ð/à×Ñôô	
ð 	×Ñ˜g×7Ñ7¸×8NÑ8NÓOÔPØ×ÑÔ2°1°eÓ<Ö=r$   )rM   rO   rN   )r|   Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__rU   r]   rb   rn   r   Ú
tp_membersrr   r   Ú__static_attributes__Ú__classdictcell__Ú__classcell__©rW   rQ   s   @@r"   rK   rK   I   s[   ù‡ € Ù1÷
3ó 
3÷"ð "÷ð ÷6ð 6ð 	‰f�S‹kð€J÷ð ÷>÷ >ó >r$   rK   c                   ó¼   a a€ ] tR t^xt oRtV3R lV 3R llt]RV3R lR ll4       tV3R lR ltV3R lR	 lt	V3R
 lR lt
V3R lR ltV3R lR ltRtVtV ;t# )ÚCUDAMemPoolContextVariablez7Represents the torch.cuda.use_mem_pool context manager.c                ó0   <€ V ^8„  d   QhRS[ RS[RS[RR/# )r   r(   r3   rP   r   N)rK   r   r   )r!   rQ   s   "€r"   r#   Ú'CUDAMemPoolContextVariable.__annotate__{   s9   ø€ ÷ Jñ Já$ðJñ ðJñ ð	Jð
 
ñJr$   c                óH   <€ Wn         W n        \        SV `  ! RR RRR/VB  R# )Útarget_valuesÚinitial_valuesNr.   )r(   r3   rT   rU   )rV   r(   r3   rP   rW   s   &&&,€r"   rU   Ú#CUDAMemPoolContextVariable.__init__{   s*   ø€ ð ŒØ(ÔÜ‰ÒÑI rÐI¸$ÐIÀ&ÔIr$   c          
      óB   <€ V ^8„  d   QhRRRS[ RS[ R,          RS[RR/# )	r   rg   r   r(   ÚdeviceNrP   r   r�   )r   r   )r!   rQ   s   "€r"   r#   r‘   †   sG   ø€ ÷ Kñ KØ'ðKá ðKñ   $Õ&ðKñ ð	Kð
 
&ñKr$   c                ó  € \        V\        4      '       g   \        V R 4       Ve%   \        V\        4      '       d^   VP                  fP   \        \        4       P                  \        P                  4      4       \        P                  P                  4       pM`VP                  4       '       d9   \        P                  P                  P                  VP!                  4       4      pM\#        RRV 2R. R7       \%        VX3/ VB # )z@torch.cuda.use_mem_pool() expected a torch.cuda.MemPool argumentz0torch.cuda.use_mem_pool with non-constant devicezdevice=zWDynamo requires the device argument to torch.cuda.use_mem_pool to be a Python constant.©Úgb_typeÚcontextÚexplanationÚhints)r&   rK   r	   r   rN   r   r/   rk   r   ÚEQUALS_MATCHr   r   r-   Úis_python_constantÚ_utilsÚ_get_device_indexÚas_python_constantr
   r�   )rg   r(   r—   rP   r3   s   &&&, r"   rm   Ú!CUDAMemPoolContextVariable.create…   s×   € ô ˜'Ô#6×7Ò7ÜØØRôð
 Š>Ü�vÔ/×0Ò0°V·\±\Ò5IäÜ+Ó-×8Ñ8¼×9RÑ9RÓSôô !Ÿ:™:×4Ñ4Ó6‰LØ×&Ñ&×(Ò(Ü Ÿ:™:×,Ñ,×>Ñ>Ø×)Ñ)Ó+ó‰Lô ØJØ! & Ð*ØuØõ	ô *¨'°<ÑJÀ6ÑJÐJr$   c                ó$   <€ V ^8„  d   QhRRRS[ /# rf   rh   )r!   rQ   s   "€r"   r#   r‘   §   s   ø€ ÷ -ñ -Ð3ð -¹ñ -r$   c                óR  a€ \        4       oSP                  \        P                  P                  P                  R V P                  P                  RV P                  /4      4       SP                  \        P                  P                  P                  4       4       V P                  VV3R l4       VP                  P                  R\        P                  P                  P                  V P                  V P                  P                  3/ 4       \        P                   ! R4      # )r(   Úmempool_devicec                  ó$   <€ S P                  4       # r7   )Úclose)Ústacks   €r"   r}   Ú2CUDAMemPoolContextVariable.enter.<locals>.<lambda>²   s   ø€ ¨%¯+©+¬-r$   Úcall_functionN)r   Úenter_contextr   ÚfxÚ	tracebackÚannotater(   rO   r3   Úpreserve_node_metaÚset_cleanup_hookÚoutputÚcreate_proxyÚopsÚbeginr   rm   )rV   rg   r©   s   &&@r"   ÚenterÚ CUDAMemPoolContextVariable.enter§   sÙ   ø€ Ü“ˆØ×ÑÜ�H‰H×Ñ×'Ñ'à˜tŸ|™|×=Ñ=Ø$ d×&7Ñ&7ðóô	
ð 	×ÑœEŸH™H×.Ñ.×AÑAÓCÔDØ×Ñ˜bÔ"7Ô8Ø
�	‰	×ÑØÜ�I‰I×Ñ×#Ñ#Ø×Ñ §¡× >Ñ >Ð?Øô		
ô  ×&Ò& tÓ,Ð,r$   c                ó*   <€ V ^8„  d   QhRRRS[ RS[ /# )r   rg   r   Úargsr   rh   )r!   rQ   s   "€r"   r#   r‘   »   s%   ø€ ÷ 
-ñ 
-Ø-ð
-Ù6Eð
-á	ñ
-r$   c                ó  € VP                   P                  R \        P                  P                  P
                  V P                  V P                  P                  3/ 4       V P                  4        \        P                  ! R4      # )r«   N)r²   r³   r   r´   r(   Úendr3   rO   Úcleanup_assertr   rm   )rV   rg   r¹   s   &&*r"   ÚexitÚCUDAMemPoolContextVariable.exit»   sf   € ð 	�	‰	×ÑØÜ�I‰I×Ñ×!Ñ!Ø×Ñ §¡× >Ñ >Ð?Øô		
ð 	×ÑÔÜ×&Ò& tÓ,Ð,r$   c                ó    <€ V ^8„  d   QhRS[ /# r+   ©Ústr)r!   rQ   s   "€r"   r#   r‘   Ç   s   ø€ ÷ ñ ™Sñ r$   c                ó   € R # )z
torch.cudar.   r\   s   &r"   Úmodule_nameÚ&CUDAMemPoolContextVariable.module_nameÇ   s   € Ùr$   c                ó    <€ V ^8„  d   QhRS[ /# r+   rÀ   )r!   rQ   s   "€r"   r#   r‘   Ê   s   ø€ ÷ ñ ™ñ r$   c                ó   € R # )Úuse_mem_poolr.   r\   s   &r"   Úfn_nameÚ"CUDAMemPoolContextVariable.fn_nameÊ   s   € Ùr$   c                ó"   <€ V ^8„  d   QhRRRR/# ru   r.   )r!   rQ   s   "€r"   r#   r‘   Í   s   ø€ ÷ 	
ñ 	
¨ð 	
¸ñ 	
r$   c                ó6   € \        R \        V 4      R. R7       R# )z#torch.cuda.use_mem_pool graph breakzCDynamo doesn't support graph breaks inside torch.cuda.use_mem_pool.r™   N)r
   rÁ   )rV   rv   s   &&r"   Úreconstruct_typeÚ+CUDAMemPoolContextVariable.reconstruct_typeÍ   s   € ô 	Ø9Ü˜“IØ]Ø÷		
r$   )r3   r(   r7   )r|   r…   r†   r‡   rˆ   rU   Ústaticmethodrm   r¶   r½   rÃ   rÈ   rÌ   rŠ   r‹   rŒ   r�   s   @@r"   r�   r�   x   sh   ù‡ € ÙA÷Jó Jð ÷Kñ Kó ðK÷B-ð -÷(
-ð 
-÷ð ÷ð ÷	
÷ 	
ð 	
r$   r�   r.   )1Ú
contextlibr   Útypingr   r   r   Útorch.fxr   r   Úbytecode_transformationr   Úexcr	   r
   r{   r   Úguardsr   r   rj   r   r   r   Úbaser   r   Úconstantr   Úctx_managerr   Útorch._dynamo.symbolic_convertr   rv   r   Útorch._library.custom_opsr   r)   r/   r<   Úregister_fakerA   r´   r(   rµ   ÚdefaultrG   r»   rK   r�   r.   r$   r"   Ú<module>rÜ      s  ðÝ  ß %ã ß +å :ß 1Ý @ß 0ß GÑ Gß )Ý &Ý 0÷ ÝHå#å /õõñ Ð¨"Ô-ôWó .ðWð
 ×Ñô	ó ð	ñ �—	‘	×!Ñ!×'Ñ'×/Ñ/Ô 0ñ ˆ>¨Ô+ôBó ,ðBð ×Ñô	ó ð	ñ �—	‘	×!Ñ!×%Ñ%×-Ñ-Ô .ô,>˜/ô ,>ô^^
Ð!8ö ^
r$   