U
    ?hŁ  ć                   @   s0   d dl mZmZ d dlmZ dd Zdd ZdS )é    )ŚdevicesŚdriver)Ś
cpu_targetc                 C   s   t j}| | ”S )z,
    Use the ABI size in the CPU target
    )r   Ztarget_contextZcalc_array_sizeof)ŚndimŚctx© r   śL/var/www/html/venv/lib/python3.8/site-packages/numba/cuda/cudadrv/ndarray.pyŚ_calc_array_sizeof   s    r	   c                 C   s   t  | ”}t ”  |”}|S )z"
    Allocate gpu data buffer
    )r   Zhost_memory_sizer   Zget_contextZmemalloc)ZaryZdatasizeZgpu_datar   r   r   Śndarray_device_allocate_data   s    
r
   N)Znumba.cuda.cudadrvr   r   Znumba.core.registryr   r	   r
   r   r   r   r   Ś<module>   s   