U |e@sTdZddlZddlZddlmZGdddZGdddejZGdd d eZdS) z Implement utils for supporting retargeting of dispatchers. WARNING: Features defined in this file are experimental. The API may change without notice. N)errorsc@s>eZdZdZejZddZddZddZ dd Z d d Z d S) RetargetCachez^Cache for retargeted dispatchers. The cache uses the original dispatcher as the key. cCs||_d|_d|_dS)Nr)container_type_cache _stat_hit _stat_missselfr P/var/www/website-v5/atlas_env/lib/python3.8/site-packages/numba/core/retarget.py__init__s zRetargetCache.__init__cCs||j|<dS)z9Save a dispatcher associated with the given key. N)r)r orig_dispZnew_dispr r r save_cacheszRetargetCache.save_cachecCs6|j|}|dkr$|jd7_n|jd7_|S)z9Load a dispatcher associated with the given key. N)rgetrr)r r outr r r load_caches  zRetargetCache.load_cachecCs |jS)z+Returns the contents of the cache. )ritemsrr r r r(szRetargetCache.itemscCs|j|jdS)z0Returns stats regarding cache hit/miss. )hitmiss)rrrr r r stats-szRetargetCache.statsN) __name__ __module__ __qualname____doc__weakrefWeakKeyDictionaryrr rrrrr r r r r s rc@s,eZdZdZejddZejddZdS) BaseRetargetz/Abstract base class for retargeting logic. cCsdS)zCheck that the retarget is compatible. This method does not return anything meaningful (e.g. None) Incompatibility is signalled via raising an exception. Nr r r r r r check_compatible6szBaseRetarget.check_compatiblecCsdS)zRetargets the given dispatcher and returns a new dispatcher-like callable. Or, returns the original dispatcher if the the target_backend will not change. Nr rr r r retarget?szBaseRetarget.retargetN)rrrrabcabstractmethodrr r r r r r3s  rc@sNeZdZdZeZddZeje dddZ ej ddZ d d Z d d Zd S) BasicRetargetaA basic retargeting implementation for a single output target. This class has two abstract methods/properties that subclasses must define. - `output_target` must return output target name. - `compile_retarget` must define the logic to retarget the given dispatcher. By default, this class uses `RetargetCache` as the internal cache. This can be modified by overriding the `.cache_type` class attribute. cCs||_dS)N) cache_typecacherr r r r VszBasicRetarget.__init__)returncCsdS)zhReturns the output target name. See numba/tests/test_retargeting.py for example usage. Nr rr r r output_targetYszBasicRetarget.output_targetcCsdS)zkReturns the retargeted dispatcher. See numba/tests/test_retargeting.py for example usage. Nr rr r r compile_retargetaszBasicRetarget.compile_retargetcCs<|j}|j}|dk r8||kr8d|d|d}t|dS)zt This implementation checks that `self.output_target == orig_disp._required_target_backend` Nz2The output target does match the required target: z != .)_required_target_backendr'r CompilerError)r r Zrequired_targetr'mr r r ris zBasicRetarget.check_compatiblecCsR|j}|j}|d|jkr |S||}|dkrJ||}|||n|}|S)zgApply retargeting to orig_disp. The retargeted dispatchers are cached for future use. target_backendN)r% targetoptionsrr'rr(r)r r r%optscachedrr r r r vs  zBasicRetarget.retargetN)rrrrrr$r r!abstractpropertystrr'r"r(rr r r r r r#Hs   r#) rr!r numba.corerrABCrr#r r r r s  &