
\K]c           @@ s   d  d l  m Z m Z d  d l Z d  d l m Z d  d l m Z m	 Z	 d e	 e j
 f d     YZ e d k r} e j   n  d S(   i    (   t   absolute_importt   print_functionN(   t   cuda(   t   unittestt   SerialMixint   TestCudaEventc           B@ s   e  Z d    Z d   Z RS(   c         C@ s   d } t  j | d t j } t  j   } t  j   } | j   t  j t j | d t j d | | j   | j   | j	   | j
 |  d  S(   Ni    t   dtypet   to(   R   t   device_arrayt   npt   doublet   eventt   recordt	   to_devicet   aranget   waitt   synchronizet   elapsed_time(   t   selft   Nt   daryt   evtstartt   evtend(    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_events.pyt   test_event_elapsed   s    
%


c         C@ s   d } t  j   } t  j | d t j } t  j   } t  j   } | j d |  t  j t j | d t j d | d | | j d |  | j	 d |  | j
   | j |  d  S(   Ni    R   t   streamR   (   R   R   R   R	   R
   R   R   R   R   R   R   R   (   R   R   R   R   R   R   (    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_events.pyt   test_event_elapsed_stream   s    +
(   t   __name__t
   __module__R   R   (    (    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_events.pyR      s   	t   __main__(   t
   __future__R    R   t   numpyR	   t   numbaR   t   numba.cuda.testingR   R   t   TestCaseR   R   t   main(    (    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_events.pyt   <module>   s   