
    sh                         S SK r S SKJrJr  S SKJr  \" S5       " S S\5      5       r\S:X  a  \ R                  " 5         gg)    N)CUDATestCaseskip_on_cudasim)captured_stdoutz4cudasim doesn't support cuda import at non-top-levelc                   >   ^  \ rS rSrSrU 4S jrU 4S jrS rSrU =r	$ )
TestVecAdd   z
Test simple vector addition
c                 t   > [        5       U l        U R                  R                  5         [        TU ]  5         g N)r   _captured_stdout	__enter__supersetUpself	__class__s    }/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/tests/doc_examples/test_vecadd.pyr   TestVecAdd.setUp   s+     !0 1'')    c                 \   > U R                   R                  S S S 5        [        TU ]  5         g r
   )r   __exit__r   tearDownr   s    r   r   TestVecAdd.tearDown   s&    &&tT48r   c                   ^	 SS K nSSKJm	  T	R                  U	4S j5       nUR                  R                  S5        SnT	R                  UR                  R	                  U5      5      nT	R                  UR                  R	                  U5      5      nT	R                  U5      nUR                  [        U5      5      " XEU5        [        UR                  5       5        Sn[        U5      U-  S-   nX(U4   " XEU5        [        UR                  5       5        UR                  R                  UR                  5       UR                  5       UR                  5       -   5        g )Nr   )cudac                 `   > TR                  S5      n[        U5      nX4:  a  X   X   -   X#'   g g )N   )gridlen)abctidsizer   s        r   f$TestVecAdd.test_ex_vecadd.<locals>.f    s4     ))A,Cq6Dz!& r   r   i    )numpynumbar   jitrandomseed	to_devicedevice_array_likeforallr   printcopy_to_hosttestingassert_equal)
r   npr$   Nr   r    r!   nthreadsnblocksr   s
            @r   test_ex_vecaddTestVecAdd.test_ex_vecadd   s    
	) 
	) 			q NN299++A./NN299++A./""1% 	
Qq!ann
 q6X%*	8
Q1%ann 	

NNNNq~~//	
r   )r   )
__name__
__module____qualname____firstlineno____doc__r   r   r7   __static_attributes____classcell__)r   s   @r   r   r      s    
,
 ,
r   r   __main__)	unittestnumba.cuda.testingr   r   numba.tests.supportr   r   r9   main r   r   <module>rF      sI     < / GH=
 =
 I=
@ zMMO r   