U
    ,‰dØ  ã                   @   sL   d dl mZ d dlmZ d dlmZ d dlmZ d dlmZm	Z	 dd„ Z
dS )	é    )Ú ensure_supported_ccs_initialized)Úunittest)Úload_testsuite)Úcuda)ÚdirnameÚjoinc                 C   s¾   t  ¡ }ttƒ}tƒ  | t| t|dƒƒ¡ t 	¡ r²| t| t|dƒƒ¡ t 
¡ }|r¨|d jdkr¨| t| t|dƒƒ¡ | t| t|dƒƒ¡ | t| t|dƒƒ¡ qºtdƒ ntd	ƒ |S )
NZnocudaZcudasimr   )é   r   ZcudadrvZcudapyZdoc_examplesz'skipped CUDA tests because GPU CC < 2.0zskipped CUDA tests)r   Z	TestSuiter   Ú__file__r   ZaddTestsr   r   r   Zis_availableZlist_devicesZcompute_capabilityÚprint)ÚloaderÚtestsÚpatternZsuiteZthis_dirZgpus© r   ú=/tmp/pip-unpacked-wheel-eu7e0c37/numba/cuda/tests/__init__.pyÚ
load_tests   s    
r   N)Znumba.cuda.testingr   Znumba.testingr   r   Znumbar   Úos.pathr   r   r   r   r   r   r   Ú<module>   s
   