§
    ’Þj“  ã                   óT   — d Z ddlZddlmZ  ej        ¦   «         ZdZd„ Zd	d„Zd„ Z	dS )
zLCoordinate reclaimable VRAM: optional expert copies precede streamed blocks.é    Né   )Ú	allocatori   c           	      óF  — t          j        |¦  «        }|| t          z   k     r| t          z   |z
  n| } d}t          t	          t
                               ¦   «         ¦  «        d„ ¬¦  «        D ]6\  }}||                     t          d| |z
  ¦  «        ¦  «        z  }|| k    r nŒ7|S )Nr   c                 ó   — | d         S )Nr   © )Úitems    ú<C:\Users\johnn\OneDrive\Documents\AI\mmgp\memory_pressure.pyú<lambda>z_reclaim.<locals>.<lambda>   s
   € ÀTÈ!ÄW€ ó    )Úkey)	r   ÚroomÚHEADROOMÚsortedÚlistÚ_clientsÚitemsÚon_pressureÚmax)ÚsizeÚdevicer   ÚfreedÚclientÚ_s         r	   Ú_reclaimr      s«   € åŒ>˜&Ñ!Ô!€DØ%)¨Dµ8©OÒ%;Ð%;ˆ4•(‰?˜TÑ!Ð!À€DØ€EÝ�D¥§¢Ñ!1Ô!1Ñ2Ô2Ð8LÐ8LÐMÑMÔMð ð ‰	ˆ�Ø�×#Ò#¥C¨¨4°%©<Ñ$8Ô$8Ñ9Ô9Ñ9ˆØ�DŠ=ˆ=ØˆEð à€Lr   c                 óp   — |t           | <   t          j        dk    rt          j        t          ¦  «         d S d S ©NÚvmm)r   r   ÚactiveÚset_pressure_callbackr   )r   Úprioritys     r	   Úregisterr!      s9   € Ø…HˆVÑÝÔ˜5Ò Ð ÝÔ'­Ñ1Ô1Ð1Ð1Ð1ð !Ð r   c                 óš   — t                                | d ¦  «         t           s&t          j        dk    rt          j        d ¦  «         d S d S d S r   )r   Úpopr   r   r   )r   s    r	   Ú
unregisterr$      sR   € Ý‡L‚L�˜ÑÔÐÝð .�	Ô(¨EÒ1Ð1ÝÔ'¨Ñ-Ô-Ð-Ð-Ð-ð.ð .Ð1Ð1r   )r   )
Ú__doc__ÚweakrefÚ r   ÚWeakKeyDictionaryr   r   r   r!   r$   r   r   r	   ú<module>r)      sw   ðà RÐ RØ €€€à Ð Ð Ð Ð Ð à$ˆ7Ô$Ñ&Ô&€Ø€ð	ð 	ð 	ð2ð 2ð 2ð 2ð.ð .ð .ð .ð .r   