o
    8ήc                     @   sx   d dl Zd dlmZ d dlmZmZ d dlmZmZ d dl	Z	eeegZ
edG dd deZedkr:e	  dS dS )	    N)	vectorize)cudafloat64)skip_on_cudasimCUDATestCasez&ufunc API unsupported in the simulatorc                   @   s   e Zd Zdd Zdd ZdS )TestCUDAVectorizeScalarArgc                 C   s\   t tdddd }tjdtjd}t|}|d|}tj|	 tjdd	tjd d S )
Nr   targetc                 S      | | S N abr   r   X/tmp/pip-target-vg8gfxp4/lib/python/numba/cuda/tests/cudapy/test_vectorize_scalar_arg.py
vector_add      zHTestCUDAVectorizeScalarArg.test_vectorize_scalar_arg.<locals>.vector_add
   )dtype      ?      )
r   signparanger   r   	to_devicetestingassert_array_almost_equalcopy_to_host)selfr   AdAvr   r   r   test_vectorize_scalar_arg   s   



z4TestCUDAVectorizeScalarArg.test_vectorize_scalar_argc                 C   s0   t tdddd }|dd}tjd| d S )Nr   r   c                 S   r
   r   r   r   r   r   r   r      r   zITestCUDAVectorizeScalarArg.test_vectorize_all_scalars.<locals>.vector_addr   g       @)r   r   r   r   assert_almost_equal)r   r   r"   r   r   r   test_vectorize_all_scalars   s   


z5TestCUDAVectorizeScalarArg.test_vectorize_all_scalarsN)__name__
__module____qualname__r#   r%   r   r   r   r   r   
   s    r   __main__)numpyr   numbar   r   r   numba.cuda.testingr   r   unittestr   r   r&   mainr   r   r   r   <module>   s    