:i8SrSSKrSSKrSSKJr SSKJrJr "SS\5r"SS \5r "S S \5r \ "5r \ Rr SS jr S rSrg)a Expose each GPU devices directly. This module implements a API that is like the "CUDA runtime" context manager for managing CUDA context stack and clean up. It relies on thread-local globals to separate the context stack management of each thread. Contexts are also shareable among threads. Only the main thread can destroy Contexts. Note: - This module must be imported by the main-thread. N)contextmanager)driverUSE_NV_BINDINGcP^\rSrSrU4SjrSrSrSrSr\ S5r Sr U=r $) _DeviceListc>US:XaU[R"5n[U5Vs/sH"n[[R"U55PM$ nnX@lU$[ [U]#U5$s snf)Nlst) rget_device_countrange_DeviceContextManager get_devicer superr __getattr__)selfattrnumdevdevidgpus __class__s ]/srv/projetos/modelo_ml_acdoc/venv/lib/python3.13/site-packages/numba/cuda/cudadrv/devices.pyr_DeviceList.__getattr__ss 5=,,.F!&v0!.*&*;*;E*BC!. 0HK[$3D990s)A0c URU$)z2 Returns the context manager for device *devnum*. r )rdevnums r __getitem___DeviceList.__getitem__$sxxcvSRURVs/sHn[U5PM sn5$s snf)Nz, )joinr str)rds r__str___DeviceList.__str__*s+yy$((3(Q#a&(3443s6c,[UR5$N)iterr rs r__iter___DeviceList.__iter__-sDHH~rc,[UR5$r')lenr r)s r__len___DeviceList.__len__0s488}rc[R"5nURnUb XsSSS5 $SSS5 g!,(df  g=f)zFReturns the active device or None if there's no active device N)rget_active_contextr)racrs rcurrent_DeviceList.current3s= & & (BYYF!|) (!) ( (s = A r) __name__ __module__ __qualname____firstlineno__rrr$r*r.propertyr3__static_attributes__ __classcell__)rs@rrrs0 : 5$$rrc6\rSrSrSrSrSrSrSrSr Sr g ) r=a9 Provides a context manager for executing in the context of the chosen device. The normal use of instances of this type is from ``numba.cuda.gpus``. For example, to execute on device 2:: with numba.cuda.gpus[2]: d_a = numba.cuda.to_device(a) to copy the array *a* onto device 2, referred to by *d_a*. cXlgr'_device)rdevices r__init___DeviceContextManager.__init__Is rc.[URU5$r')getattrr@)ritems rr!_DeviceContextManager.__getattr__Lst||T**rcV[RURR5 gr')_runtimeget_or_create_contextr@idr)s r __enter___DeviceContextManager.__enter__Os&&t||7rcTURR5R5 gr')r@get_primary_contextpop)rexc_typeexc_valexc_tbs r__exit___DeviceContextManager.__exit__Rs ((*..0rc SRUS9$)Nzr))formatr)s rr$_DeviceContextManager.__str__Vs+222==rr?N) r5r6r7r8__doc__rBrrLrTr$r:rrrr=s  +81>rrcX\rSrSrSrSr\S5rSrSr Sr Sr S r S r S rS rg )_RuntimeZzuEmulate the CUDA runtime context management. It owns all Devices and Contexts. Keeps at most one Context per Device c[5Ul[R"5Ul[R "5Ul[R"5Ulgr') rr threadinglocal_tlscurrent_thread _mainthreadRLock_lockr)s rrB_Runtime.__init__as>M OO% %335__& rc#(# [R"5 UR5nURS5nUR U5 Sv UR U5 SSS5 g!UR U5 f=f!,(df  g=f7f)aiEnsure a CUDA context is available inside the context. On entrance, queries the CUDA driver for an active CUDA context and attaches it in TLS for subsequent calls so they do not need to query the CUDA driver again. On exit, detach the CUDA context from the TLS. This will allow us to pickup thirdparty activated CUDA context in any top-level Numba CUDA API. N)rr1_get_attached_contextrJ_set_attached_context)roldctxnewctxs rensure_context_Runtime.ensure_contextnsw & & (//1F//5F  & &v . 3**62) (**62) (s4B3B A+B" B+A>>B B BcUc&UR5nUcURU5$U$[(a [U5nUR U5$)zReturns the primary context and push+create it if needed for *devnum*. If *devnum* is None, use the active CUDA context (must be primary) or create a new one with ``devnum=0``. )rh_get_or_create_context_uncachedrint_activate_context_for)rr attached_ctxs rrJ_Runtime.get_or_create_contextsS >557L#;;FCC##~V--f5 5rcUR [R"5nU(d#URS5sSSS5 sSSS5 $URUR R 5n[(a+[UR5n[UR5nO,URRnURRnXE:waSn[URU55eUR5 UsSSS5 sSSS5 $!,(df  O=fSSS5 g!,(df  g=f)zRSee also ``get_or_create_context(devnum)``. This version does not read the cache. rNz5Numba cannot operate on non-primary CUDA context {:x})rerr1rqrrrOrrphandlecontext_handlevalue RuntimeErrorrWprepare_for_use)rrr2ctx ctx_handle ac_ctx_handlemsgs rro(_Runtime._get_or_create_context_uncachedsZZ**,55a8-,Z))BII.BBDC%~%(_ (+B,=,=(> %(ZZ%5%5 (*(9(9(?(? !2 4*3::m+DEE'')'-,Z-,,ZZs.D7D D7B;D D7 D* &D77 EcUR URUnUR5nUR5nUbXCLa [ S5eUR 5 UsSSS5 $!,(df  g=f)NzCannot switch CUDA-context.)rerrOrhrxpush)rrgpurk cached_ctxs rrq_Runtime._activate_context_fors` ZZ))F#C,,.F335J%**B"#@AA KKMZZs AA** A8c0[URSS5$)Nattached_context)rErar)s rrh_Runtime._get_attached_contextstyy"4d;;rc$XRlgr')rar)rrzs rri_Runtime._set_attached_contexts %( "rc[R"5b[R"5bM[R"5UR:XaUR 5 gg)zaClear all contexts in the thread. Destroy the context if and only if we are in the main thread. N)rpop_active_contextr_rbrc_destroy_all_contextsr)s rreset_Runtime.resetsQ '')5 '')5  # # %)9)9 9  & & ( :rcJURHnUR5 M gr')rr)rrs rr_Runtime._destroy_all_contextss99C IIKr)rercrarN)r5r6r7r8rYrBrrlrJrorqrhrirrr:rZrrr\r\ZsC '33&6 8 <) )rr\c,[RU5$)zVGet the current device or use a device by device number, and return the CUDA context. )rIrJ)rs r get_contextrs  ) )& 11rcF^[R"T5U4Sj5nU$)z A decorator that ensures a CUDA context is available when *fn* is executed. Note: The function *fn* cannot switch CUDA-context. cr>[R5 T"U0UD6sSSS5 $!,(df  g=fr')rIrl)argskwsfns r_require_cuda_context.require_context.._require_cuda_contexts(  $ $ &t#s#' & &s( 6) functoolswraps)rrs` rrequire_contextrs( __R$$ ! rc,[R5 g)zReset the CUDA subsystem for the current thread. In the main thread: This removes all CUDA contexts. Only use this at shutdown or for cleaning up between tests. In non-main threads: This clear the CUDA context stack only. N)rIrrZrrrrs NNrr')rYrr_ contextlibrrrobjectrrr\rIrrrrrZrrrsc %*&$&&$R>F>:tvtn :}}2 ! r