o
    ۾i                     @   sL   d dl Zd dlmZ d dlmZmZ G dd deZedkr$e	  dS dS )    N)cuda)unittestCUDATestCasec                   @   s   e Zd Zdd ZdS )TestCudaComplexc                 C   sX   t ddd }tjdtjd}| }|d|jf |d | t||d  d S )Nzvoid(complex128[:], complex128)c                 S   s   t d}| |  |7  < d S )N   )r   grid)abi r   _/home/ubuntu/.local/lib/python3.10/site-packages/numba/cuda/tests/cudapy/test_complex_kernel.pyfoo   s   
z2TestCudaComplex.test_cuda_complex_arg.<locals>.foo   )dtyper   y               @)	r   jitnparange
complex128copyshape
assertTrueallclose)selfr   r   a0r   r   r   test_cuda_complex_arg   s   
z%TestCudaComplex.test_cuda_complex_argN)__name__
__module____qualname__r   r   r   r   r   r      s    r   __main__)
numpyr   numbar   numba.cuda.testingr   r   r   r   mainr   r   r   r   <module>   s    