o i<@sddlZddlmZddlmZmZmZmZddlm Z ddlm Z ddl m Z m Z ddlZddlZddlmZmZddlmZefd d Ze d Gd d d e Ze d Gddde ZedkrhedSdS)N) namedtuple)voidint32float32float64) guvectorize)cuda)skip_on_cudasim CUDATestCase)NumbaPerformanceWarning TypingError)override_configcCsTtt|ddddf|ddddf|ddddfgddddd}|S)Nz(m,n),(n,p)->(m,p)rtargetc Ssv|j\}}|j\}}t|D]*}t|D]#}d|||f<t|D]}|||f|||f|||f7<q qqdSNrshaperange) ABCmnpijkra/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/cuda/tests/cudapy/test_gufunc.py matmulcores      *z*_get_matmulcore_gufunc..matmulcore)rr)dtyperrrr_get_matmulcore_gufuncs >  r!z&ufunc API unsupported in the simulatorc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/S)0TestCUDAGufunccCszt}d}tj|ddtjd|dd}tj|ddtjd|dd}|||}t||}|t||dSNr r!nparangerreshapematmul assertTrueallcloseselfgufunc matrix_ctrrrGoldrrrtest_gufunc_small!  z TestCUDAGufunc.test_gufunc_smallcCst}d}tj|ddtjd|dd}tj|ddtjd|dd}t|}|||}t||}| t ||dSr#) r!r)r*rr+r to_device copy_to_hostr,r-r.)r0r1r2rrdBrr3rrrtest_gufunc_auto_transfer/s  z(TestCUDAGufunc.test_gufunc_auto_transfercCszt}d}tj|ddtjd|dd}tj|ddtjd|dd}|||}t||}|t||dS)Nr$r%r&r'r(r/rrr test_gufunc?r5zTestCUDAGufunc.test_gufunccCs~t}d}tj|ddtjddddd}tj|ddtjddddd}|||}t||}|t||dS)Ndr$r%r&r'r(r/rrrtest_gufunc_hidimMs$$  z TestCUDAGufunc.test_gufunc_hidimcCspttd}tjddd}tjdd}t||}|||}tj||||t|d}tj||dS)Nr& )r?rA) r!rr)randomrandnr,testingassert_allclosetile)r0r1XYgoldres1res2rrrtest_gufunc_new_axisYs   z#TestCUDAGufunc.test_gufunc_new_axisc Cst}d}tj|ddtjd|dd}tj|ddtjd|dd}t}t||}t||}tjd|j |d}|||||d}|j |d } | t ||} | t| | dS) Nr:r$r%r&r')r:r$r')rr stream)outrM)rM)r!r)r*rr+rrMr6 device_arrayr r7 synchronizer,r-r.) r0r1r2rrrMdAr8dCrr3rrrtest_gufunc_streamhs"    z!TestCUDAGufunc.test_gufunc_streamcCsjtttddtddgddddd}tjdtjdd}t|}|||d tj||dS) N(x)->(x)rrcS t|jD]}||||<qdSNrsizerrrrrrcopyz&TestCUDAGufunc.test_copy..copyr?r&rArNrrrr)r* zeros_likerDrEr0rZrrrrr test_copys   zTestCUDAGufunc.test_copycCslttddtddfgddddd}tjdtjdd}t|}|||d |t||dS) NrTrrcSrUrVrWrYrrrrZr[z9TestCUDAGufunc.test_copy_unspecified_return..copyr?r&rAr\)rrr)r*r^r-r.r_rrrtest_copy_unspecified_returns   z+TestCUDAGufunc.test_copy_unspecified_returncCsntttddtddgddddd}tjdtjdd}t|}|||d |t||dS) NrTrrcSrUrVrWrYrrrrZr[z*TestCUDAGufunc.test_copy_odd..copy r&rAr\)rrrr)r*r^r-r.r_rrr test_copy_odds   zTestCUDAGufunc.test_copy_oddcCstttddddftddddfgddddd}tjdtjddd d }t|}|||d |t||dS) Nz(x, y)->(x, y)rrcSs@t|jdD]}t|jdD] }|||f|||f<qqdS)NrrArr)rrxyrrrcopy2ds z*TestCUDAGufunc.test_copy2d..copy2dr&r'rAr\) rrrr)r*r+r^r-r.)r0rgrrrrr test_copy2ds,   zTestCUDAGufunc.test_copy2dcstttddtddgdddddtjfdd}tdd }t|}d }| t ||d ||WddS1sHwYdS) Nz(n)->(n)rrcSs$t|jdD]}||||<qdSrrd)rbrrrr gufunc_copyszDTestCUDAGufunc.test_not_supported_call_from_jit..gufunc_copycs ||SrVr)rrkrlrrcuda_jits zATestCUDAGufunc.test_not_supported_call_from_jit..cuda_jitrz#Untyped global name 'gufunc_copy'.*)rArA) rrrrjitr)r*astyper^assertRaisesRegexr )r0rnrrkmsgrrmr test_not_supported_call_from_jits  "z/TestCUDAGufunc.test_not_supported_call_from_jitc Cstdgddddd}tjdd}tjdd}t|jd d}td d Ltj d d -}||||| |d j t | dt|d j| dt|d jWdn1sewYWddSWddS1s}wYdS)N(void(float32[:], float32[:], float32[:]) (n),(n)->(n)rrcS0|jd}t|D] }||||||<q dSrrarkdistlenrrrrnumba_dist_cuda  zMTestCUDAGufunc.test_inefficient_launch_configuration..numba_dist_cudarorrCUDA_LOW_OCCUPANCY_WARNINGSrATrecordz Grid sizez low occupancy)rr)rBrandrqzerosrr warningscatch_warnings assertEqualcategoryr assertInstrmessage)r0r|ryrkrzwrrr%test_inefficient_launch_configurations"   "z4TestCUDAGufunc.test_inefficient_launch_configurationc Cstdgdddddd}tjdd d }tjdd d }t|}td d 5tj dd }||||| t |dWdn1sOwYWddSWddS1sgwYdS)NrurvTr)nopythonrcSrwrrrxrrrnumba_dist_cuda2r}zLTestCUDAGufunc.test_efficient_launch_configuration..numba_dist_cuda2ir)ir$r~rArr) rr)rBrrqr+r^r rrrr{)r0rryrkrzrrrr#test_efficient_launch_configurations&    "z2TestCUDAGufunc.test_efficient_launch_configurationcCsdd}tttddtddgdddd||t}tttddtddgdddd|Wdn1s@wY|dt|jdS) NcSdSrVrrrrrrfooz.TestCUDAGufunc.test_nopython_flag..foorTrT)rrFznopython flag must be True)rrr assertRaises TypeErrorrr exception)r0rraisesrrrtest_nopython_flags  z!TestCUDAGufunc.test_nopython_flagcCsdd}|t}tttddtddgddddd|Wdn1s+wYd}t|j}||dt|||t|d d }d d |D}|t d d gt |dS)NcSrrVrrrrrrrz.TestCUDAGufunc.test_invalid_flags..foorTrTF)rwhat1ever2z/The following target options are not supported:,cSsg|]}|dqS)z'" )strip).0rrrr sz5TestCUDAGufunc.test_invalid_flags..rr) rrrrrrrrr{rsplitset)r0rrheadrsitemsrrrtest_invalid_flagss  z!TestCUDAGufunc.test_invalid_flagscCstttddtddgddddd}tjdtjd}}|t}||||dWdn1s8wYd }|t|j |dS) NrTrrcSrrVr)inprNrrrrz2TestCUDAGufunc.test_duplicated_output..foor?r&r\z()rrcSs6d}tt|D] }|||||7}q||d<dSr)rr{)rerfrsrrrr gu_reduces z1TestCUDAGufunc.check_tuple_arg..gu_reducerA)axis)rrr)sumasarrayrD assert_equal)r0ryrkrrexpectedrrrcheck_tuple_args&  zTestCUDAGufunc.check_tuple_argcCsd}d}|||dS)N)?@@@@@)?@ @@@@)rr0ryrkrrrtest_tuple_of_tuple_arg+sz&TestCUDAGufunc.test_tuple_of_tuple_argcCsRtdd}|dddd|ddddf}|d d d d|d d ddf}|||dS)NPointrerfzrrrrrrrrrrrr)rr)r0rryrkrrrtest_tuple_of_namedtuple_arg2s     z+TestCUDAGufunc.test_tuple_of_namedtuple_argcCs8tdtdf}tdtdf}|||dS)Nrrrr)r)rrrrrrtest_tuple_of_array_arg:sz&TestCUDAGufunc.test_tuple_of_array_argcCst}||jddS)Nr)r!r__name__)r0r1rrrtest_gufunc_nameAszTestCUDAGufunc.test_gufunc_namecCs~|t}tttddtddgddddd}Wdn1s'wYt|j}|d||d|dS)Nz(m)->(m)rrcSrrVr)rerfrrrfGrz.TestCUDAGufunc.test_bad_return_type..fz+guvectorized functions cannot return valueszspecifies int32 return type)rrrrrrr)r0terrsrrrtest_bad_return_typeEs $   z#TestCUDAGufunc.test_bad_return_typecCs ttddtddtddfgddddd}td}|t }||Wdn1s4wYt|j}|d||d||d ||t}|||||Wdn1sgwYt|j}|d||d||d |dS) Nz (m),(m)->(m)rrcSrrVrrrrrrPz;TestCUDAGufunc.test_incorrect_number_of_pos_args..fr'%gufunc accepts 2 positional argumentszor 3 positional argumentsGot 1 positional argument.zGot 4 positional arguments. rrr)r*rrrrrr0rarrrrsrrr!test_incorrect_number_of_pos_argsOs&$            z0TestCUDAGufunc.test_incorrect_number_of_pos_argsN)r __module__ __qualname__r4r9r;r>rLrSr`rarcrjrtrrrrrrrrrrrrrrrrr"s0    r"c@s4eZdZddZddZddZddZd d Zd S) TestMultipleOutputscCstttddtddtddgddddd}tjdtjdd}t|}t|}||||tj||tj||dS) N (x)->(x),(x)rrcSs,t|jD]}||||<||||<qdSrVrWrrrrrrrrZms zKTestMultipleOutputs.test_multiple_outputs_same_type_passed_in..copyr?r&rAr])r0rZrrrrrr)test_multiple_outputs_same_type_passed_inls&    z=TestMultipleOutputs.test_multiple_outputs_same_type_passed_incCstttddtddtddgddddd}tjdtjdd}t|}t|}||||tj||tj|d |dS) NrrrcS0t|jD]}||||<||d||<qdSNr$rWrrrrcopy_and_double~ zRTestMultipleOutputs.test_multiple_outputs_distinct_values..copy_and_doubler?r&rAr$r]r0rrrrrrr%test_multiple_outputs_distinct_values|s&    z9TestMultipleOutputs.test_multiple_outputs_distinct_valuescCs|tttddtddtddgddddd}tjdtjdd}||\}}tj||tj|d |dS) NrrrcSrrrWrrrrrrzLTestMultipleOutputs.test_multiple_output_allocation..copy_and_doubler?r&rAr$)rrrr)r*rDrErrrrtest_multiple_output_allocations&  z3TestMultipleOutputs.test_multiple_output_allocationcCstttddtddtddgddddd}tjdtjdd}t|}tj|tjd}||||tj||tj|td |dS) NrrrcSr)NrrWrrrrcopy_and_multiplyrzJTestMultipleOutputs.test_multiple_output_dtypes..copy_and_multiplyr?r&rAr) rrrrr)r*r^rDrE)r0rrrrrrrtest_multiple_output_dtypess&   z/TestMultipleOutputs.test_multiple_output_dtypescCsttddtddtddtddfgddddd}td}|t }||Wdn1s9wYt|j}|d||d||d ||t}||||||Wdn1smwYt|j}|d||d||d |dS) Nz(m),(m)->(m),(m)rrcSrrVr)rerfrrrrrrrz@TestMultipleOutputs.test_incorrect_number_of_pos_args..fr'rzor 4 positional argumentsrzGot 5 positional arguments.rrrrrrs&.            z5TestMultipleOutputs.test_incorrect_number_of_pos_argsN)rrrrrrrrrrrrrjs  r__main__)numpyr) collectionsrnumbarrrrrrnumba.cuda.testingr r unittestrnumba.core.errorsr r numba.tests.supportr r!r"rrmainrrrrs(     M\