
    `i                     t    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	                     dS dS )    N)cuda)unittestCUDATestCasec                       e Zd Zd ZdS )TestCudaComplexc                 0   t          j        d          d             }t          j        dt          j                  }|                                } |d|j        f         |d           |                     t          j        ||dz                        d S )Nzvoid(complex128[:], complex128)c                 N    t          j        d          }| |xx         |z  cc<   d S )N   )r   grid)abis      /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/numba/cuda/tests/cudapy/test_complex_kernel.pyfooz2TestCudaComplex.test_cuda_complex_arg.<locals>.foo   s(    	!AaDDDAIDDDDD       )dtyper
   y               @)	r   jitnparange
complex128copyshape
assertTrueallclose)selfr   r   a0s       r   test_cuda_complex_argz%TestCudaComplex.test_cuda_complex_arg   s    	3	4	4	 	 
5	4	 Iar}---VVXXAqwJ2ArBw//00000r   N)__name__
__module____qualname__r    r   r   r   r      s#        	1 	1 	1 	1 	1r   r   __main__)
numpyr   numbar   numba.cuda.testingr   r   r   r   mainr"   r   r   <module>r(      s              5 5 5 5 5 5 5 5
1 
1 
1 
1 
1l 
1 
1 
1 zHMOOOOO r   