U ?h@s|ddlmZmZddlmZmZddlmZddlm Z m Z ddl m Z e ZdZGdddeZeZeZGd d d eZd d Zd dZddZddZddZGdddeZGdddeZGdddeZGdddeZGdddeZGdd d eZGd!d"d"eZ eed#<eed<eed<eed<eed$<eed <eed%<e ed&<eed'Z!edZ"ee!e"<eee"<d(S)))ABCabstractmethod)DelayedRegistry CPUDispatcher)jit)InternalTargetMismatchErrorNumbaValueError)localcpucseZdZfddZZS)_TargetRegistrycsVzt|WStk rPd}dddtD}t|||dYnXdS)Nz7No target is registered against '{}', known targets: {} cSs&g|]\}}|ddd|qS)z < z -> ).0kvrrM/var/www/html/venv/lib/python3.8/site-packages/numba/core/target_extension.py sz/_TargetRegistry.__getitem__..)super __getitem__KeyErrorjointarget_registryitemsrformat)selfitemmsgZknown __class__rrrs z_TargetRegistry.__getitem__)__name__ __module__ __qualname__r __classcell__rrrrr sr c@s(eZdZdZddZddZddZdS) target_overridezWContext manager to temporarily override the current target with that prescribed.cCsttdt|_||_dS)Ntarget)getattr_active_context_active_context_default _orig_targetr%)rnamerrr__init__"sztarget_override.__init__cCs |jt_dSN)r%r'rrrr __enter__'sztarget_override.__enter__cCs |jt_dSr,)r)r'r%)rtyvaltbrrr__exit__*sztarget_override.__exit__N)r r!r"__doc__r+r.r2rrrrr$sr$cCs ttdtS)zReturns the current target r%)r&r'r(rrrrcurrent_target.sr4cCsPt|jjdkr|jdj}nttd}|dkrHd}t||n|SdS)za Gets the local target from the call stack if available and the TLS override if not. rNz7The target found is not registered.Given target was {}.) lenZ callstack_stackr%rgetr4 ValueErrorr)contextr%rrrrget_local_target4sr:cCst|S)zrrr_get_local_target_checkedQs   rBc@seZdZeddZdS) JitDecoratorcCstSr,)NotImplementedr-rrr__call__pszJitDecorator.__call__N)r r!r"rrErrrrrCnsrCc@seZdZdZeddZdS)Targetz Implements a target cCs t||S)zAReturns True if this target inherits from 'other' False otherwise) issubclass)clsotherrrrr@xszTarget.inherits_fromN)r r!r"r3 classmethodr@rrrrrFusrFc@seZdZdZdS)GenericzmMark the target as generic, i.e. suitable for compilation on any target. All must inherit from this. Nr r!r"r3rrrrrK~srKc@seZdZdZdS)CPUzMark the target as CPU. NrLrrrrrMsrMc@seZdZdZdS)GPUzOMark the target as GPU, i.e. suitable for compilation on a GPU target. NrLrrrrrNsrNc@seZdZdZdS)CUDAzMark the target as CUDA. NrLrrrrrOsrOc@seZdZdZdS)NPyUfunczMark the target as a ufunc NrLrrrrrPsrPZgenericZgpucudaZnpyufunc)Zkey_typeN)#abcrrZnumba.core.registryrrZnumba.core.decoratorsrZnumba.core.errorsrr threadingr tlsr'r(r rZ jit_registryobjectr$r4r:r<r?rBrCrFrKrMrNrOrPr=Z cpu_targetrrrrsD