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	   c/home/ubuntu/veenaModal/venv/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   

