o Tæ«dn ã@sîddlZddlmZmZmZmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5ddl6m7Z7dd l8m9Z9dd l:m;Z;dd lZ>dd l?Tdd l?m@Z@ddlAmBZBmCZCmDZDddlEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMddlNmOZOeOjPZQZPe D]ZReSejTeUeRjUeRƒeRjVD] ZWeSejTeUeWeRƒqÖqÇ[R[ dd„ZXdd„ZYdd„ZZe9 [¡dS)éNé)Ú threadIdxÚblockIdxÚblockDimÚgridDimÚlaneidÚwarpsizeÚsyncwarpÚsharedÚlocalÚconstÚatomicÚshfl_sync_intrinsicÚvote_sync_intrinsicÚmatch_any_syncÚmatch_all_syncÚthreadfence_blockÚthreadfence_systemÚ threadfenceÚselpÚpopcÚbrevÚclzÚffsÚfmaÚcbrtÚcgÚ activemaskÚ lanemask_ltÚ nanosleepÚfp16Ú_vector_type_stubs)ÚgridÚgridsizeÚ syncthreadsÚsyncthreads_andÚsyncthreads_countÚsyncthreads_or)ÚCudaSupportError) ÚBaseCUDAMemoryManagerÚHostOnlyCUDAMemoryManagerÚGetIpcHandleMixinÚ MemoryPointerÚ MappedMemoryÚ PinnedMemoryÚ MemoryInfoÚ IpcHandleÚset_memory_manager)Úruntime)Únvvm)Ú initialize)ÚKernelRuntimeError)ÚjitÚdeclare_device)Ú*)Ú _auto_device)ÚInÚOutÚInOut)Úall_syncÚany_syncÚeq_syncÚ ballot_syncÚ shfl_syncÚ shfl_up_syncÚshfl_down_syncÚ shfl_xor_sync)Ú reductioncCs0d}ztjj}Wn tyYnw|ot ¡S)z†Returns a boolean to indicate the availability of a CUDA GPU. This will initialize the driver if it hasn't been initialized. F)ÚdriverÚ is_availabler(r3)Údriver_is_available©rIúi/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/cuda/device_init.pyrG,s   ÿ rGcCst ¡S)aReturns True if the CUDA Runtime is a supported version. Unsupported versions (e.g. newer versions than those known to Numba) may still work; this function provides a facility to check whether the current Numba version is tested and known to work with the current runtime version. If the current version is unsupported, the caller can decide how to act. Options include: - Continuing silently, - Emitting a warning, - Generating an error or otherwise preventing the use of CUDA. )r2Úis_supported_versionrIrIrIrJrK?srKcCstjjS)z§Returns None if there was no error initializing the CUDA driver. If there was an error initializing the driver, a string describing the error is returned. )rFÚinitialization_errorrIrIrIrJÚ cuda_errorPsrM)\ÚsysÚstubsrrrrrrr r r r r rrrrrrrrrrrrrrrrrrr r!Ú intrinsicsr"r#r$r%r&r'Ú cudadrv.errorr(Únumba.cuda.cudadrv.driverr)r*r+r,r-r.r/r0r1Únumba.cuda.cudadrv.runtimer2Úcudadrvr3Ú numba.cudar4Úerrorsr5Ú decoratorsr6r7Úapir9Úargsr:r;r<Úintrinsic_wrapperr=r>r?r@rArBrCrDÚkernelsrEÚReduceÚreduceÚvector_type_stubÚsetattrÚmodulesÚ__name__ÚaliasesÚaliasrGrKrMÚinitialize_allrIrIrIrJÚs4„  ,     (   ÿ