σ \ K]c@`sπdZddlmZmZmZddlZddlmZddlm Zm Z de fd„ƒYZ e ƒZ[ d e fd „ƒYZdad „Zd „Zd „ZejƒZdd„Zdd„Zd„Zd„Zd„ZdS(s! Expose each GPU device directly i(tprint_functiontabsolute_importtdivisionN(t servicelibi(thsatContextt_culistcB`szeZdZd„Zed„ƒZd„Zd„Zd„Zd„Z d„Z d„Z e Z d „Z ed „ƒZRS( s)A thread local list of GPU instances cC`s d|_dS(N(tNonet_lst(tself((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt__init__scC`s"|js|jƒ|_n|jS(N(Rt _init_gpus(R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt_gpuss cC`s1g}x$tjD]}|jt|ƒƒqW|S(N(tdrivert componentstappendtCU(R tgpustcom((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR scC`s |j|S(N(R (R titem((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt __getitem__scC`s|jj|ƒS(N(R R(R R((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR scC`s t|jƒS(N(tlenR (R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt__len__#scC`s t|jƒS(N(tboolR (R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt __nonzero__&scC`s t|jƒS(N(titerR (R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt__iter__)scC`sx|D]}|jƒqWdS(N(treset(R tgpu((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR.s cC`stjS(s>Get the current GPU object associated with the thread (t_custackttop(R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pytcurrent2s(t__name__t __module__t__doc__R tpropertyR R RRRRRt__bool__RR(((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR s        RcB`sGeZd„Zd„Zd„Zd„Zd„Zd„Zd„ZRS(cC`s||_d|_dS(N(t_cuRt_context(R tcu((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR >s cC`s.|jdƒrt|ƒ‚nt|j|ƒS(sRedirect to self._gpu t_(t startswithtAttributeErrortgetattrR%(R tkey((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt __getattr__BscC`s t|jƒS(N(treprR%(R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt__repr__IscC`s+|jdkr$|jjƒ|_n|jS(s@Associate the context of this GPU to the running thread N(R&RR%tcreate_context(R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pytassociate_contextLscC`s|jƒtj|ƒdS(N(R1Rtpush(R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt __enter__Us cC`s0tƒ|kst‚|jjƒtjƒdS(N(t _get_devicetAssertionErrorR&tpopR(R texc_typetexc_valtexc_tb((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt__exit__Ys cC`s&|jr"|jjƒd|_ndS(N(R&RR(R ((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR^s  ( R R!R R-R/R1R3R:R(((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR=s     cC`sHtdkrDgtjD]}|js|^qd}t|ƒantS(Ni(t _cpu_contextRR tagentst is_componentt_Context(tat cpu_agent((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pytget_cpu_contextgs )cC`st|S(N(tcus(ti((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pytget_gpuoscC`s ttƒS(N(RRB(((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt get_num_gpusrscC`s#tstjt|ƒƒntjS(s=Get the current device or use a device by device number. (RR2RDR(tdevnum((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR4yscC`std|ƒjƒS(s]Get the current device or use a device by device number, and return the HSA context. RF(R4R1(RF((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt get_contextscC`s&gttƒƒD]}t|ƒ^qS(N(trangeRERG(RC((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pytget_all_contextsˆsc`s"tjˆƒ‡fd†ƒ}|S(s? A decorator to ensure a context for the HSA subsystem c`stƒˆ||ŽS(N(RG(targstkws(tfn(s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyt_require_cu_context‘s(t functoolstwraps(RLRM((RLs7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pytrequire_contextŒscC`stjƒtjƒdS(N(RBRRtclear(((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyR™s (R"t __future__RRRRNtnumbaRR RRR>tobjectRRBRRR;RARDREtTLStackRR4RGRIRPR(((s7lib/python2.7/site-packages/numba/roc/hsadrv/devices.pyts$ / '