
    sh                     n    S SK rS SKJr  S SKJrJr   " S S\5      r\S:X  a  \R                  " 5         gg)    N)cuda)unittestCUDATestCasec                       \ rS rSrS rSrg)TestCudaComplex   c                 "   [         R                  " S5      S 5       n[        R                  " S[        R                  S9nUR                  5       nUSUR                  4   " US5        U R                  [        R                  " X#S-   5      5        g )Nzvoid(complex128[:], complex128)c                 H    [         R                  " S5      nX==   U-  ss'   g )N   )r   grid)abis      /Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/cudapy/test_complex_kernel.pyfoo2TestCudaComplex.test_cuda_complex_arg.<locals>.foo   s    		!ADAID       )dtyper   y               @)	r   jitnparange
complex128copyshape
assertTrueallclose)selfr   r   a0s       r   test_cuda_complex_arg%TestCudaComplex.test_cuda_complex_arg   sn    	3	4	 
5	 IIar}}-VVXAqwwJ2ABw/0r    N)__name__
__module____qualname____firstlineno__r    __static_attributes__r"   r   r   r   r      s    	1r   r   __main__)
numpyr   numbar   numba.cuda.testingr   r   r   r#   mainr"   r   r   <module>r-      s2      5
1l 
1 zMMO r   