o 8ήc@s\ddlZddlmZddlmZmZddZddZGdd d eZ e d kr,e dSdS) N)cuda)unittest CUDATestCasecCs$t|jdD]}||||<qdSNr)rangeshapeinpoutir K/tmp/pip-target-vg8gfxp4/lib/python/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)devicezvoid(int32[:,:], int32[:,:])d)dtype )rr) rjitrrrnparangeint32reshape zeros_like)selfcucopyr r r r r test_slice_as_args  z!TestCudaSlicing.test_slice_as_argcCs0d}t|}tt|}t||dd<dSr)rr device_arraylen to_device)rNaarrr r r test_assign_empty_slicesz'TestCudaSlicing.test_assign_empty_sliceN)__name__ __module__ __qualname__r r'r r r r rs r__main__) numpyrnumbarnumba.cuda.testingrrrrrr(mainr r r r s