gdZddlZddlZddlmZGddZGddejZGdd 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>eZdZdZejZdZdZdZ dZ dZ dS) RetargetCachez^Cache for retargeted dispatchers. The cache uses the original dispatcher as the key. cT||_d|_d|_dS)Nr)container_type_cache _stat_hit _stat_missselfs I/granary/edev/quantum/lib/python3.11/site-packages/numba/core/retarget.py__init__zRetargetCache.__init__s'))++ c||j|<dS)z9Save a dispatcher associated with the given key. N)r)r orig_dispnew_disps r save_cachezRetargetCache.save_caches"* Irc|j|}||xjdz c_n|xjdz c_|S)z9Load a dispatcher associated with the given key. N)rgetr r )r routs r load_cachezRetargetCache.load_cachesCkooi(( ; OOq OOO NNa NN rc4|jS)z+Returns the contents of the cache. )ritemsr s r rzRetargetCache.items(s{  """rc |j|jdS)z0Returns stats regarding cache hit/miss. )hitmiss)r r r s r statszRetargetCache.stats-s~t???rN) __name__ __module__ __qualname____doc__weakrefWeakKeyDictionaryrrrrrrrr rr sx.N *** ### @@@@@rrcReZdZdZejdZejdZdS) BaseRetargetz/Abstract base class for retargeting logic. cdS)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 rs r check_compatiblezBaseRetarget.check_compatible6s  rcdS)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)s r retargetzBaseRetarget.retarget? rN)rr r!r"abcabstractmethodr*r,r%rr r'r'3sZ         rr'cneZdZdZeZdZejde fdZ ej dZ dZ dZdS) 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. c8||_dS)N) cache_typecacher s r rzBasicRetarget.__init__Vs__&& rreturncdS)zhReturns the output target name. See numba/tests/test_retargeting.py for example usage. Nr%r s r output_targetzBasicRetarget.output_targetYr-rcdS)zkReturns the retargeted dispatcher. See numba/tests/test_retargeting.py for example usage. Nr%r)s r compile_retargetzBasicRetarget.compile_retargetar-rcp|j}|j}|#||krd|d|d}tj|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_backendr7r CompilerError)r rrequired_targetr7ms r r*zBasicRetarget.check_compatibleisj $<*  &//>%>>+:>>>*1--- ' &//rc|j}|j}|d|jkr|S||}|,||}|||n|}|S)zgApply retargeting to orig_disp. The retargeted dispatchers are cached for future use. target_backend)r4 targetoptionsrr7rr9r)r rr4optscachedrs r r,zBasicRetarget.retargetvs  & 88$ % %); ; ; !!),, >'' 22C   Y , , , ,C rN)rr r!r"rr3rr.abstractpropertystrr7r/r9r*r,r%rr r1r1Hs  J'''  s         . . .rr1) r"r.r# numba.corerrABCr'r1r%rr rIs  #@#@#@#@#@#@#@#@L     37   *?????L?????r