U ?h@sPdZddlZddlmZmZddlmZmZGdddeZ e dkrLe dS)z Test basic language features N)cudafloat64)unittest CUDATestCasec@s$eZdZddZddZddZdS)TestLangcsLdtdfdd}tt}|d||t|kdS)N)?g@g@void(float64[:])cstD]\}}|||<qdS)N) enumerate)aivtupS/var/www/html/venv/lib/python3.8/site-packages/numba/cuda/tests/cudapy/test_lang.pyfoosz$TestLang.test_enumerate..foor)rjitnpzeroslen assertTrueall)selfrr rr rtest_enumerate s  zTestLang.test_enumeratecsjddtdfdd}td}|d|t}t}|t|||kdS)N)r)g@gffffff@g@rcs0d}tD]\}}|||7}q||d<dS)Nr)zip)r cr jt1t2rrrszTestLang.test_zip..foorr)rrrrarrayrrsum)rrr brrr!rtest_zips    zTestLang.test_zipcCs,tddd}td}|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[:,:])cSs^tdD] }tjj}qtdD] }tjj}qtjdt}d|||f<|||f|||f<dS)Nrrrrr)rangerZ threadIdxxyZsharedr$r)Zaryr Ztxr tysmrrr"cuda_kernel_api_in_multiple_blocks1s     zCTestLang.test_issue_872..cuda_kernel_api_in_multiple_blocksr()rr(N)rrrr)rr.r rrrtest_issue_872)s zTestLang.test_issue_872N)__name__ __module__ __qualname__rr'r/rrrrr s r__main__) __doc__numpyrZnumbarrZnumba.cuda.testingrrrr0mainrrrrs 4