o
    ˜à·iY  ã                   @   sd   d dl mZ d dlmZ d dlmZmZ G dd„ deƒZeG dd„ deƒƒZG dd	„ d	ee ƒZ	d
S )é    )Ú	dataclass)ÚAttentionBackend)ÚBaseMambaAttentionMetadataÚ!BaseMambaAttentionMetadataBuilderc                   @   s4   e Zd Zedefdd„ƒZeded fdd„ƒZdS )ÚMamba1AttentionBackendÚreturnc                   C   s   dS )NÚMAMBA1_ATTN© r	   r	   r	   ú\/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm/v1/attention/backends/mamba1_attn.pyÚget_name   ó   zMamba1AttentionBackend.get_nameÚMamba1AttentionMetadataBuilderc                   C   s   t S )N)r   r	   r	   r	   r
   Úget_builder_cls   r   z&Mamba1AttentionBackend.get_builder_clsN)Ú__name__Ú
__module__Ú__qualname__ÚstaticmethodÚstrr   Útyper   r	   r	   r	   r
   r      s
    r   c                   @   s   e Zd ZdS )ÚMamba1AttentionMetadataN)r   r   r   r	   r	   r	   r
   r      s    r   c                   @   s   e Zd ZU eZdZeed< dS )r   FÚsupports_update_block_tableN)r   r   r   r   Úmetadata_clsr   ÚboolÚ__annotations__r	   r	   r	   r
   r      s   
 r   N)
Údataclassesr   Úvllm.v1.attention.backendr   Ú%vllm.v1.attention.backends.mamba_attnr   r   r   r   r   r	   r	   r	   r
   Ú<module>   s   

ÿ