ó \ K]c@`s~ddlmZmZmZddlmZddlmZdefd„ƒYZ defd„ƒYZ e ƒZ e j Z dS( i(tprint_functiontdivisiontabsolute_import(t defaultdict(tconfigtRewritecB`s)eZdZd„Zd„Zd„ZRS(s8Defines the abstract base class for Numba rewrites. cC`s ||_dS(s+Constructor for the Rewrite class. N(tpipeline(tselfR((s6lib/python2.7/site-packages/numba/rewrites/registry.pyt__init__ scC`stS(s]Overload this method to check an IR block for matching terms in the rewrite. (tFalse(Rtfunc_irtblockttypemapt calltypes((s6lib/python2.7/site-packages/numba/rewrites/registry.pytmatchscC`stdƒ‚dS(sgOverload this method to return a rewritten IR basic block when a match has been found. s Abstract Rewrite.apply() called!N(tNotImplementedError(R((s6lib/python2.7/site-packages/numba/rewrites/registry.pytapplys(t__name__t __module__t__doc__RRR(((s6lib/python2.7/site-packages/numba/rewrites/registry.pyRs  tRewriteRegistrycB`s;eZdZeddgƒZd„Zd„Zd„ZRS(s+Defines a registry for Numba rewrites. sbefore-inferencesafter-inferencecC`sttƒ|_dS(siConstructor for the rewrite registry. Initializes the rewrites member to an empty list. N(Rtlisttrewrites(R((s6lib/python2.7/site-packages/numba/rewrites/registry.pyR#sc`s;ˆˆjkr%tdˆfƒ‚n‡‡fd†}|S(sf Decorator adding a subclass of Rewrite to the registry for the given *kind*. sinvalid kind %rc`s?t|tƒs'tdj|ƒƒ‚nˆjˆj|ƒ|S(Ns {0} is not a subclass of Rewrite(t issubclassRt TypeErrortformatRtappend(t rewrite_cls(tkindR(s6lib/python2.7/site-packages/numba/rewrites/registry.pyt do_register0s  (t_kindstKeyError(RRR((RRs6lib/python2.7/site-packages/numba/rewrites/registry.pytregister)sc C`s†||jkst‚|j}|jƒ}x|j|D] }||ƒ}t|jƒƒ}xå|rC|jƒ\} } |j|| |j |j ƒ} | r_t j s­t j rçtddƒtd|jƒ| jƒtddƒn|jƒ} | || <|j| | fƒt j s"t j r@| jƒtddƒq@q_q_Wq8Wx7|jƒD])\} } | || krU| jƒqUqUWdS(sŽGiven a pipeline and a dictionary of basic blocks, exhaustively attempt to apply all registered rewrites to all basic blocks. t_iFsREWRITING (%s):i<N(RtAssertionErrortblockstcopyRRtitemstpopRR R RtDEBUGtDUMP_IRtprintRtdumpRRtverify( RRRR R#t old_blocksRtrewritet work_listtkeyR tmatchest new_block((s6lib/python2.7/site-packages/numba/rewrites/registry.pyR8s2         (RRRt frozensetRRR R(((s6lib/python2.7/site-packages/numba/rewrites/registry.pyRs   N(t __future__RRRt collectionsRtnumbaRtobjectRRtrewrite_registryR tregister_rewrite(((s6lib/python2.7/site-packages/numba/rewrites/registry.pyts ;