§
    .`ƒip  ã                   ó|   — d dl mZ d dlZd dlmZ  G d„ dee¦  «        Z G d„ de¦  «        Z G d„ d	e¦  «        ZdS )
é    )ÚABCN)ÚLoadStoreSpecc                   ó6   — e Zd ZdZdee         fd„Zdefd„ZdS )ÚBlockIDsLoadStoreSpeczF
    Spec for loading/storing KV blocks from given block numbers.
    Ú	block_idsc                 óP   — t          j        |t           j        ¬¦  «        | _        d S )N)Údtype)ÚnpÚarrayÚint64r   )Úselfr   s     ún/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/v1/kv_offload/mediums.pyÚ__init__zBlockIDsLoadStoreSpec.__init__   s   € Ýœ )µ2´8Ð<Ñ<Ô<ˆŒˆˆó    Úreturnc                 ó*   — t          | j        ¦  «        S )N)Úreprr   )r   s    r   Ú__repr__zBlockIDsLoadStoreSpec.__repr__   s   € ÝD”NÑ#Ô#Ð#r   N)	Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚlistÚintr   Ústrr   © r   r   r   r   
   sX   € € € € € ðð ð= $ s¤)ð =ð =ð =ð =ð$˜#ð $ð $ð $ð $ð $ð $r   r   c                   ó.   — e Zd ZdZedefd„¦   «         ZdS )ÚGPULoadStoreSpecz<
    Spec for loading/storing a KV block to GPU memory.
    r   c                  ó   — dS )NÚGPUr   r   r   r   ÚmediumzGPULoadStoreSpec.medium   ó   € àˆur   N©r   r   r   r   Ústaticmethodr   r!   r   r   r   r   r      óE   € € € € € ðð ð ðCð ð ð ñ „\ðð ð r   r   c                   ó.   — e Zd ZdZedefd„¦   «         ZdS )ÚCPULoadStoreSpecz<
    Spec for loading/storing a KV block to CPU memory.
    r   c                  ó   — dS )NÚCPUr   r   r   r   r!   zCPULoadStoreSpec.medium%   r"   r   Nr#   r   r   r   r'   r'       r%   r   r'   )	Úabcr   Únumpyr
   Úvllm.v1.kv_offload.abstractr   r   r   r'   r   r   r   ú<module>r-      sÃ   ðð Ð Ð Ð Ð Ð à Ð Ð Ð à 5Ð 5Ð 5Ð 5Ð 5Ð 5ð	$ð 	$ð 	$ð 	$ð 	$˜M¨3ñ 	$ô 	$ð 	$ðð ð ð ð Ð,ñ ô ð ðð ð ð ð Ð,ñ ô ð ð ð r   