o Û¾i¶ã@sÀddlZddlZddlZddlmZddlmZddlm Z m Z mZddlm Z ejs.finalizer)r)r ÚRuntimeErrorrrrÚweakrefÚproxyÚcontextÚctypesÚc_void_prÚcudadrvÚdriverÚAutoFreePointer)rÚsizerÚctxÚptrrrrÚmemalloc s ÿzDeviceOnlyEMMPlugin.memalloccCó d|_dS©NT)r ©rrrrÚ initialize>s zDeviceOnlyEMMPlugin.initializecCr,r-)rr.rrrÚresetBs zDeviceOnlyEMMPlugin.resetcCsd|_tjdddS)NTé é@)ÚfreeÚtotal)rrÚ MemoryInfor.rrrÚget_memory_infoKsz#DeviceOnlyEMMPlugin.get_memory_infocCsd|_d|jjS)NTzDummy IPC handle for alloc %s)rÚdevice_pointerÚvalue)rÚmemoryrrrÚget_ipc_handlePsz"DeviceOnlyEMMPlugin.get_ipc_handlecCódS)Nrrr.rrrÚinterface_versionWsz%DeviceOnlyEMMPlugin.interface_version) Ú__name__Ú __module__Ú__qualname__Ú__doc__r r+r/r0r6r:Úpropertyr<Ú __classcell__rrrrrs rc@seZdZdZedd„ƒZdS)ÚBadVersionEMMPluginz?A plugin that claims to implement a different interface versioncCr;)Nérr.rrrr<_sz%BadVersionEMMPlugin.interface_versionN)r=r>r?r@rAr<rrrrrC\srCz+EMM Plugins not supported on CUDA simulatorcsXeZdZdZ‡fdd„Z‡fdd„Zdd„Zdd „Zd d„Zdd „Z e dd„ƒZ‡ZS)ÚTestDeviceOnlyEMMPluginzu Tests that the API of an EMM Plugin that implements device allocations only is used correctly by Numba. cs tƒ ¡t ¡t t¡dSr)r ÚsetUprÚcloseÚset_memory_managerrr.rrrrFkó zTestDeviceOnlyEMMPlugin.setUpcs tƒ ¡t ¡dtjj_dSr)r ÚtearDownrrGr%r&Ú_memory_managerr.rrrrJqrIz TestDeviceOnlyEMMPlugin.tearDowncCs´t ¡j}t d¡}t |¡}| |j¡| |j d¡| |j d|j¡t d¡}t |¡}| |j d¡| |j d|j¡~| d|j ¡| d|j ¡~| d|j ¡dS)Né rérD)rÚcurrent_contextÚmemory_managerÚnpÚarangeÚdevice_array_likeÚ assertTruerÚassertEqualrrÚnbytesÚassertNotInÚassertIn)rÚmgrÚarr_1Úd_arr_1Úarr_2Úd_arr_2rrrÚ test_memallocws z%TestDeviceOnlyEMMPlugin.test_memalloccCs| t ¡jj¡dSr)rSrrNrOr r.rrrÚtest_initialized_in_context”sz3TestDeviceOnlyEMMPlugin.test_initialized_in_contextcCs"t ¡}| ¡| |jj¡dSr)rrNr0rSrOr)rr)rrrÚ test_reset™sz"TestDeviceOnlyEMMPlugin.test_resetcCs>t ¡}| ¡}| |jj¡| |jd¡| |jd¡dS)Nr1r2) rrNr6rSrOrrTr3r4)rr)ÚmeminforrrÚtest_get_memory_infožs z,TestDeviceOnlyEMMPlugin.test_get_memory_infocCsDt d¡}t |¡}| ¡}t ¡}| |jj¡| d|j ¡dS)NrDzDummy IPC handle for alloc 1)rPrQrrRr:rNrSrOrrWÚ_ipc_handle)rÚarrÚd_arrÚipchr)rrrÚtest_get_ipc_handle¥s z+TestDeviceOnlyEMMPlugin.test_get_ipc_handle) r=r>r?r@rFrJr]r^r_rarrfrBrrrrrEdsrEc@seZdZdZdd„ZdS)ÚTestBadEMMPluginVersionzV Ensure that Numba rejects EMM Plugins with incompatible version numbers. cCsJ| t¡ }t t¡Wdƒn1swY| dt|jƒ¡dS)Nzversion 1 required)ÚassertRaisesrrrHrCrWÚstrÚ exception)rÚraisesrrrÚtest_bad_plugin_version¹sÿz/TestBadEMMPluginVersion.test_bad_plugin_versionN)r=r>r?r@rlrrrrrg²srgÚ__main__)r#ÚnumpyrPr ÚnumbarÚ numba.corerÚnumba.cuda.testingrrrÚnumba.tests.supportrÚENABLE_CUDASIMÚHostOnlyCUDAMemoryManagerrrCrErgr=ÚmainrrrrÚs"QMÿ