U
    ,‰d.  ã                   @   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   ú;/tmp/pip-unpacked-wheel-eu7e0c37/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   r   r   Úhasattrr   r   r   r   r   r   Ú<module>   s    
