
    `i                         d dl Z d dlmZmZ d dlmZ  ed           G d de                      Zedk    r e j                     dS dS )    N)CUDATestCaseskip_on_cudasim)captured_stdoutz4cudasim doesn't support cuda import at non-top-levelc                   2     e Zd ZdZ fdZ fdZd Z xZS )
TestVecAddz%
    Test simple vector addition
    c                     t                      | _        | j                                         t                                                       d S N)r   _captured_stdout	__enter__supersetUpself	__class__s    }/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/numba/cuda/tests/doc_examples/test_vecadd.pyr   zTestVecAdd.setUp   s=     !0 1 1'')))    c                     | j                             d d d            t                                                       d S r	   )r
   __exit__r   tearDownr   s    r   r   zTestVecAdd.tearDown   s:    &&tT4888r   c                 4  	 dd l }ddlm	 	j        	fd            }|j                            d           d}	                    |j                            |                    }	                    |j                            |                    }	                    |          } |                    t          |                    |||           t          |                                           d}t          |          |z  dz   } |||f         |||           t          |                                           |j                            |                                |                                |                                z              d S )Nr   )cudac                                          d          }t          |          }||k     r| |         ||         z   ||<   d S d S )N   )gridlen)abctidsizer   s        r   fz$TestVecAdd.test_ex_vecadd.<locals>.f    sH     ))A,,Cq66DTzz3!C&# zr   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_vecaddzTestVecAdd.test_ex_vecadd   s    
	) 	) 	) 	) 
	) 		q NN29++A..//NN29++A..//""1%% 	QAq!!!ann
 q66X%*'8
Q1%%%ann 	
NNNNq~~///	
 	
 	
 	
 	
r   )__name__
__module____qualname____doc__r   r   r3   __classcell__)r   s   @r   r   r      sj                 
,
 ,
 ,
 ,
 ,
 ,
 ,
r   r   __main__)	unittestnumba.cuda.testingr   r   numba.tests.supportr   r   r4   main r   r   <module>r?      s     < < < < < < < < / / / / / / GHH=
 =
 =
 =
 =
 =
 =
 IH=
@ zHMOOOOO r   