ó
\K]c           @@ sŸ   d  d l  m Z m Z d  d l Z d  d l m Z m Z m Z d  d l	 m
 Z
 m Z d „  Z d „  Z d e e
 j f d „  ƒ  YZ e d	 k r› e
 j ƒ  n  d S(
   i    (   t   print_functiont   absolute_importN(   t   cudat   float32t   int32(   t   unittestt   SerialMixinc         C@ s0   x) t  | j d ƒ D] } |  | | | <q Wd  S(   Ni    (   t   ranget   shape(   t   inpt   outt   i(    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_slicing.pyt   foo   s    c         C@ s@   t  j d ƒ } t |  | d  d  … f | | d  d  … f ƒ d  S(   Ni   (   R   t   gridt   cufoo(   R	   R
   R   (    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_slicing.pyt   copy   s    t   TestCudaSlicingc           B@ s   e  Z d  „  Z RS(   c         C@ sx   t  j d d t ƒt ƒ a t  j d ƒ t ƒ } t j d d t j ƒj	 d d ƒ } t j
 | ƒ } | d | | ƒ d  S(	   Ns   void(int32[:], int32[:])t   devices   void(int32[:,:], int32[:,:])id   t   dtypei
   i   (   i   i
   (   R   t   jitt   TrueR   R   R   t   npt   arangeR   t   reshapet
   zeros_like(   t   selft   cucopyR	   R
   (    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_slicing.pyt   test_slice_as_arg   s
    $(   t   __name__t
   __module__R   (    (    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_slicing.pyR      s   t   __main__(   t
   __future__R    R   t   numpyR   t   numbaR   R   R   t   numba.cuda.testingR   R   R   R   t   TestCaseR   R   t   main(    (    (    sC   lib/python2.7/site-packages/numba/cuda/tests/cudapy/test_slicing.pyt   <module>   s   		