U ?hC@sHddlmZddlmZGdddeZGdddeZeZejZ dS)) defaultdict)configc@s*eZdZdZd ddZddZddZdS) Rewritez8Defines the abstract base class for Numba rewrites. NcCsdS)z+Constructor for the Rewrite class. N)selfstaterrN/var/www/html/venv/lib/python3.8/site-packages/numba/core/rewrites/registry.py__init__ szRewrite.__init__cCsdS)z]Overload this method to check an IR block for matching terms in the rewrite. Fr)rfunc_irblocktypemap calltypesrrrmatchsz Rewrite.matchcCs tddS)zgOverload this method to return a rewritten IR basic block when a match has been found. z Abstract Rewrite.apply() called!N)NotImplementedErrorrrrrapplysz Rewrite.apply)N)__name__ __module__ __qualname____doc__r rrrrrrrs rc@s4eZdZdZeddgZddZddZdd Zd S) RewriteRegistryz+Defines a registry for Numba rewrites. zbefore-inferencezafter-inferencecCstt|_dS)ziConstructor for the rewrite registry. Initializes the rewrites member to an empty list. N)rlistrewritesrrrrr !szRewriteRegistry.__init__cs*jkrtdffdd}|S)zf Decorator adding a subclass of Rewrite to the registry for the given *kind*. zinvalid kind %rcs,t|tstd|j||S)Nz {0} is not a subclass of Rewrite) issubclassr TypeErrorformatrappend) rewrite_clskindrrr do_register.s  z-RewriteRegistry.register..do_register)_kindsKeyError)rrr rrrregister's zRewriteRegistry.registercCs$||jkst|jj}|}|j|D]}||}t|}|r(|\}} | |j| |j |j } | r@t j svt jrtdtd|j| td|} | ||<||| ft j st jr@| tdq@q(|D]\}} | ||kr| qddlm} | |j} | dS)zGiven a pipeline and a dictionary of basic blocks, exhaustively attempt to apply all registered rewrites to all basic blocks. ZF______________________________________________________________________zREWRITING (%s):Z<____________________________________________________________r)postprocN)r!AssertionErrorr blockscopyrritemspoprr r rDEBUGZDUMP_IRprintrdumprrverify numba.corer$Z PostProcessorrun)rrrr&Z old_blocksrZrewriteZ work_listkeyr matchesZ new_blockr$Z post_procrrrr6s:         zRewriteRegistry.applyN) rrrr frozensetr!r r#rrrrrrs  rN) collectionsrr.robjectrrZrewrite_registryr#Zregister_rewriterrrrs  E