
    xKg3                     l    d dl Zd dlmZ d dlmZmZ  G d de      Zedk(  r ej                          yy)    N)cuda)unittestCUDATestCasec                       e Zd Zd Zd Zy)TestCudaEventc                    d}t        j                  |t        j                        }t        j                         }t        j                         }|j                          t        j                  t        j                  |t        j                        |       |j                          |j                          |j                          |j                  |       y )N    dtype)to)r   device_arraynpdoubleeventrecord	to_devicearangewaitsynchronizeelapsed_time)selfNdaryevtstartevtends        h/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/cuda/tests/cudadrv/test_events.pytest_event_elapsedz TestCudaEvent.test_event_elapsed   s      "))4::<ryy"))4>f%    c                    d}t        j                         }t        j                  |t        j                        }t        j
                         }t        j
                         }|j                  |       t        j                  t        j                  |t        j                        ||       |j                  |       |j                  |       |j                          |j                  |       y )Nr	   r
   )stream)r   r    )r   r    r   r   r   r   r   r   r   r   r   r   )r   r   r    r   r   r   s         r   test_event_elapsed_streamz'TestCudaEvent.test_event_elapsed_stream   s      "))4::<v&ryy"))4fMV$6"f%r   N)__name__
__module____qualname__r   r!    r   r   r   r      s    &&r   r   __main__)
numpyr   numbar   numba.cuda.testingr   r   r   r"   mainr%   r   r   <module>r+      s4      5&L &> zHMMO r   