U ?h@sZddlZddlmZddlZddlmZejddZGdddeZ e dkrVe dS) N)cuda) CUDATestCasecCs(td}||jkr$||d7<dSNrgridsize)xir U/var/www/html/venv/lib/python3.8/site-packages/numba/cuda/tests/cudapy/test_forall.pyfoos  r c@s,eZdZddZddZddZddZd S) TestForAllcCs8td}|}t|j|tj||ddS)N r)nparangecopyr forallrtestingassert_array_almost_equal)selfarrorigr r r test_forall_1s zTestForAll.test_forall_1cCsptddd}tjdtjd}tjdtjd}|}d}||j|||tjj ||||dddS) Nz%void(float32, float32[:], float32[:])cSs0td}||jkr,|||||||<dSrr)ar yr r r r bars  z%TestForAll.test_forall_2..bar )ZdtypegX9v?)decimal) rjitrrZfloat32rrrrr)rrr rZoldyrr r r test_forall_2s zTestForAll.test_forall_2cCstd}td|dS)Nrr)rrr r)rrr r r test_forall_no_work$s zTestForAll.test_forall_no_workc Cs6|t}tdW5QRX|dt|jdS)Nz,Can't create ForAll with negative task count)Z assertRaises ValueErrorr rZassertInstr exception)rZraisesr r r test_forall_negative_work*s  z$TestForAll.test_forall_negative_workN)__name__ __module__ __qualname__rr!r"r'r r r r rsr__main__) numpyrZnumbarZunittestZnumba.cuda.testingrr r rr(mainr r r r s   $