ó \ K]c@@sÛddlmZmZddlZddljjZddl m Z m Z m Z ddl m Z ddl mZddl mZddlmZmZedƒd eejfd „ƒYƒZed kr×ejƒndS( i(tprint_functiontabsolute_importN(tvoidtfloat32tfloat64(t guvectorize(tcuda(tunittest_support(tskip_on_cudasimt SerialMixins&ufunc API unsupported in the simulatortTestCUDAGufunccB@s}eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „ZRS( cC@s tttdd…dd…ftdd…dd…ftdd…dd…fƒgdddƒd„ƒ}|}d|_d}tj|dddtjƒj|ddƒ}tj|dd dtjƒj|dd ƒ}|||ƒ}tj||ƒ}|j tj ||ƒƒdS( Ns(m,n),(n,p)->(m,p)ttargetRc S@s¥|j\}}|j\}}x€t|ƒD]r}xit|ƒD][}d|||fWq+WdS(Ni(tshapetrange( tAtBtCtmtntptitjtk((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyt matmulcoresiiitdtypei( RRRt max_blocksizetnptarangetreshapetuttmatrix_multiplyt assertTruetallclose(tselfRtgufunct matrix_ctRRRtGold((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_gufunc_smallsW  & & c C@s5tttdd…dd…ftdd…dd…ftdd…dd…fƒgdddƒd„ƒ}|}d|_d}tj|dddtjƒj|ddƒ}tj|dd dtjƒj|dd ƒ}tj|ƒ}|||ƒj ƒ}t j ||ƒ}|j tj ||ƒƒdS( Ns(m,n),(n,p)->(m,p)R Rc S@s¥|j\}}|j\}}x€t|ƒD]r}xit|ƒD][}d|||fWq+WdS(Ni(R R ( RRRRRRRRR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyR.siiiRi(RRRRRRRRt to_devicet copy_to_hostRRRR ( R!RR"R#RRtdBRR$((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_gufunc_auto_transfer,sW  & & cC@s tttdd…dd…ftdd…dd…ftdd…dd…fƒgdddƒd„ƒ}|}d|_d}tj|ddd tjƒj|ddƒ}tj|dd d tjƒj|dd ƒ}|||ƒ}tj||ƒ}|j tj ||ƒƒdS( Ns(m,n),(n,p)->(m,p)R Rc S@s¥|j\}}|j\}}x€t|ƒD]r}xit|ƒD][}d|||fWq+WdS(Ni(R R ( RRRRRRRRR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyRKsiiéiiRi( RRRRRRRRRRR (R!RR"R#RRRR$((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyt test_gufuncIsW  & & cC@s&tttdd…dd…ftdd…dd…ftdd…dd…fƒgdddƒd„ƒ}|}d|_d}tj|ddd tjƒjdd ddƒ}tj|dd d tjƒjdd dd ƒ}|||ƒ}tj||ƒ}|j tj ||ƒƒdS( Ns(m,n),(n,p)->(m,p)R Rc S@s¥|j\}}|j\}}x€t|ƒD]r}xit|ƒD][}d|||fWq+WdS(Ni(R R ( RRRRRRRRR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyRfsiidiiRii( RRRRRRRRRRR (R!RR"R#RRRR$((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_gufunc_hidimdsW  22cC@stttdd…dd…ftdd…dd…ftdd…dd…fƒgdddƒd„ƒ}|}tjjdddƒ}tjjddƒ}tj||ƒ}|||ƒ}tjj ||ƒ||tj |dƒƒ}tjj ||ƒdS( Ns(m,n),(n,p)->(m,p)R Rc S@s¥|j\}}|j\}}x€t|ƒD]r}xit|ƒD][}d|||fWq+WdS(Ni(R R ( RRRRRRRRR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyRsi ii(i ii( RRRRtrandomtrandnRRttestingtassert_allclosettile(R!RR"tXtYtgoldtres1tres2((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_gufunc_new_axis}sW cC@s)tttdd…dd…ftdd…dd…ftdd…dd…fƒgdddƒd„ƒ}|}d|_d}tj|ddd tjƒj|ddƒ}tj|dd d tjƒj|dd ƒ}d |_|||ƒ}tj||ƒ}|j tj ||ƒƒdS( Ns(m,n),(n,p)->(m,p)R Rc S@s¥|j\}}|j\}}x€t|ƒD]r}xit|ƒD][}d|||fWq+WdS(Ni(R R ( RRRRRRRRR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyRšsiiéiiRii ( RRRRRRRRRRR (R!RR"R#RRRR$((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_gufunc_adjust_blocksize˜sW  & &  c C@s™tttdd…dd…ftdd…dd…ftdd…dd…fƒgdddƒd„ƒ}|}d|_d}tj|ddd tjƒj|ddƒ}tj|dd d tjƒj|dd ƒ}tjƒ}tj ||ƒ}tj ||ƒ}tj d dd |j d |ƒ} |||d | d |ƒ} | j d |ƒ} |j ƒtj||ƒ} |jtj| | ƒƒdS(Ns(m,n),(n,p)->(m,p)R Rc S@s¥|j\}}|j\}}x€t|ƒD]r}xit|ƒD][}d|||fWq+WdS(Ni(R R ( RRRRRRRRR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyR¶siiéiiRiR tstreamtout(iéii(RRRRRRRRR8R&t device_arrayRR't synchronizeRRRR ( R!RR"R#RRR8tdAR(tdCRR$((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_gufunc_stream´s&W  & &  ! cC@s„ttttƒgdddƒd„ƒ}tjddtjƒd}tj|ƒ}||d|ƒ|jtj||ƒƒdS( Ns(x)->(x)R RcS@s,x%t|jƒD]}||||(x)R RcS@s,x%t|jƒD]}||||(x, y)R RcS@sZxSt|jdƒD]>}x5t|jdƒD] }|||f|||f(c C@s˜d„}ttttƒgddddtƒ|ƒ|jtƒ4}ttttƒgddddtƒ|ƒWdQX|jdt|j ƒƒdS(NcS@sdS(N((RR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pytfooss(x)->(x)R Rtnopythonsnopython flag must be True( RRRtTruet assertRaisest TypeErrortFalset assertEqualtstrt exception(R!RHtraises((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_nopython_flags   c C@sêd„}|jtƒ:}ttttƒgddddtdtƒ|ƒWdQXd}t|jƒ}|j |t |ƒ |ƒ|t |ƒj ƒj dƒ}g|D]}|j d ƒ^q©}|j t ddgƒt |ƒƒdS( NcS@sdS(N((RR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyRHss(x)->(x)R Rtwhat1tever2s/The following target options are not supported:t,s'" (RKRLRRRRJRMRORPRNtlentstriptsplittset(R!RHRQtheadtmsgtitemsR((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_invalid_flagss "c C@sttttƒgdddƒd„ƒ}tjddtjƒ}}|jtƒ}|||d|ƒWdQX|jt|j ƒdƒdS( Ns(x)->(x)R RcS@sdS(N((tinpR9((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyRH#si RR9s>cannot specify 'out' as both a positional and keyword argument( RRRRtzerosRKt ValueErrorRNRORP(R!RHR^R9RQ((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyttest_duplicated_output"s ,(t__name__t __module__R%R)R*R+R6R7R>RBRCRGRRR]Ra(((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyR s       %     t__main__(t __future__RRtnumpyRtnumpy.core.umath_teststcoret umath_testsRtnumbaRRRRRRtunittesttnumba.cuda.testingRR tTestCaseR Rbtmain(((sBlib/python2.7/site-packages/numba/cuda/tests/cudapy/test_gufunc.pyts  ÿ!