o
    i                     @   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   target_contextcalc_array_sizeof)ndimctx r	   W/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/cuda/cudadrv/ndarray.py_calc_array_sizeof   s   
r   c                 C   s   t | }t |}|S )z"
    Allocate gpu data buffer
    )r   host_memory_sizer   get_contextmemalloc)arydatasizegpu_datar	   r	   r
   ndarray_device_allocate_data   s   
r   N)numba.cuda.cudadrvr   r   numba.core.registryr   r   r   r	   r	   r	   r
   <module>   s    