U ,d@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 7/tmp/pip-unpacked-wheel-eu7e0c37/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. )hitZmiss)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)rrrrabcabstractmethodrrr 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 != .)Z_required_target_backendr&rZ 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. Ztarget_backendN)r$Z targetoptionsrr&rr'r)r r r$optscachedrr r r rvs  zBasicRetarget.retargetN)rrrrrr#r r abstractpropertystrr&r!r'rrr r r r r"Hs   r") rr rZ numba.corerrABCrr"r r r r s  &