
    Ph                         d dl Z d dlmZ  e j                  d      defd       Z e j                  d      defd       Zy)    N)get_interface_for_devicereturnc                  2    	 dd l } | d uS # t        $ r Y yw xY w)Nr   F)tritonImportError)r   s    ^C:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/utils/_triton.pyhas_triton_packager	      s'    T!! s   
 	c                  .    d }  |        xr
 t               S )Nc                      t        d      } | j                         xr' | j                  j                         j                  dk\  S )Ncuda   )r   is_availableWorkerget_device_propertiesmajor)device_interfaces    r   is_cuda_compatible_with_tritonz2has_triton.<locals>.is_cuda_compatible_with_triton   sA    3F;))+ K ''==?EEJ	
    )r	   )r   s    r   
has_tritonr      s    
 *+D0B0DDr   )	functoolstorch._dynamo.device_interfacer   	lru_cacheboolr	   r    r   r   <module>r      s^     C TD   TED E Er   