ó \ K]c@ sddlmZddlmZddlZddlmZmZddl m Z m Z m Z ddl mZe dƒde e jfd „ƒYƒZe d ƒd e e jfd „ƒYƒZd e e jfd„ƒYZe dƒde e jfd„ƒYƒZedkre jƒndS(iÿÿÿÿ(tdivision(tcontextmanagerN(tcudatconfig(tunittesttskip_on_cudasimt SerialMixin(tcaptured_stderrsnot supported on CUDASIMtTestDeallocationcB seZd„Zd„ZRS(cC s©tjƒj}|jƒ|jt|ƒdƒxGttjƒD]6}tj t j dƒƒ|jt|ƒ|dƒq?Wtj t j dƒƒ|jt|ƒdƒdS(Nii( Rtcurrent_contextt deallocationstcleart assertEqualtlentrangeRtCUDA_DEALLOCS_COUNTt to_devicetnptarange(tselftdeallocsti((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_max_pending_counts cC s;tjƒ}|j}|jƒ|jt|ƒdƒ|jƒ}d}tj}zÚ||j t_|j |j |ddƒtj t j|ddt jƒƒ|jt|ƒdƒtj t j|j |jdt jƒƒ|jt|ƒdƒtj t jddt jƒƒ|jt|ƒdƒWd|t_XdS( Nii itdeltaiitdtypei@B(RR R R R R tget_memory_infoRtCUDA_DEALLOCS_RATIOttotaltassertAlmostEqualt_max_pending_bytesRRtonestint8t_size(RtctxRtmit max_pendingt old_ratio((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_max_pending_bytess"     #)(t__name__t __module__RR%(((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyR s s&defer_cleanup has no effect in CUDASIMtTestDeferCleanupcB s#eZd„Zd„Zd„ZRS(cC såtjdƒ}tj|ƒ}tjƒj}|jƒ|jt|ƒdƒtj ƒftj|ƒ}~|jt|ƒdƒ~|jt|ƒdƒ|jƒ|jt|ƒdƒWdQX|jƒ|jt|ƒdƒdS(Niiii( RRRRR R R R R t defer_cleanup(Rtharrtdarr1Rtdarr2((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyt test_basicAs    c C stjdƒ}tj|ƒ}tjƒj}|jƒ|jt|ƒdƒtj ƒ™tj ƒftj|ƒ}~|jt|ƒdƒ~|jt|ƒdƒ|jƒ|jt|ƒdƒWdQX|jƒ|jt|ƒdƒWdQX|jƒ|jt|ƒdƒdS(Niiii( RRRRR R R R R R)(RR*R+RR,((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyt test_nestedSs$      c C s7tjdƒ}tj|ƒ}tjƒj}|jƒ|jt|ƒdƒdt fd„ƒY}|j |ƒftj ƒStj|ƒ}~|jt|ƒdƒ|jƒ|jt|ƒdƒ|‚WdQXWdQX|jƒ|jt|ƒdƒ~|jt|ƒdƒ|jƒ|jt|ƒdƒdS(Niit CustomErrorcB seZRS((R&R'(((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyR/osi( RRRRR R R R R t Exceptiont assertRaisesR)(RR*R+RR/R,((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_exceptionhs(     (R&R'R-R.R2(((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyR(?s  tTestDeferCleanupAvailcB seZd„ZRS(cC stjƒWdQXdS(N(RR)(R((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_context_managerƒs (R&R'R4(((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyR3‚stTestDelcB s\eZdZed„ƒZd„Zd„Zd„Zd„Zd„Z d„Z d„Z RS( sJ Ensure resources are deleted properly without ignored exception. cc s;tƒ}dV|jjƒWdQX|j|jƒƒdS(N(RR R t assertFalsetgetvalue(RR!tcap((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pytcheck_ignored_exceptionŽs cC s5tjƒ}|jƒ}|j|ƒ~WdQXdS(N(RR t create_streamR9(RR!tstream((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyt test_stream•s  cC s5tjƒ}|jƒ}|j|ƒ~WdQXdS(N(RR t create_eventR9(RR!tevent((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyt test_event›s  cC s8tjƒ}|jdƒ}|j|ƒ~WdQXdS(Ni (RR t memhostallocR9(RR!tmem((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_pinned_memory¡s cC s>tjƒ}|jddtƒ}|j|ƒ~WdQXdS(Ni tmapped(RR R@tTrueR9(RR!RA((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_mapped_memory§s cC s8tjƒ}|jdƒ}|j|ƒ~WdQXdS(Ni (RR tmemallocR9(RR!RA((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_device_memory­s cC s dtfd„ƒY}tjdƒ}tjƒ}|jjƒ|j|ƒºtj|ƒWdQXtj|ƒWdQXtj ƒ1tj|ƒWdQXtj|ƒWdQXWdQXy tj|ƒ |‚WdQXWn'|k rtj|ƒWdQXnXWdQXdS(NtPinnedExceptioncB seZRS((R&R'(((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyRH¶si( R0RtzerosRR R R R9tpinnedR)(RRHtarrR!((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_pinned_contextmanager³s(     cC sdtfd„ƒY}tjdƒ}tjƒ}|jjƒ|j|ƒÆtj|ƒ}WdQXtj|ƒ}WdQXtj ƒ5tj|ƒ}WdQXtj|ƒ}WdQXWdQXy"tj|ƒ }|‚WdQXWn)|k rtj|ƒ}WdQXnXWdQXdS(NtMappedExceptioncB seZRS((R&R'(((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyRMÒsi( R0RRIRR R R R9RCR)(RRMRKR!tmarr((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyttest_mapped_contextmanagerÏs(     ( R&R't__doc__RR9R<R?RBRERGRLRO(((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyR5‰s      t__main__(t __future__Rt contextlibRtnumpyRtnumbaRRtnumba.cuda.testingRRRtnumba.tests.supportRtTestCaseRR(R3R5R&tmain(((sJlib/python2.7/site-packages/numba/cuda/tests/cudadrv/test_deallocations.pyts  2 B a