
    .`iY                         d dl mZ d dlmZ d dlmZmZ  G d de          Ze G d de                      Z G d d	ee                   Z	d
S )    )	dataclass)AttentionBackend)BaseMambaAttentionMetadata!BaseMambaAttentionMetadataBuilderc                   R    e Zd Zedefd            Zeded         fd            ZdS )Mamba1AttentionBackendreturnc                      dS )NMAMBA1_ATTN r       z/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/v1/attention/backends/mamba1_attn.pyget_namezMamba1AttentionBackend.get_name   s    }r   Mamba1AttentionMetadataBuilderc                      t           S )N)r   r   r   r   get_builder_clsz&Mamba1AttentionBackend.get_builder_cls   s    --r   N)__name__
__module____qualname__staticmethodstrr   typer   r   r   r   r   r      sf        c    \ .T"BC . . . \. . .r   r   c                       e Zd ZdS )Mamba1AttentionMetadataN)r   r   r   r   r   r   r   r      s        Dr   r   c                   "    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*          +L(------r   r   N)
dataclassesr   vllm.v1.attention.backendr   %vllm.v1.attention.backends.mamba_attnr   r   r   r   r   r   r   r   <module>r#      s    " ! ! ! ! ! 6 6 6 6 6 6       . . . . .- . . . 	 	 	 	 	8 	 	 	. . . . .%&=>. . . . .r   