
    sh3                     n    S SK rS SKJr  S SKJrJr   " S S\5      r\S:X  a  \R                  " 5         gg)    N)cuda)unittestCUDATestCasec                        \ rS rSrS rS rSrg)TestCudaEvent   c                    Sn[         R                  " U[        R                  S9n[         R                  " 5       n[         R                  " 5       nUR                  5         [         R                  " [        R                  " U[        R                  S9US9  UR                  5         UR                  5         UR                  5         UR                  U5        g )N    dtype)to)r   device_arraynpdoubleeventrecord	to_devicearangewaitsynchronizeelapsed_time)selfNdaryevtstartevtends        x/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudadrv/test_events.pytest_event_elapsed TestCudaEvent.test_event_elapsed   s      "))4::<ryy"))4>f%    c                    Sn[         R                  " 5       n[         R                  " U[        R                  S9n[         R
                  " 5       n[         R
                  " 5       nUR                  US9  [         R                  " [        R                  " U[        R                  S9X2S9  UR                  US9  UR                  US9  UR                  5         UR                  U5        g )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_stream'TestCudaEvent.test_event_elapsed_stream   s      "))4::<v&ryy"))4MV$6"f%r     N)__name__
__module____qualname____firstlineno__r   r#   __static_attributes__r%   r    r   r   r      s    &&r    r   __main__)
numpyr   numbar   numba.cuda.testingr   r   r   r&   mainr%   r    r   <module>r0      s2      5&L &> zMMO r    