o 8ήc@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 :/tmp/pip-target-vg8gfxp4/lib/python/numba/core/retarget.py__init__s  zRetargetCache.__init__cCs||j|<dS)z9Save a dispatcher associated with the given key. N)r)r orig_dispnew_dispr r r save_cachezRetargetCache.save_cachecCs8|j|}|dur|jd7_|S|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(s zRetargetCache.itemscCs|j|jdS)z0Returns stats regarding cache hit/miss. )hitmiss)rrrr r r stats-rzRetargetCache.statsN) __name__ __module__ __qualname____doc__weakrefWeakKeyDictionaryrr rrrrr r r r r s rc@s,eZdZdZejddZejddZdS) BaseRetargetz/Abstract base class for retargeting logic. cCdS)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_compatiblecCr )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 r!r r r retarget?zBaseRetarget.retargetN)rrrrabcabstractmethodr"r#r r r r r3s  rc@sNeZdZdZeZddZejde fddZ 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__returncCr )zhReturns the output target name. See numba/tests/test_retargeting.py for example usage. Nr rr r r output_targetYr$zBasicRetarget.output_targetcCr )zkReturns the retargeted dispatcher. See numba/tests/test_retargeting.py for example usage. Nr r!r r r compile_retargetar$zBasicRetarget.compile_retargetcCs@|j}|j}|dur||krd|d|d}t|dSdS)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 required_targetr+mr r r r"is zBasicRetarget.check_compatiblecCsT|j}|j}|d|jkr|S||}|dur&||}||||S|}|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,r"r#r r r r r'Hs   r') rr%r numba.corerrABCrr'r r r r s &