o ۾i@s^ddlZddlmZddlZddlmZejddZGdddeZ e dkr-e dSdS) N)cuda) CUDATestCasecCs,td}||jkr||d7<dSdSNrgridsize)xir W/home/ubuntu/.local/lib/python3.10/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[:])cSs4td}||jkr|||||||<dSdSrr)ar yr r r r bars  z%TestForAll.test_forall_2..bar )dtypegX9v?)decimal) rjitrrfloat32rrrrr)rrr roldyrr 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_workcCsJ|t }tdWdn1swY|dt|jdS)Nz,Can't create ForAll with negative task count) assertRaises ValueErrorr rassertInstr exception)rraisesr r r test_forall_negative_work*s  z$TestForAll.test_forall_negative_workN)__name__ __module__ __qualname__rr$r%r-r r r r rs  r__main__) numpyrnumbarunittestnumba.cuda.testingrr!r rr.mainr r r r s   $