U ?h@sTddlZddlmZddlmZddlZddZGdddeZe dkrPe dS) N)cuda) CUDATestCasecCs$|||tjd}||d<dS)Nr)viewnpint32)byte_arrstartstopoutputvalr \/var/www/html/venv/lib/python3.8/site-packages/numba/cuda/tests/cudapy/test_array_methods.pyreinterpret_array_typesrc@seZdZddZdS)TestCudaArrayMethodsc Cst}t|}tjdtjd}ttjj}t dd|D]Z}||}||| tjd}tj dtjd}|d|||||d} | || q8dS)z= Reinterpret byte array as int32 in the GPU. )dtyper)rrN) rrZjitrZarangeZuint8rritemsizerangerZzerosZ assertEqual) selfZpyfuncZkernelrrrr expectr gotr r r test_reinterpret_array_types z0TestCudaArrayMethods.test_reinterpret_array_typeN)__name__ __module__ __qualname__rr r r r r sr__main__) numpyrZnumbarZnumba.cuda.testingrZunittestrrrmainr r r r s