o
    *Ñ¹gy  ã                   @   sH   d dl Z d dlmZ e  d¡defdd„ƒZe  d¡defdd„ƒZdS )é    N)Úget_interface_for_deviceÚreturnc                  C   s(   z	dd l } | d uW S  ty   Y dS w )Nr   F)ÚtritonÚImportError)r   © r   úW/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/site-packages/torch/utils/_triton.pyÚhas_triton_package   s   
ÿr   c                  C   s   dd„ } | ƒ o	t ƒ S )Nc                  S   s    t dƒ} |  ¡ o| j ¡ jdkS )NZcudaé   )r   Zis_availableZWorkerZget_device_propertiesÚmajor)Zdevice_interfacer   r   r   Úis_cuda_compatible_with_triton   s   þz2has_triton.<locals>.is_cuda_compatible_with_triton)r   )r   r   r   r   Ú
has_triton   s   r   )Ú	functoolsZtorch._dynamo.device_interfacer   Ú	lru_cacheÚboolr   r   r   r   r   r   Ú<module>   s    	