o
    ÅÙ¾iR  ã                   @   sv   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d„ ƒZejdd„ ƒZejdd	„ ƒZd
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                  C   s8   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)Útarget© r   úN/home/ubuntu/.local/lib/python3.10/site-packages/triton_kernels/target_info.pyr      s   



r   c                   C   ó
   t ddƒS ©Né
   r   ©r   r   r   r   r   r   +   ó   
r   c                   C   r   r   r   r   r   r   r   r	   0   r   r	   c                   C   s   t j d¡jS )Nr   )ÚtorchÚcudaÚget_device_propertiesÚmulti_processor_countr   r   r   r   r
   5   s   r
   )r   ÚtritonÚtriton.languageÚlanguager   Útriton.language.target_infor   r   r   r   r   Ú__all__Úconstexpr_functionr   r   r	   r
   r   r   r   r   Ú<module>   s    


