o
    i3                     @   sL   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r$e	  dS 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    dtype)to)r   device_arraynpdoubleeventrecord	to_devicearangewaitsynchronizeelapsed_time)selfNdaryevtstartevtend r   b/home/ubuntu/transcripts/venv/lib/python3.10/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   numbar   numba.cuda.testingr   r   r   r   mainr   r   r   r   <module>   s    