
    wg                     l    d dl Zd dlmZ d dlmZmZ  G d de      Zedk(  r ej                          yy)    N)cuda)unittestCUDATestCasec                       e Zd Zd Zy)TestCudaComplexc                 $   t        j                  d      d        }t        j                  dt        j                        }|j                         } |d|j                  f   |d       | j                  t        j                  ||dz                y )Nzvoid(complex128[:], complex128)c                 H    t        j                  d      }| |xx   |z  cc<   y )N   )r   grid)abis      p/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/numba/cuda/tests/cudapy/test_complex_kernel.pyfooz2TestCudaComplex.test_cuda_complex_arg.<locals>.foo   s    		!AaDAID       )dtyper
   y               @)	r   jitnparange
complex128copyshape
assertTrueallclose)selfr   r   a0s       r   test_cuda_complex_argz%TestCudaComplex.test_cuda_complex_arg   sr    	3	4	 
5	 IIar}}-VVXAqwwJ2ArBw/0r   N)__name__
__module____qualname__r    r   r   r   r      s    	1r   r   __main__)
numpyr   numbar   numba.cuda.testingr   r   r   r   mainr"   r   r   <module>r(      s4      5
1l 
1 zHMMO r   