o I h @s,dZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z mZmZmZddlmZddlmZmZmZmZdd lmZeejGd d d ZeejGd d d ZejdkrtdejdefddZnejdkrdejdefddZn dejdefddZGdddZejdkrddZ nejdkrddZ nddZ dedddefddZ!defd d!Z"ddefd#d$Z#defd%d&Z$de%efd'd(Z&d)eee%efde%efd*d+Z'd)eee%efde%efd,d-Z(de%efd.d/Z)de%efd0d1Z*defd2d3Z+defd4d5Z,de%efd6d7Z- dd8e.d9e.de/e%e.e e.e.gdfffd:d;Z0de.de%e.fd?d@Z2dAe.fdBdCZ3eejGdDdEdEZ4d>e.de%e.fdFdGZ5dHe e.de.fdIdJZ6dKe%e4ddfdLdMZ7dNe8de%e4fdOdPZ9dKe%e4de8fdQdRZ:dSe%edTe.de/e8e8ffdUdVZ;dWee?ej@ZAd]eddfd^d_ZBdSe%ed`e.fdadbZCdcddZDdee8dSe%efdfdgZEdSe%ede%e4fdhdiZFdKe%eddfdjdkZGdSe%eddfdldmZHdSe%efdndoZIdSe%eddfdpdqZJdrdsZKdSe%ede%efdtduZLdSe%eddfdvdwZMdSe%eddfdxdyZNdzd{d|d}ZOdSe%eddfd~dZPdejQdSe%eddfddZRdSe%ede.fddZSde.fddZTdddZUdddZVdeWfddZXddZYe?ejZZ[e?ej\Z]e?ej^Z_e?ej`Zade.fddZbddSe%efddZcddZdde%eWfddZeddejQfddZfdSe%ede%eWde c CrAr ) rr*r+r,r-r.r/r0r2rCrrrrDqrEc Cs$t|j|j|j|j|j|j|jdSr )rr*r+r,r-r.r/r0rCrrrrDsc@seZdZdefddZdS) _NotProvidedrcCdS)NrHrrrrrrz_NotProvided.__repr__N)r#r$r%r)rrrrrrHsrHr> cCs|dvS)N) LOAD_ATTR LOAD_GLOBALLOAD_SUPER_ATTRrnamerrrinst_has_op_bitsr8rRcCs|dkS)NrNrrPrrrrRr8cCrI)NFrrPrrrrRrJ)r,r-rcCst|r|dur td|durd}n|du|tu|du}|dkr'td|dur4t|ts4tdttj|||||dS)a At most one of `arg`, `argval`, and `target` can be not None/_NotProvided. This is to prevent ambiguity, e.g. does create_instruction("LOAD_CONST", 5) mean load the constant at co_consts[5], or load the constant 5? If `arg` is not provided, it will be computed during assembly from `argval` or `target`. Bits in the args of instructions LOAD_GLOBAL, LOAD_ATTR (3.12+), and LOAD_SUPER_ATTR modify the behavior of the instruction. In this case, we allow both `arg` and `argval` to be set. The value of `arg` here is expected to be the value of the op bits and the true value of `arg` will be computed during assembly. If `arg` is not set, the bits are assumed to be 0. Nz*target cannot be specified for instructionrr z@only one of arg, argval, and target can be not None/_NotProvidedz#instruction arg must be int or None)r*r+r,r-r)rR RuntimeErrorrH isinstancer'rdisopmap)rQr,r-rcntrrrcreate_instructions rXcCtjdkrdnd}t||dS)NrF JUMP_FORWARD JUMP_ABSOLUTErsys version_inforX)rinstrrrcreate_jump_absolute raTcCs&|r t|s Jd|td|dS)z In general we should only create `LOAD_CONST` for immutable objects, but sometimes it's convenient _and safe_ for Dynamo create `LOAD_CONST` for mutable objects. In such cases, use `checked=False`. zunsafe constant LOAD_CONSTr-)rrX)valcheckedrrrcreate_load_consts rgcCstjdkr tdddStdS)NrFCOPYr r,DUP_TOPr]rrrrcreate_dup_tops  rkcCs~|dkrgStjdkrddt|ddDStjdkr'|dkr'td|d |d kr8td gd |d gStd|dgS)a Returns a "simple" sequence of instructions that rotates TOS to the n-th position in the stack. For Python < 3.11, returns a single ROT_* instruction. If no such instruction exists, an error is raised and the caller is expected to generate an equivalent sequence of instructions. For Python >= 3.11, any rotation can be expressed as a simple sequence of swaps. r rFcSsg|]}td|dqS)SWAPri)rX.0r@rrr z create_rot_n..r> zrotate z not supported for Python < 3.10ROT_)TWOTHREEFOURr ROT_Nri)r^r_rangeAttributeErrorrXnrrr create_rot_ns  r inst_or_instscs t|tr |gn|fdd}fdd}tjdkr6tdjr-|ds-|dStdgStjdkrftdjrL|dsL|dSd jd kr]|d s]|d StdgStjd krtd jr||d s||d StdgS) a Appends or prepends a PUSH_NULL instruction to `inst_or_insts`, depending on Python version. Used when you know that `inst_or_insts` generates a callable that will be called. NOTE: Assumes `inst_or_insts` is a single instruction or sequence of instructions that pushes exactly 1 object to the stack that is to be called. It is important that you include ALL instructions that construct the callable - not just the first instruction/a prefix. Will attempt to use the NULL push bit for instructions with such bits (LOAD_GLOBAL 3.11+, LOAD_ATTR 3.12+, LOAD_SUPER_ATTR). In this case, instructions WILL be modified. cs$|jdus J|jd@dkSNr riidxinstsrrinst_has_bit_setsz'add_push_null..inst_has_bit_setcs(|jdus J|jdO_dSrrirrrr set_inst_bitsz#add_push_null..set_inst_bitr=rq PUSH_NULLrKrrNrF)rTrr^r_rRr+rX)rrrrrr add_push_nulls4        rcCst|tr |g}n|}tjdkr|Stjdkrdnd}||jdkr?||jdus+J||jd@dkr?||jdO_|StjdkrM|tdg}|Stdg|}|S) zLike add_push_null, but the low bit of LOAD_ATTR/LOAD_SUPER_ATTR is not set, due to an expected CALL_FUNCTION_EX instruction. rFr=rqrrNNr r)rTrr^r_r+r,rX)rrrrrradd_push_null_call_function_ex3s   rcCstjdkr=g}|r$|tdtjdkr|dn|d}|t|tjdkr2|td|d|td |d|Std |dgS) aS Creates a sequence of instructions that makes a function call. `push_null` is used in Python 3.11+ only. It is used in codegen when a function call is intended to be made with the NULL + fn convention, and we know that the NULL has not been pushed yet. We will push a NULL and rotate it to the correct position immediately before making the function call. `push_null` should be True if no NULL is pushed for the callable. Conversely, `push_null` should be False if a NULL was pushed for the callable. Prefer using `push_null=False` when possible since we will not need to rotate NULL to the right place, which is less efficient. Generally, you should codegen a function by using `add_push_null` then `create_call_function` with `push_null=False`. Example of when to set push_null False: insts = [ create_instruction("LOAD_GLOBAL", argval="torch"), create_instruction("LOAD_ATTR", argval="nn"), create_instruction("LOAD_ATTR", argval="functional"), create_instruction("LOAD_ATTR", argval="relu"), ] insts = add_push_null(insts) insts.append(create_instruction("LOAD_FAST", argval="x")) insts.extend(create_call_function(1, False)) Example of when to set push_null True: insts = [create_instruction("LOAD_FAST", x)] for should_wrap, wrapper_name in wrappers: if should_wrap: insts.extend([ create_instruction("LOAD_GLOBAL", argval="wrapper1"), create_instruction("SWAP", arg=2), *create_call_function(1, True), ) rFrr=r r rKPRECALLriCALL CALL_FUNCTION)r^r_appendrXextendr)nargs push_nulloutputrotsrrrcreate_call_functionPs ) rcCsHtjdkr td|dgStjdkrtd|dtd|dgStd|dgS)NrKrrirFr CALL_METHODr])rrrrcreate_call_methods    rcCs$tjdkr tdd|dStd|dS)NrKrMr )r,r- LOAD_METHODrdr]rPrrrcreate_load_methods  rcCrY)NrF BEFORE_WITH SETUP_WITHr\r])rr+rrrcreate_setup_withrbrc Cstjdkr td|dgS|dkrgS td|ddtdtddd td td tdtd tddd td tdtd g tdtdtd|ddS)NrFrlrir BUILD_LISTrjrcrqrd BINARY_SUBSCR ROT_THREE STORE_SUBSCRreverserPOP_TOPUNPACK_SEQUENCE)r^r_rXrrr}rrr create_swaps.     rlinenobytenocs*tjdksJgfdd}|fS)z Used to create typing.CodeType.co_lnotab See https://github.com/python/cpython/blob/main/Objects/lnotab_notes.txt This is the internal format of the line number table if Python < 3.10 rrcs|ks|krAtdt|d}tdt|d}|dks&|dks&J|7|7||d@f|ks|ksdSdS)Nrimaxminr) lineno_new byteno_new byte_offset line_offsetrrlnotabrrupdateszlnotab_writer..updater^r_)rrrrrr lnotab_writers rcs`tjdkr tjdks Jg|ddfddfdd}fdd }||fS) z Used to create typing.CodeType.co_linetable See https://github.com/python/cpython/blob/main/Objects/lnotab_notes.txt This is the internal format of the line number table for Python 3.10 rrrFrcs~|dks|dkr=tdt|d}tdt|d}|dks"|dks"J||8}||8}||d@f|dks|dksdSdS)Nrirrr) byteno_delta lineno_deltarr) linetablerr_updatesz%linetable_310_writer.._updatecs&|}||||dSr r)rrr)rrrrrrrs  z$linetable_310_writer..updatecs|dSr r) total_bytes)rrrrrrz!linetable_310_writer..endr) first_linenorrr)rrrrrrlinetable_310_writers   rr~cCsX|dksJ|d@g}|dL}|dkr*|ddO<||d@|dL}|dks|S)z~ 6-bit chunk encoding of an unsigned integer See https://github.com/python/cpython/blob/3.11/Objects/locations.md r?rq@)r)r~brrr encode_varints  rrcs.tjdksJg|dfdd }|fS)z Used to create typing.CodeType.co_linetable See https://github.com/python/cpython/blob/3.11/Objects/locations.md This is the internal format of the line number table for Python 3.11 rFr2r1csfrjnd}fdd}|durd}n|}||dkr,||d|d8}|dks|||dS)Ncsd|kr dksJJd}r?jdur?jdur?jdur?jdur?d|djjjdjdf}n d|d|dkrT| d>dB}n|dK}t||D] }t|qadS)Nrrr )r end_lineno col_offsetend_col_offsetrrr)deltasize other_varintsr~)rr2rrr s*     z5linetable_311_writer..update.._updaterr)r)r2 inst_sizerrrrr)r2rrs  z$linetable_311_writer..updateN)r2r1r)rrrrrlinetable_311_writers .rc@s6eZdZUeed<eed<eed<eed<eed<dS)ExceptionTableEntryrrrrrN)r#r$r%r'r&r(rrrrrMs  rcCsv|dksJ|d@g}|dL}|dkr"||d@|dL}|dks|tt|dD] }||dO<q.|S)zR Similar to `encode_varint`, but the 6-bit chunks are ordered in reverse. rrrr r)rrr{len)r~rr@rrrencode_exception_table_varintWs  r bytes_itercCs@t|}|d@}|d@r|dK}t|}||d@O}|d@s |S)z5 Inverse of `encode_exception_table_varint`. rrr)next)rrrerrrdecode_exception_table_varintgs rtabcCsftt|dD](}||j||jkr.||j||djkr.||dj||djks0JqdS)z Verifies that a list of ExceptionTableEntries will make a well-formed jump table: entries are non-empty, sorted, and do not overlap. r N)r{rrr)rr@rrrcheck_exception_tablets "rexntabc Cst|}g}z3 t|d}t|d}||d}t|d}t|}|d?}t|d@} |t||||| qtyHt||YSw)z Parse the exception table according to https://github.com/python/cpython/blob/3.11/Objects/exception_handling_notes.txt Tr r )iterrr(rr StopIterationr) r exntab_iterrrlengthrrdlrrrrrparse_exception_tables"      rcCsg}|D]@}t|jd}|ddO<|||j|jd}|t|d|t|jd|jd>|j}|t|qt|S)zd Inverse of parse_exception_table - encodes list of exception table entries into bytes. r rr )rrrrrrrbytes)rrentry first_entryrrrrrassemble_exception_tables r instructions firstlinenoc Csng}tjdkrpt|\}}d}t|D][\}}|jdkr;d}|d7}dD]} ||| jdkr9||| j|_nq$n t|d|}d}||j|d}|jpQd} ||j | d@ft t|ddD]} |dqfqn?tjd kr|t |\}}nt |\}}} |D]}|j d ur||j t||jpd} ||j | d@fqtjd kr| t|t|t|fS) z)Do the opposite of dis.get_instructions()rFr EXTENDED_ARGr r r r>r r)rrrrN)r^r_r enumerater+r2instruction_sizer,rr*r{rrr/rr) rrcoder update_linenonum_extr@r`rjr,_rrrrassemblesF           roffset_to_instr.cCs2dD]}|||jtjkr|||SqdS)zU Get the instruction located at a given offset, accounting for EXTENDED_ARGs )rr rurNr*rUr)rr.r~rrr_get_instruction_by_offsets rcCsBdd|D}|D]}|jtjvs|jtjvrt||j|_q dS)z9Replace jump targets with pointers to make editing easiercSsi|]}|j|qSr)r.rnr`rrr z$virtualize_jumps..N)r*rUhasjabshasjrelrr-r)r jump_targetsr`rrrvirtualize_jumpss r instructioncCsrtjdkr tdd|jvr|jdd|_nd|jvr%|jdd|_ntdtj|j|_|jt vs7JdS)NrFz+Cannot flip jump direction in Python < 3.11FORWARDBACKWARDz-Instruction is not a forward or backward jump) r^r_rSr+replacer|rUrVr* _REL_JUMPS)rrrrflip_jump_directions   rrcCsD||}dD]}||kr|||jtjkr|||}q|S|S)zp i.e. get the first EXTENDED_ARG instruction (if any) when targeting instructions[idx] with a jump. rr)rrrr.rrr_get_instruction_fronts rcCsNttjttj}|D]5}|j|vrB|jtjvrB|jj|jkr4tj dkr*t dd|j vr3t |q tj dkrBd|j vrBt |q t |t|}|D]W}|j|vrt|||j}|jtjvr~tj dkrl|j|_n-tj dkrzt|jd|_nt dtt|j|jt||_tj dkr|jd_|j|_d|j|_qMd S) zJFill in args for virtualized jump target after instructions may have movedrFz*Got negative jump offset for Python < 3.11rrrrr z+Python 3.11+ should not have absolute jumpszto N)setrUrunionrr*rr.r^r_rSr+rupdate_offsetsr rr,r'absrr-r4)rjumpsr`indexofrrrrdevirtualize_jumpssB          r exn_tab_bytescst|}dd|Dtdt|z3fdd}|\}}|D]}|j|jkr;|\}}|j|jks0|j|jkrGt||_q(WdSt yTYdSw)zDReplace exception table entries with pointers to make editing easiercSsi|] }tt|j|qSr)rr'r.rrrrr7sz.virtualize_exception_table..rcst}tkr"|jkr"d7tkr"|jksdks(Jd}tt|jt|t|j|j|j}||fS)Nr r) rrrrrrrrr)r end_offset inst_entryend_offset_idx exn_tab_iterroffsetsrrstep=s    z(virtualize_exception_table..stepN) rsortedkeysrr.rrcopyr3r)rrexn_tabr rrr`rr rvirtualize_exception_table4s&       rc sit|}|D]F}|jrNt|||jjj}tt|jjjt|jjd}t|||jj j}||f}||jj |jj f}|vrJ||ksJJ||<qt ddd}dggfdd} |D]j}rdd |dkr| rdd |dksxrψdd|dkr|d krdd ksJJtdd} | |dkrˈt| |ddgdR|d|qjr| stS) zMCompute exception table in list format from instructions with exn_tab_entriesr cS|d|d fSNrr rtrrrz)compute_exception_table..keyrcsZr)}|dkr+tt|d|dg|R|dddSdSdS)zT Pop the key_stack and append an exception table entry if possible. r rr N)poprrrrexn_dictr key_stacknextirrrs "z$compute_exception_table..poprqr )r r3rrr.rr'rrrrrrrrrrr) rrr`rrrrre keys_sortedrleftrrrcompute_exception_table]sb    >   r"cCsg}|D]N}||j||jf}|r,|dd|dkr,||r,|dd|dks|rM|dd|dkrJ|dkrJ|ddksMJJ||qdS)z Checks `tab` is a properly sorted list of nested InstructionExnTabEntry's, i.e. no entries partially overlap. "Properly sorted" means entries are sorted by increasing starts, then decreasing ends. rqr rN)rrrr)rr entry_stackrrrrr!check_inst_exn_tab_entries_nesteds > r$cst|}i|D]#}|jr+||jj||jjf}|vr&|j|ks&J|j|<qfddtdddD}t|||D]}t||j||jdD] }t|||_qSqDdS)z~ Copies exception table entries to all instructions in an entry's range. Supports nested exception table entries. cg|]}|qSrr)rnrentriesrrrosz4propagate_inst_exn_table_entries..cSrrrrrrrrrz2propagate_inst_exn_table_entries..rr N) r r3rrrrr$r{r)rrr`rsorted_entriesrr@rr&r propagate_inst_exn_table_entriess(     r)cCst|}t}t|D]K\}}|jrVtjdksJt|j|vs"J|t|j|j}|j|vs4J|j |vs;J|j |vsBJ||j|krS||j ksVJJq dS)aJ Checks that exn_tab_entries of instructions are valid. An entry's start, end, and target must be in instructions. Instructions with an exn_tab_entry are located within the entry's start and end instructions. Instructions do not share exn_tab_entries. Implicitly checks for no duplicate instructions. rFN) r rrr3r^r_r6addrrr)rrexn_tab_entry_setr@r`rrrr check_inst_exn_tab_entries_valids (r,cCsdd|D|dd<dS)NcSsg|] }|jtjkr|qSrrrmrrrroz'strip_extended_args..r)rrrrstrip_extended_argssr.cCs|jr|jj|urt|dkr|d|j_|ddD] }t|j|_|j|_q|dj|_|dj|_|dj|_|dj|_|dj |_ |g|ddS)Nr rqr) r3rrrr2r*r+r,r-r)old_inst new_instsr`rrroverwrite_instructions         r1cCsJtjdksJddd}|D]}|j|vr"||j|_tj|j|_q|S)zFLOAD_METHOD puts a NULL on the stack which causes issues, so remove itrFrMr)rr)r^r_r+rUrVr*)rrewritesr`rrrremove_load_call_method s   r3cCsg}|D]H}d|jvrGtdtd|jvd}|j|_tjdkr.td|jvr'dnd|jd }ntd |jd }td dd ||g}|t ||q| |q||dd<dS) N_NONEIS_OPNOTrirKrPOP_JUMP_FORWARD_IF_TRUEPOP_JUMP_BACKWARD_IF_TRUEr\POP_JUMP_IF_TRUErcrd) r+rXr'r,r-r^r_rrr1r)rr0r`is_opjump_op replace_instsrrrremove_jump_if_nones(     r=cCsg}|D]A}|||jdvrEt|jdd}|jr&|jj|ur&||j_t|j|_|j|_tj d|_ d|_d|_ d|_ ||q||dd<dS)N) BINARY_SLICE STORE_SLICESLICESUBSCR BUILD_SLICEr ) rr+rXrr3rrr2rUrVr*r,r-)rr0r` subscr_instrrrremove_binary_store_slice5s     rD) LOAD_FASTrE) STORE_FASTrF)rFrE)LOAD_FAST_LOAD_FASTSTORE_FAST_STORE_FASTSTORE_FAST_LOAD_FASTcCstg}|D]-}|jtvr,t|j\}}|j\}}t||dt||dg}|t||q||q||dd<dS)Nrd)r+ FUSED_INSTSr-rXrr1r)rr0r`inst0inst1argval0argval1r<rrrremove_fused_load_storePs     rOrc CsF|jpd|jpd}t|jsdSg}t|D]\}}|||jdkr|jdkr||d}|jdkrt j dkr?|jdksXt j d krNt j dkrN|jd ksXt j d kr|jd krd |vs^J|t d d d|jd}||vrz|t d |dn |t d|dd|_d|_|jd kr||d}d|_d|_q||dd<dS)z3convert super() with no args into explicit arg formrNrNsuperr rrKrrFrr __class__ LOAD_DEREFrdrEr ) co_cellvars co_freevarsr co_varnamesrrr+r-r,r^r_rX) rr cell_and_freerrr`r first_var call_instrrrexplicit_superas8           rYcs@gfdd}|D]}|jtjkrd|_nn|jrG|jdkrG|dtd|jd?dtd|jd ?dtd|jd ?dn=|jrl|jd krl|d td|jd ?dtd|jd ?dn|jr|jd kr|dtd|jd ?d|q tt|}|dksJ|dd<|S)z,Fill in correct argvals for EXTENDED_ARG opscs.t|D]}rdjtjkrqdS)Nrq)r{r*rUrr)r~rrrr maybe_pop_ns z&fix_extended_args..maybe_pop_nrir>rririr rr N)r*rUrr,rrXr)rr[r`addedrrZrfix_extended_argss,     r_cCs0ddl}tjdkrd|jjjj|jdSdS)NrrFr r )torchr^r__C_dynamo eval_framepy_opcode_cachesr*)r`r`rrrrs rcCs,d}|D]}|j|ks J|t|7}qdSNrr.rrr.r`rrr check_offsetss rhcCs$d}|D] }||_|t|7}qdSrerfrgrrrrs rcGsVtttt|}dd|gt|ddt|d|dDgD}dd|S)NcSs g|] }ddd|DqS) css|]}|dVqdS)03Nr)rnxrrr sz)debug_bytes...)join)rnr,rrrroszdebug_bytes..cSsg|] \}}t||kqSr)r')rnarrrrror-rqzbytes mismatch  )r{rmaprlistziprm)argsindexresultrrr debug_bytessrwcCsRt|dddd}|j|jksJt|j|j|j|jks'Jt|j|jdS)z.T)safeN)transform_code_objectco_coderw co_lnotab)rdoderrr debug_checkss"r~cCsHt|dD] \}}||ur|Sq|d|f7<t|ddS)N co_constsr )rr) code_optionsrer@vrrrget_const_indexs rc sddtdDdtffdd }tjdkr;|dus Jddtd Dd dtd d Dn=t|sAJitD]}z ||}||<WqGty]Ynwfd dd Dfddd d Dtt D]fdd}j dkrȈj t usJtjdkrj dusJ|j d>ttj d_ q~|j _ q~j dkr j t usJtjdkrj dusJ|j d>ttj d_ q~|j _ q~j dkr>j dusJj t us%J|j d>ttj dd_ q~j tvrtjdksNJtj tsYJt j dkseJtfddj D}|dd>|dd@_ q~jtvr|rtjdkrj vrj _ q~j _ q~jtvr|r|j _ q~jtvr|rj _ q~jtvrj durtj }|dksJ|_ q~dS)NcSi|]\}}||qSrrrnrrQrrrrzfix_vars..co_namesrcsbz|}W|Sty0t}|<gd|Rd<tdtks-JY|Sw)Nr)KeyErrorr)rQr)rnamesrrget_name_indexs  z fix_vars..get_name_indexrFcSrrrrrrrrrrUcSrrrrrrrrsrSrTci|]}||qSrrrnrQallnamesrrrrcrrrrrrrrscsjtuSr )r-rHr)r@rrrshould_compute_arg sz$fix_vars..should_compute_argrNr r rMrKrOr=c3s(|]}|vr |n|VqdSr rr) freenamesvarnamesrrrl/s  zfix_vars..rru)rr'r^r_callable itertoolscount IndexErrorr{rr+r-rHr,rrJrTtupler* HAS_LOCALHAS_NAMEHAS_FREE HAS_CONSTr)rrvarname_from_opargrrrQr arg_tupler)rrrr@rrrrfix_varss              rcCsP|D]#}|jtur%|jtvs|jtvs|jtvs|jtvr%|jdvr%d|_qdS)N)rNrMrO) r-rHr*rrrrr+r,)rr`rrrclear_instruction_argsNs      rcCsdg}|d|gdtjdkr|d|dtjdkr)|dn|d tjdkr8|d |d d g|S) N co_argcountco_posonlyargcount) co_kwonlyargcount co_nlocals co_stacksizeco_flagsr{rrrU co_filenameco_namerF co_qualnameco_firstlinenorr co_linetabler|co_exceptiontablerTrS)rrr^r_)rrrr get_code_keys_s&         rFcs\t}fdd|D}t|d|dksJt|}t||||t|||dS)Ncsi|]}|t|qSr)getattrrnkrrrrrpz)transform_code_object..rUrr )rrcleaned_instructionsr clean_and_assemble_instructions)rtransformationsryrrrrrrrzs  rzrrcst|tdd<d}tjdkr"tjfdd|D}|j}t||dd}|r=t|t |t t |}|s-t |t |d\}}tjd krT|d <n|d <|d <t|d <t|dhtdhkstJtjdkrtt|d<|tjfdd|DfS)NrUrrFcr%rrrrrrrorz3clean_and_assemble_instructions..)rTrrrr|rr{rrrcr%rrrrrrror)r,rr^r_typesCodeType_varname_from_opargrrrr(r_rrrrrrr")rrrrtmp_codedirtybytecoderrrrrs8      rcCs$|D] }|jdkr||j|_qdS)NKW_NAMES)r+r,r-)rconstsr`rrrpopulate_kw_names_argvals   rcCst||}t|Sr )_cached_cleaned_instructions_clone_instructionsrryrrrrrs rcCspdd|D}tt||}d|d<|D]!}||j|_|j}r5t||j||j||j|j|j|_q|S)NcSs>g|]}t|j|j|j|j|j|j|j|j|j |j |j qSr) rr*r+r,r-r.r/r0r2rr3r4rmrrrros z'_clone_instructions..) dictrsrr3rrrrr)rcopiedremapr@rrrrrs"  rcCstttt|}t|tjdkrt||j t |j |t |t ||sWtjdkr2t|tjdkrs         '  : 7 + #;  "*  0) J   &!       k'   )%