ó ¦–Õ\c!@sàddlZddlZddljZddlmZddlm Z ej dƒZ d„d„d„d„d „d „d „d „d „d„d„d„d„ej d„dej jddƒƒej d„dej jddƒƒej d„dej jddƒƒd„d„d„d„d„ej d„dej jddƒƒd„d „d!„d"„d#„ej d$„dej jdd%ƒƒgZej jd&eƒd'„ƒZej jd(d)d*gƒd+„ƒZdS(,iÿÿÿÿN(tsizeof(t assert_eqtcupycCs|S(N((tx((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyt tcCs tj|ƒS(N(tdatexpm1(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR RcCsd|S(Ni((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|dS(Ni((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|dS(Ni((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs||S(N((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs||S(N((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|dS(Ni((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|dd…dfS(Ni((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|dd…ddd…fS(Nii(tNone(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|jS(N(tT(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCstj|dƒS(Niii(iii(Rt transpose(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs |jƒS(N(tsum(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|jtj|jdƒƒS(Niÿÿÿÿ(tdottnptarangetshape(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRtmarkstreasonscupy.dot(numpy) failscCs|jtj|jdƒƒS(Niÿÿÿÿ(R R teyeR(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs,tj|tj|jd ƒdddgƒS(Nitaxesii(ii(ii(Rt tensordotR tonesR(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|jddƒS(Ntaxisi(R (R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyRRcCs|jddƒS(NRi(tmax(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR RcCs|jddƒS(NRii(ii(R (R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR!RcCs|jtjƒS(N(tastypeR t complex128(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR"RcCs|jd„ƒS(NcSs|dS(Ni((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR#R(t map_blocks(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR#RcCs |jdƒS(Ni(tround(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR$Rscupy doesn't support roundcCs,|j|jd|jd|jdfƒS(Niii(treshapeR(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR&RcCs t|ƒS(N(tabs(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR'RcCs |dkS(Ngà?((R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR(RcCs |jdƒS(Ni(iii(trechunk(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR)RcCs |jdƒS(Nii(iii(R(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR*RcCstjd||ƒS(Nsijk,ijk(Rteinsum(R((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyR+RsDdepends on resolution of https://github.com/numpy/numpy/issues/12974tfunccCs­tjjd ƒ}|jƒ}tj|dd dtƒ}tj|dd ƒ}||ƒ}||ƒ}t||ƒ|jr©|jddƒ}t |tj ƒs©t ‚ndS( Niiitchunksitasarrayt schedulerssingle-threaded(iii(iii(iii( RtrandomtgetRt from_arraytFalseRRtcomputet isinstancetndarraytAssertionError(R tctntdctdntddctddntresult((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyt test_basic1s     tdtypetf4tf8cCs7tjjdd|ƒ}t|ƒ|jks3t‚dS(NiiiR4(iii(RR$RtnbytesR+(R4R,((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pyt test_sizeofBs(tnumpyR tpytestt dask.arraytarrayRt dask.sizeofRtdask.array.utilsRt importorskipRtparamtmarktxfailt functionst parametrizeR3R8(((s9lib/python2.7/site-packages/dask/array/tests/test_cupy.pytsR