U
    ?h.                     @   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rle
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   s   ddl m}  | S )Nr   CUDAVectorize)numba.cuda.vectorizersr   r    r   I/var/www/html/venv/lib/python3.8/site-packages/numba/np/ufunc/__init__.pyinit_cuda_vectorize   s    z"_init.<locals>.init_cuda_vectorizec                  S   s   ddl m}  | S )Nr   CUDAGUFuncVectorize)r   r   r   r   r   r   init_cuda_guvectorize   s    z$_init.<locals>.init_cuda_guvectorizecuda)r   Ztarget_registryZondemandr   )r   r   r   r   r   _init   s    r   N)Znumba.np.ufunc.decoratorsr   r   r   r   Znumba.np.ufunc._internalr   r   r   Znumba.np.ufuncr	   r
   Znumba.np.ufunc.parallelr   r   r   r   r   r   hasattrr   r   r   r   r   r   <module>   s    
