
    .`iR                         d dl Z d dlZd dlmZ d dlmZmZmZm	Z	m
Z
 g dZej        d             Zej        d             Zej        d             Zd ZdS )    N)cuda_capability_geqis_cudais_hipis_hip_cdna3is_hip_cdna4)	r   get_cdna_versionhas_tma_gatherhas_native_mxfpr   r   r   r   num_smsc                      t           j                                        } | j        dk    rdS | j        dk    rdS | j        dk    rdS dS )z
    Gets the AMD architecture version, i.e. CDNA3 or CDNA4, currently
    only supports 3 (gfx942) or 4 (gfx950). Returns -1 if it is not AMD
    hardware or unsupported architecture
    hipgfx942   gfx950   )tltarget_infocurrent_targetbackendarch)targets    /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/third_party/triton_kernels/target_info.pyr   r      sR     ^**,,F~r{hq{hq2    c                  "    t          dd          S N
   r   r    r   r   r	   r	   +       r1%%%r   c                  "    t          dd          S r   r   r   r   r   r
   r
   0   r    r   c                  J    t           j                            d          j        S )Nr   )torchcudaget_device_propertiesmulti_processor_countr   r   r   r   r   5   s    :++A..DDr   )r#   tritontriton.languagelanguager   triton.language.target_infor   r   r   r   r   __all__constexpr_functionr   r	   r
   r   r   r   r   <module>r-      s                        
 
 
     & & & & & &E E E E Er   