U
    9%e3                     @   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   c/var/www/html/Darija-Ai-API/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
   