U
    hâËd3  ã                   @   sH   d dl Zd dlmZ d dlmZmZ G dd„ deƒZedkrDe 	¡  dS )é    N)Úcuda)ÚunittestÚCUDATestCasec                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTestCudaEventc                 C   sl   d}t j|tjd�}t  ¡ }t  ¡ }| ¡  t jtj|tjd�|d� | ¡  | ¡  | 	¡  | 
|¡ d S )Né    ©Zdtype)Úto)r   Údevice_arrayÚnpÚdoubleÚeventÚrecordÚ	to_deviceÚarangeÚwaitÚsynchronizeÚelapsed_time)ÚselfÚNÚdaryÚevtstartÚevtend© r   ú]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/cuda/tests/cudadrv/test_events.pyÚtest_event_elapsed   s    z TestCudaEvent.test_event_elapsedc                 C   s‚   d}t  ¡ }t j|tjd�}t  ¡ }t  ¡ }|j|d� t jtj|tjd�||d� |j|d� |j	|d� | 
¡  | |¡ d S )Nr   r   )Ústream)r   r   )r   r   r	   r
   r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r   Útest_event_elapsed_stream   s    z'TestCudaEvent.test_event_elapsed_streamN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r      s   r   Ú__main__)
Únumpyr
   Znumbar   Znumba.cuda.testingr   r   r   r   Úmainr   r   r   r   Ú<module>   s
   