
    .`i                         d dl Z d dlmZ dZdZdZ ed           G d d	                      Z ed           G d
 de                      Z ed           G d de                      ZdS )    N)	dataclassg      @g      |@g      n@T)frozenc                   J    e Zd ZU dZej        dz  ed<   dej        fdZd ZdS )BaseFlexDataNdtypexc                 H    | j         |S |                    | j                   S N)r   viewselfr   s     |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/third_party/triton_kernels/numerics.pyr   zBaseFlexData.view   s#    :Hvvdj!!!    c                 h    | j         |j         j        dk    r|S |                    | j                   S )N   )r   itemsizer   r   s     r   reinterpretzBaseFlexData.reinterpret   s2    :!1A!5!5Hvvdj!!!r   )	__name__
__module____qualname__r   torch__annotations__Tensorr   r    r   r   r   r   	   sY          $E5;$$$"el " " " "
" " " " "r   r   c                   D    e Zd ZU dZej        dz  ed<   ed             ZdS )
InFlexDataNscalec                 D    | j         dnt          | j                   dk    S )NFr   )r   lenr   s    r   is_per_batchzInFlexData.is_per_batch   s     
*uuDJ!0CCr   )	r   r   r   r   r   r   r   propertyr!   r   r   r   r   r      sK         !%E5<$%%%D D XD D Dr   r   c                   p    e Zd ZU dZej        dz  ed<   dZej        dz  ed<   dZej        dz  ed<   d Z	dS )OutFlexDataNexpected_scaleactual_scalechecksum_scalec              #   @   K   | j         V  | j        V  | j        V  d S r
   )r%   r&   r'   r    s    r   __iter__zOutFlexData.__iter__'   s9      !!!!!!!!!!r   )
r   r   r   r%   r   r   r   r&   r'   r)   r   r   r   r$   r$   !   sj         *.NEL4'...(,L%,%,,,*.NEL4'..." " " " "r   r$   )	r   dataclassesr   MAX_FINITE_FLOAT8E5MAX_FINITE_FLOAT8E4NVMAX_FINITE_FLOAT8E4B8r   r   r$   r   r   r   <module>r.      s    ! ! ! ! ! !    $" " " " " " " " $D D D D D D D D $" " " " ", " " " " "r   