o Zh @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-rZcntrrrcreate_instructions rWcCtjdkrdnd}t||dS)NrF JUMP_FORWARD JUMP_ABSOLUTErsys version_inforW)rinstrrrcreate_jump_absolute r`TcCs&|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-)rrW)valcheckedrrrcreate_load_consts rfcCstjdkr tdddStdS)NrFCOPYr r,DUP_TOPr\rrrrcreate_dup_tops  rjcCs~|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)SWAPrh)rW.0r@rrr z create_rot_n..r> zrotate z not supported for Python < 3.10ZROT_)ZTWOZTHREEZFOURr ROT_Nrh)r]r^rangeAttributeErrorrWnrrr create_rot_ns  rz 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 rhidxinstsrrinst_has_bit_setsz'add_push_null..inst_has_bit_setcs(|jdus J|jdO_dSr|rhr}rrr set_inst_bitsz#add_push_null..set_inst_bitr=rp PUSH_NULLrKrrNrF)rTrr]r^rRr+rW)r{rrrrr 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=rprrNNr r)rTrr]r^r+r,rW)r{rr~rrradd_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 rKPRECALLrhCALL CALL_FUNCTION)r]r^appendrWextendrz)nargsZ push_nulloutputZrotsrrrcreate_call_functionPs ) rcCsHtjdkr td|dgStjdkrtd|dtd|dgStd|dgS)NrKrrhrFr CALL_METHODr\)rrrrcreate_call_methods    rcCs$tjdkr tdd|dStd|dS)NrKrMr )r,r- LOAD_METHODrcr\rPrrrcreate_load_methods  rcCrX)NrFZ BEFORE_WITH SETUP_WITHr[r\)rr+rrrcreate_setup_withrarc Cstjdkr td|dgS|dkrgS td|ddtdtddd td td tdtd tddd td tdtd g tdtdtd|ddS)NrFrkrhr BUILD_LISTrirbrprc BINARY_SUBSCR ROT_THREE STORE_SUBSCRreverserPOP_TOPUNPACK_SEQUENCE)r]r^rWrrrxrrr 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 rqcs|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 rqrFrcs~|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   rrycCsX|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?rp@)r)rybrrr 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)deltasizeZ other_varintsry)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)rrrvlen)ryrr@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)rrrdrrrdecode_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)rvrrr)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) rZ 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)rrentryZ 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)rrrqN)r]r^r enumerater+r2instruction_sizer,rr*rvrrr/rr) rrcoderZ update_linenoZnum_extr@r_rjr,_rrrrassemblesF           roffset_to_instr.cCs2dD]}|||jtjkr|||SqdS)zU Get the instruction located at a given offset, accounting for EXTENDED_ARGs )rr rtrNr*rUr)rr.ryrrr_get_instruction_by_offsets rcCsBdd|D}|D]}|jtjvs|jtjvrt||j|_q dS)z9Replace jump targets with pointers to make editing easiercSsi|]}|j|qSr)r.rmr_rrr z$virtualize_jumps..N)r*rUhasjabshasjrelrr-r)rZ 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+replacerwrUrVr* _REL_JUMPS)rrrrflip_jump_directions   rr~cCsD||}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)rr~rr.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.11rrrqr z+Python 3.11+ should not have absolute jumpszto N)setrUrunionrr*rr.r]r^rSr+rupdate_offsetsr rr,r'absrr-r4)rZjumpsr_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_entryZend_offset_idxZ exn_tab_iterroffsetsrrstep=s    z(virtualize_exception_table..stepN) rsortedkeysrr.rrcopyr3r)rrexn_tabrrrr_rrrvirtualize_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)poprrrrZexn_dictrZ key_stacknextirrr s "z$compute_exception_table..poprpr )r r3rrr.rr'rrrrrrrrrrr) rrr_rrrr rdZ keys_sortedr leftrr rcompute_exception_table]sb    >   rcCsg}|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. rpr rN)rrr r)rrZ entry_stackrr rrr!check_inst_exn_tab_entries_nesteds > rcst|}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)rmr entriesrrrnsz4propagate_inst_exn_table_entries..cSrrrrrrrrrz2propagate_inst_exn_table_entries..rr N) r r3rrrrrrvr)rrr_r Zsorted_entriesrr@rrr propagate_inst_exn_table_entriess(     rcCst|}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)rrZexn_tab_entry_setr@r_rrrr check_inst_exn_tab_entries_valids (rcCsdd|D|dd<dS)NcSsg|] }|jtjkr|qSrrrlrrrrnz'strip_extended_args..r)rrrrstrip_extended_argssrcCs|jr|jj|urt|dkr|d|j_|ddD] }t|j|_|j|_q|dj|_|dj|_|dj|_|dj|_|dj |_ |g|ddS)Nr rpr) r3rrrr2r*r+r,r-r)Zold_inst new_instsr_rrroverwrite_instructions         rcCsJtjdksJddd}|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*)rZrewritesr_rrrremove_load_call_method s   rcCsg}|D]H}d|jvrGtdtd|jvd}|j|_tjdkr.td|jvr'dnd|jd }ntd |jd }td dd ||g}|t ||q| |q||dd<dS) NZ_NONEIS_OPNOTrhrKrZPOP_JUMP_FORWARD_IF_TRUEZPOP_JUMP_BACKWARD_IF_TRUEr[POP_JUMP_IF_TRUErbrc) r+rWr'r,r-r]r^rrrr)rrr_Zis_opZjump_op replace_instsrrrremove_jump_if_nones(     rcCsg}|D]A}|||jdvrEt|jdd}|jr&|jj|ur&||j_t|j|_|j|_tj d|_ d|_d|_ d|_ ||q||dd<dS)N)Z BINARY_SLICEZ STORE_SLICEZSLICEZSUBSCR BUILD_SLICEr ) rr+rWrr3rrr2rUrVr*r,r-)rrr_Z subscr_instrrrremove_binary_store_slice5s     r!) LOAD_FASTr") STORE_FASTr#)r#r")ZLOAD_FAST_LOAD_FASTZSTORE_FAST_STORE_FASTZSTORE_FAST_LOAD_FASTcCstg}|D]-}|jtvr,t|j\}}|j\}}t||dt||dg}|t||q||q||dd<dS)Nrc)r+ FUSED_INSTSr-rWrrr)rrr_Zinst0Zinst1Zargval0Zargval1rrrrremove_fused_load_storePs     r%rc 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_DEREFrcr"r ) co_cellvars co_freevarsr co_varnamesrrr+r-r,r]r^rW) rrZ cell_and_freerr~r_r Z first_varZ call_instrrrexplicit_superas8           r,cs@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)Nrp)rvr*rUrr )ryrrrr maybe_pop_ns z&fix_extended_args..maybe_pop_nrir>rrhrir rr N)r*rUrr,rrWr)rr.r_addedrr-rfix_extended_argss,     r2cCs0ddl}tjdkrd|jjjj|jdSdS)NrrFr r )torchr]r^Z_CZ_dynamoZ eval_frameZpy_opcode_cachesr*)r_r3rrrrs rcCs,d}|D]}|j|ks J|t|7}qdSNrr.rrr.r_rrr check_offsetss r7cCs$d}|D] }||_|t|7}qdSr4r5r6rrrrs rcGsVtttt|}dd|gt|ddt|d|dDgD}dd|S)NcSs g|] }ddd|DqS) css|]}|dVqdS)Z03Nr)rmxrrr sz)debug_bytes...)join)rmr,rrrrnszdebug_bytes..cSsg|] \}}t||kqSr)r')rmarrrrrnrrpzbytes mismatch  )rvrmaprlistzipr;)argsindexresultrrr debug_bytessrEcCsRt|dddd}|j|jksJt|j|j|j|jks'Jt|j|jdS)z.T)safeN)transform_code_objectco_coderE co_lnotab)rZdoderrr debug_checkss"rKcCsHt|dD] \}}||ur|Sq|d|f7<t|ddS)N co_constsr )rr) code_optionsrdr@vrrrget_const_indexs rOc 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|]\}}||qSrrrmr~rQrrrrzfix_vars..co_namesrcsbz|}W|Sty0t}|<gd|Rd<tdtks-JY|Sw)NrS)KeyErrorr)rQr~)rMnamesrrget_name_indexs  z fix_vars..get_name_indexrFcSrPrrrQrrrrrRr+cSrPrrrQrrrrsr)r*ci|]}||qSrrrmrQallnamesrrrrRcrWrrrXrYrrrscsjtuSr )r-rHr)r@rrrshould_compute_arg sz$fix_vars..should_compute_argrNr r rMrKrOr=c3s(|]}|vr |n|VqdSr rrX) freenamesvarnamesrrr:/s  zfix_vars..rrt)rr'r]r^callable itertoolscount IndexErrorrvrr+r-rHr,rr$rTtupler* HAS_LOCALHAS_NAMEHAS_FREE HAS_CONSTrO)rrMvarname_from_opargrVr~rQr[Z arg_tupler)rZrMr\r@rrUr]rfix_varss              ricCsP|D]#}|jtur%|jtvs|jtvs|jtvs|jtvr%|jdvr%d|_qdS)N)rNrMrO) r-rHr*rdrerfrgr+r,)rr_rrrclear_instruction_argsNs      rjcCsdg}|d|gdtjdkr|d|dtjdkr)|dn|d tjdkr8|d |d d g|S) N co_argcountco_posonlyargcount) co_kwonlyargcount co_nlocals co_stacksizeco_flagsrIrLrSr+ co_filenameco_namerFZ co_qualnameco_firstlinenorq co_linetablerJco_exceptiontabler*r))rrr]r^)rrrr get_code_keys_s&         rvFcs\t}fdd|D}t|d|dksJt|}t||||t|||dS)Ncsi|]}|t|qSr)getattrrmkrrrrroz)transform_code_object..r+rnr )rvrcleaned_instructionsr clean_and_assemble_instructions)rZtransformationsrGrrMrrrzrrHs  rHrrMcst|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)Nr+rnrFcrrrrxrMrrrnrz3clean_and_assemble_instructions..)rhTrsrqrJrtrIrorlrucrrrrxr}rrrnr)rrr]r^typesCodeTypeZ_varname_from_opargrirrr(r2rrrrrrr)rrrMrhZtmp_codeZdirtybytecoderrr}rr|s8      r|cCs$|D] }|jdkr||j|_qdS)NZKW_NAMES)r+r,r-)rZconstsr_rrrpopulate_kw_names_argvals   rcCst||}t|Sr )_cached_cleaned_instructions_clone_instructionsrrGrrrrr{s r{cCspdd|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/r0r2rr3r4rlrrrrns z'_clone_instructions..) dictrArr3rrrrr)rZcopiedZremapr@rrrrrs"  rcCstttt|}t|tjdkrt||j t |j |t |t ||sWtjdkr2t|tjdkrs         '  : 7 + #;  "*  0) J   &!       k'   )%