ó
\K]c           @@ s€   d  d l  m Z m Z d  d l Z d  d l j Z d  d l m	 Z	 d „  Z
 d e j f d „  ƒ  YZ e d k r| e j ƒ  n  d S(   i    (   t   print_functiont   absolute_importN(   t   rocc         C@ s3   t  j d ƒ } | |  j k  r/ | | |  | <n  d  S(   Ni    (   R   t   get_global_idt   size(   t   outt   inpt   i(    (    sA   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_autojit.pyt   copy_kernel	   s    t   TestAutoJitc           B@ s   e  Z d  „  Z d „  Z RS(   c         C@ s]   t  j t ƒ } t j d ƒ } t j | ƒ } | j | j ƒ | | ƒ t j j	 | | ƒ d  S(   Ni
   (
   R   t   jitR   t   npt   aranget
   zeros_liket   forallR   t   testingt   assert_equal(   t   selft   kernelR   R   (    (    sA   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_autojit.pyt   test_autojit_kernel   s
    c         @ s­   t  j d t ƒ d „  ƒ ‰  t  j ‡  f d †  ƒ } t j d ƒ } | j ƒ  } t j d ƒ } | j | j ƒ | | ƒ |  j t j	 | | k ƒ ƒ t j
 j | | | ƒ d  S(   Nt   devicec         S@ s   |  | S(   N(    (   t   at   b(    (    sA   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_autojit.pyt   inner   s    c         @ s@   t  j d ƒ } | |  j k  r< ˆ  |  | | | ƒ |  | <n  d  S(   Ni    (   R   R   R   (   t   At   BR   (   R   (    sA   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_autojit.pyt   outer   s    i
   (   R   R
   t   TrueR   R   t   copyR   R   t   assertFalset   allR   R   (   R   R   R   t   AorigR   (    (   R   sA   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_autojit.pyt   test_autojit_device   s    (   t   __name__t
   __module__R   R    (    (    (    sA   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_autojit.pyR	      s   	t   __main__(   t
   __future__R    R   t   numpyR   t   numba.unittest_supportt   unittest_supportt   unittestt   numbaR   R   t   TestCaseR	   R!   t   main(    (    (    sA   lib/python2.7/site-packages/numba/roc/tests/hsapy/test_autojit.pyt   <module>   s   	