
    PiT                         d dl mZ d dlmZ dgZd Zde_        ed             Zedd            Zed             Z	ed	             Z
ed
             ZdS )    )driver)constexpr_functioncurrent_targetc                  h    	 t           j        } n# t          $ r Y d S w xY w|                                 S )N)r   activeRuntimeErrorget_current_target)active_drivers    s/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/triton/language/target_info.pyr   r      sF       tt ++---s    
Tc                  <    t                      } | d uo
| j        dk    S )Ncudar   backendtargets    r   is_cudar      s$    F:&.F"::    c                     t                      }||j        dk    rdS t          |j        t                    sJ |j        | dz  |z   k    S )z
    Determines whether we have compute capability >= (major, minor) and
    returns this as a constexpr boolean. This can be used for guarding
    inline asm implementations that require a certain compute capability.
    Nr   F
   )r   r   
isinstancearchint)majorminorr   s      r   cuda_capability_geqr      sS     F~611ufk3''''';%"*u,,,r   c                  <    t                      } | d uo
| j        dk    S )Nhipr   r   s    r   is_hipr   '   s$    F9&.E"99r   c                  <    t                      } | d uo
| j        dk    S )Ngfx942r   r   r   s    r   is_hip_cdna3r"   -   $    F9&+"99r   c                  <    t                      } | d uo
| j        dk    S )Ngfx950r!   r   s    r   is_hip_cdna4r&   3   r#   r   N)r   )triton.runtimer   triton.runtime.jitr   __all__r   __triton_builtin__r   r   r   r"   r&    r   r   <module>r,      s    ! ! ! ! ! ! 1 1 1 1 1 1
. . . %) ! ; ; ;
 
- 
- 
- 
- : : :
 : : :
 : : : : :r   