1@j|dZddlZddlmZmZddlmZmZGddeZ e dkrej dSdS)z Test basic language features N)cudafloat64)unittest CUDATestCasec eZdZdZdZdZdS)TestLangcdtjdfd}tjt }|d||tj|kdS)N)?g@g@void(float64[:])c<tD] \}}|||< dS)N) enumerate)aivtups RC:\Projects\Historical\venv\Lib\site-packages\numba/cuda/tests/cudapy/test_lang.pyfooz$TestLang.test_enumerate..foos1!#  1!  r)rjitnpzeroslen assertTrueall)selfrrrs @rtest_enumeratezTestLang.test_enumerate s $ % %     & %  HSXX  D !  qCx(()))))rclddtjdfd}tjd}|d|tj}tj}|tj|||zkdS)N)r)g@gffffff@g@r cRd}tD] \}}|||zz }||d<dS)Nr)zip)rcrjt1t2s rrzTestLang.test_zip..foos?AB   1QU AaDDDrrr)rrrrarrayrrsum)rrrbr$r&r's @@rtest_zipzTestLang.test_zips   $ % %      & %  HQKKD ! HRLL HRLL qQUKKMM12233333rctjdd}tjd}|d|dS)a? Ensure that typing and lowering of CUDA kernel API primitives works in more than one block. Was originally to ensure that macro expansion works for more than one block (issue #872), but macro expansion has been replaced by a "proper" implementation of all kernel API functions. zvoid(float64[:,:])ctdD]}tjj}tdD]}tjj}tjdt}d|||f<|||f|||f<dS)Nr r!r r!r )ranger threadIdxxysharedr(r)aryrtxr%tysms r"cuda_kernel_api_in_multiple_blockszCTestLang.test_issue_872..cuda_kernel_api_in_multiple_blocks1s1XX & &^%1XX & &^%""6733BBr2vJRV*CBKKKrr.)rr.N)rrrr)rr8rs rtest_issue_872zTestLang.test_issue_872)sV & ' ' % % ( ' % HV  5*95a88888rN)__name__ __module__ __qualname__rr+r9rrrr sA * * *444"99999rr__main__) __doc__numpyrnumbarrnumba.cuda.testingrrrr:mainr=rrrDs 555555551919191919|191919h zHMOOOOOr