
    
f                     .    d dl mZmZ d dlmZ d Zd ZdS )    )devicesdriver)
cpu_targetc                 D    t           j        }|                    |           S )z,
    Use the ABI size in the CPU target
    )r   target_contextcalc_array_sizeof)ndimctxs     :lib/python3.11/site-packages/numba/cuda/cudadrv/ndarray.py_calc_array_sizeofr      s      
#C  &&&    c                 z    t          j        |           }t          j                                        |          }|S )z"
    Allocate gpu data buffer
    )r   host_memory_sizer   get_contextmemalloc)arydatasizegpu_datas      r   ndarray_device_allocate_datar      s5     &s++H"$$--h77HOr   N)numba.cuda.cudadrvr   r   numba.core.registryr   r   r    r   r   <module>r      sW    . . . . . . . . * * * * * *' ' '    r   