B   ›\ž ã@s¤ddlmZmZmZddlZddlmZddlmZddlm Z m Z m Z Gdd „d e ƒZ Gd d „d eƒZeƒZGd d „d e jƒZGdd„de jƒZeƒZeed<dS)é)Úprint_functionÚdivisionÚabsolute_importNé)Úcpu)ÚTargetDescriptoré)Ú dispatcherÚutilsÚtypingc@s"eZdZdZdZejdd„ƒZdS)Ú_NestedContextNc cs6|j|jf}z||_||_dVWd|\|_|_XdS)N)Ú_typing_contextÚ_target_context)ÚselfÚtyping_contextÚtarget_contextZ old_nested©rú5lib/python3.7/site-packages/numba/targets/registry.pyÚnesteds   z_NestedContext.nested)Ú__name__Ú __module__Ú __qualname__r rÚ contextlibÚcontextmanagerrrrrrr sr c@sTeZdZejZeƒZej dd„ƒZ ej dd„ƒZ e dd„ƒZ e dd„ƒZd d „Zd S) Ú CPUTargetcCs t |j¡S)N)rZ CPUContextr)rrrrÚ_toplevel_target_contextsz"CPUTarget._toplevel_target_contextcCst ¡S)N)r ZContext)rrrrÚ_toplevel_typing_context$sz"CPUTarget._toplevel_typing_contextcCs|jj}|dk r|S|jSdS)z5 The target context for CPU targets. N)Ú_nestedrr)rrrrrr)szCPUTarget.target_contextcCs|jj}|dk r|S|jSdS)z5 The typing context for CPU targets. N)rr r)rrrrrr4szCPUTarget.typing_contextcCs|j ||¡S)zˆ A context manager temporarily replacing the contexts with the given ones, for the current thread of execution. )rr)rrrrrrÚnested_context?szCPUTarget.nested_contextN)rrrrZCPUTargetOptionsZoptionsr rr Zcached_propertyrrÚpropertyrrrrrrrrs rc@seZdZeZdS)Ú CPUDispatcherN)rrrÚ cpu_targetZ targetdescrrrrrr Ksr cs,eZdZdZ‡fdd„Z‡fdd„Z‡ZS)ÚTargetRegistrya1 A registry of API implementations for various backends. Attributes ---------- ondemand: A dictionary of target-name -> function, where function is executed the first time a target is used. It is used for deferred initialization for some targets (e.g. gpu). cs tt|ƒj||Žt ¡|_dS)N)Úsuperr"Ú__init__r Ú UniqueDictÚondemand)rÚargsZkws)Ú __class__rrr$[szTargetRegistry.__init__cs2||jkr"|j|ƒ||<|j|=tt|ƒ |¡S)N)r&r#r"Ú __getitem__)rÚitem)r(rrr)_s zTargetRegistry.__getitem__)rrrÚ__doc__r$r)Ú __classcell__rr)r(rr"Os  r"r)Z __future__rrrrÚrZ descriptorsrr r r Úobjectr rr!Z Dispatcherr r%r"Zdispatcher_registryrrrrÚs  -