ó
\K]c           @` sX   d  d l  m Z m Z m Z d d l m Z m Z d  d l m Z d „  Z	 d „  Z
 d S(   i    (   t   print_functiont   absolute_importt   divisioni   (   t   devicest   driver(   t
   cpu_targetc         C` s   t  j } | j |  ƒ S(   s,   
    Use the ABI size in the CPU target
    (   R   t   target_contextt   calc_array_sizeof(   t   ndimt   ctx(    (    s9   lib/python2.7/site-packages/numba/cuda/cudadrv/ndarray.pyt   _calc_array_sizeof   s    	c         C` s(   t  j |  ƒ } t j ƒ  j | ƒ } | S(   s"   
    Allocate gpu data buffer
    (   R   t   host_memory_sizeR   t   get_contextt   memalloc(   t   aryt   datasizet   gpu_data(    (    s9   lib/python2.7/site-packages/numba/cuda/cudadrv/ndarray.pyt   ndarray_device_allocate_data   s    N(   t
   __future__R    R   R   t    R   R   t   numba.targets.registryR   R
   R   (    (    (    s9   lib/python2.7/site-packages/numba/cuda/cudadrv/ndarray.pyt   <module>   s   	