o
    i.                     @   s   d dl mZmZmZmZ d dlmZmZmZ d dl	m
Z
mZ d dlmZmZmZmZmZmZ ee
dr6e
jZ[
[dd Ze  [dS )	    )	VectorizeGUVectorize	vectorizeguvectorize)PyUFunc_NonePyUFunc_ZeroPyUFunc_One)	_internalarray_exprs)threading_layerget_num_threadsset_num_threadsget_thread_idset_parallel_chunksizeget_parallel_chunksizePyUFunc_ReorderableNonec                  C   s,   dd } dd }| t jjd< |tjjd< d S )Nc                  S      ddl m}  | S )Nr   CUDAVectorize)numba.cuda.vectorizersr   r    r   T/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/np/ufunc/__init__.pyinit_cuda_vectorize      z"_init.<locals>.init_cuda_vectorizec                  S   r   )Nr   CUDAGUFuncVectorize)r   r   r   r   r   r   init_cuda_guvectorize   r   z$_init.<locals>.init_cuda_guvectorizecuda)r   target_registryondemandr   )r   r   r   r   r   _init   s   r    N)numba.np.ufunc.decoratorsr   r   r   r   numba.np.ufunc._internalr   r   r   numba.np.ufuncr	   r
   numba.np.ufunc.parallelr   r   r   r   r   r   hasattrr   r    r   r   r   r   <module>   s    
