o
    YÛ·iØ  ã                   @   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 
¡ }|rU|d jdkrU| t| t|dƒƒ¡ | t| t|dƒƒ¡ | t| t|dƒƒ¡ |S tdƒ |S td	ƒ |S )
NÚnocudaÚcudasimr   )é   r   ÚcudadrvÚcudapyÚdoc_examplesz'skipped CUDA tests because GPU CC < 2.0zskipped CUDA tests)r   Ú	TestSuiter   Ú__file__r   ÚaddTestsr   r   r   Úis_availableÚlist_devicesÚcompute_capabilityÚprint)ÚloaderÚtestsÚpatternÚsuiteÚthis_dirÚgpus© r   úO/home/ubuntu/vllm_env/lib/python3.10/site-packages/numba/cuda/tests/__init__.pyÚ
load_tests   s    ýÿr   N)Únumba.cuda.testingr   Únumba.testingr   r   Únumbar   Úos.pathr   r   r   r   r   r   r   Ú<module>   s    