o
    e                     @   s   d dl Z d dlZz
d dlmZ dZW n ey   dZY nw dd Zdd Ze j	j
ed	d
e j	jddddd Ze j	j
ed	d
e j	jddddd ZdS )    N)cudaFTc                 C   s   | dd|f   d S N    )kernelstreamr   r   JD:\Projects\ConvertPro\env\Lib\site-packages\cuda/benchmarks/test_numba.pylaunch_empty   s   r	   c                 C   s   | dd|f | d S r   r   )r   r   argr   r   r   launch   s   r   zNumba is not installed)reasonnumbai  )groupZ
min_roundsc                 C   s.   t  }t jdd }| t|| t   d S )Nc                   S   s   d S )Nr   r   r   r   r   empty_kernel   s   z6test_launch_latency_empty_kernel.<locals>.empty_kernel)r   r   jitr	   synchronize)	benchmarkr   r   r   r   r    test_launch_latency_empty_kernel   s
   
r   c                 C   sB   t  }t jdtj|d}t jdd }| t||| t   d S )Nr   )Zdtyper   c                 S   s   d| d< d S )Ng        r   r   )arrayr   r   r   small_kernel,   s   z6test_launch_latency_small_kernel.<locals>.small_kernel)r   r   Zdevice_arraynpZfloat32r   r   r   )r   r   r
   r   r   r   r    test_launch_latency_small_kernel%   s   
r   )Zpytestnumpyr   r   r   Z
skip_testsImportErrorr	   r   markZskipifr   r   r   r   r   r   r   <module>   s    