U ?h@sXddlZddlmZddlmZmZddZddZGdd d eZ e d krTe dS) N)cuda)unittest CUDATestCasecCs$t|jdD]}||||<qdSNr)rangeshapeinpoutir V/var/www/html/venv/lib/python3.8/site-packages/numba/cuda/tests/cudapy/test_slicing.pyfoosrcCs0td}t||ddf||ddfdS)N)rgridcufoorr r r copy s rc@seZdZddZddZdS)TestCudaSlicingcCsTtjdddtatdt}tjdtjddd}t |}|d||dS) Nzvoid(int32[:], int32[:])T)Zdevicezvoid(int32[:,:], int32[:,:])d)Zdtype )rr) rZjitrrrnpZarangeZint32ZreshapeZ zeros_like)selfZcucopyr r r r r test_slice_as_args  z!TestCudaSlicing.test_slice_as_argcCs0d}t|}tt|}t||dd<dSr)rrZ device_arraylenZ to_device)rNaZarrr r r test_assign_empty_slicesz'TestCudaSlicing.test_assign_empty_sliceN)__name__ __module__ __qualname__rrr r r r rs r__main__) numpyrZnumbarZnumba.cuda.testingrrrrrrmainr r r r s