a
    ò.;bµ  ã                   @   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 ee
dƒrde
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Ú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   ú6lib/python3.9/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_guvectorizeZ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   Úhasattrr   r   r   r   r   r   Ú<module>   s   
