o ۾i@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)InternalTargetMismatchErrorNonexistentTargetError)localcpucseZdZfddZZS)_TargetRegistrycsLzt|WSty%d}dddtD}t|||dw)Nz7No target is registered against '{}', known targets: {} cSs&g|]\}}|ddd|qS)z < z -> ).0kvrrO/home/ubuntu/.local/lib/python3.10/site-packages/numba/core/target_extension.py sz/_TargetRegistry.__getitem__..)super __getitem__KeyErrorjointarget_registryitemsrformat)selfitemmsgknown __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__#s ztarget_override.__init__cCs |jt_dSN)r&r(rrrr __enter__( ztarget_override.__enter__cCs |jt_dSr-)r*r(r&)rtyvaltbrrr__exit__+r0ztarget_override.__exit__N)r!r"r#__doc__r,r/r4rrrrr% s  r%cCs ttdtS)zReturns the current target r&)r'r(r)rrrrcurrent_target/s r6cCsJt|jjdkr|jdj}nttd}|dur#d}t|||S)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 {}.) len callstack_stackr&rgetr6 ValueErrorr)contextr&rrrrget_local_target5sr=cCst|S)z target_hwrrrresolve_dispatcher_from_strLsrBcCs*t|}t|}||st||||S)aReturns the local target if it is compatible with the given target name during a type resolution; otherwise, raises an exception. Parameters ---------- tyctx: typing context hwstr: str target name to check against reason: str Reason for the resolution. Expects a noun. Returns ------- target_hw : Target Raises ------ InternalTargetMismatchError )r?r= inherits_fromr)tyctxhwstrreasonhw_clazzrArrr_get_local_target_checkedRs   rHc@seZdZeddZdS) JitDecoratorcCstSr-)NotImplementedr.rrr__call__qszJitDecorator.__call__N)r!r"r#rrKrrrrrIosrIc@seZdZdZeddZdS)Targetz Implements a target cCs t||S)zAReturns True if this target inherits from 'other' False otherwise) issubclass)clsotherrrrrCys zTarget.inherits_fromN)r!r"r#r5 classmethodrCrrrrrLvsrLc@eZdZdZdS)GenericzmMark the target as generic, i.e. suitable for compilation on any target. All must inherit from this. Nr!r"r#r5rrrrrRrRc@rQ)CPUzMark the target as CPU. NrSrrrrrUrTrUc@rQ)GPUzOMark the target as GPU, i.e. suitable for compilation on a GPU target. NrSrrrrrVrTrVc@rQ)CUDAzMark the target as CUDA. NrSrrrrrWrTrWc@rQ)NPyUfunczMark the target as a ufunc NrSrrrrrXrTrXgenericgpucudanpyufunc)key_typeN)#abcrrnumba.core.registryrrnumba.core.decoratorsrnumba.core.errorsrr threadingr tlsr(r)r r jit_registryobjectr%r6r=r?rBrHrIrLrRrUrVrWrXr@ cpu_targetrrrrsF