9j  dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZddlmZmZmZmZddlmZddlmZmZmZmZddlmZerdd lmZej>d Gd d Z ej>d GddZ!ejDdk\rdejBde!fdZ#n4ejDdk\rdejBde!fdZ#ndejBde!fdZ#GddZ$ejDdk\r de%de&fdZ'n ejDdk\r de%de&fdZ'nde%de&fdZ'de$ddde%de(dzd edzd!e!dzde!f d"Z)d!e!de!fd#Z*d!e!de&fd$Z+dd%ed&e&de!fd'Z,de-e!fd(Z.de!fd)Z/d*e(de-e!fd+Z0d,e!e-e!zde-e!fd-Z1d,e!e-e!zde-e!fd.Z2d/e(d0e&de-e!fd1Z3 dd2e&d0e&d3e&de-e!fd4Z4d/e(de-e!fd5Z5de%de!fd6Z6d!e!de!fd7Z7d*e(de-e!fd8Z8 dd9e(dzd:e(dzd;e&de-e!fd<Z9de(de-e!fd=Z:d>e(de-e!fd?Z;d@ede-e!fdAZdDe(de?e-e(e e(e(gdfe e(gdfffdEZ@d*e(de-e(fdFZAejDdk\r$dDe(de?e-e(e ejdze(gdfffdGZCej>d GdHdIZDd*e(de-e(fdJZEdKe e(de(fdLZFdMe-eDddfdNZGdOeHde-eDfdPZIdMe-eDdeHfdQZJdRe-e!dSe(de?eHeHffdTZKdUeLe(e!fdVe(de!dzfdWZMdRe e!ddfdXZNeOejZQdYe!ddfdZZRdRe-e!d[e(de!fd\ZSdRe-e!ddfd]ZTd^eHdRe-e!ddfd_ZUdRe-e!de-eDfd`ZVdMe-e daeLe!e(fddfdbZWdRe-e!ddfdcZXdRe-e!ddfddZYdRe-e!ddfdeZZdfe!dge-e!de-e!fdhZ[dRe-e!de-e!fdiZ\dRe-e!ddfdjZ]dRe-e!ddfdkZ^dldmdndodpZ_dRe-e!ddfdqZ`dRe-e!ddfdrZadRe-e!ddfdsZbdtejdRe-e!ddfduZddRe-e!de(fdvZedwe!de(fdxZfdRee!ddfdyZgdRee!ddfdzZhd{eHde%fd|Zidtejddfd}ZjeOejZleOejZneOejZpeOejZrd~eLe%efd%ede(fdZs ddRe-e!d~eLe%efde defdzddfdZtdRe-e!ddfdZuejde-e%fdZw ddtejde e-e!eLe%efgedfde&de?ejedffdZxdRe-e!de-e%d~eLe%efde?e-e!ejffdZydRee!deddfdZzddtejde&de-e!fdZ{dRee!de-e!fdZ|ej ddtejde&dee!fdZ}ejZdde%de&de%fdZdtejde&fdZ dde defde eefdzde&de&de-e!f dZy)a This module provides utilities for analyzing, transforming and manipulating Python bytecode. It includes functionality for: - Converting between different bytecode formats and versions - Virtualizing jumps and managing jump targets - Handling exception tables and their entries - Managing instruction offsets and extended arguments - Providing a clean API for bytecode modification and transformation - Supporting Python version-specific bytecode features - Generating bytecode from template functions The module is designed to work across different Python versions (3.7+) and handles version-specific bytecode differences transparently. N)CallableIterableIteratorMappingSequence)AnycastOptional TYPE_CHECKING)config) get_indexofpropagate_line_numsremove_extra_line_numsstacksize_analysis)is_safe_constant)DynamoTracerOutputT)slotsc\eZdZUded<ded<ded<eed<eed<defdZd edefd Z y ) InstructionExnTabEntry Instructionstartendtargetdepthlastireturnc d|jjd|jjd|jjd|jd|j d S)NzInstructionExnTabEntry(start=z, end=z , target=z, depth=z, lasti=))rshort_inst_reprrrrrselfs e/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/_dynamo/bytecode_transformation.py__repr__zInstructionExnTabEntry.__repr__1sl+DJJ,F,F,H+IJ88++-./kk11345ZZL A 7 oc(t|tsy|j|juxrh|j|juxrN|j|juxr4|j |j k(xr|j |j k(SNF) isinstancerrrrrr)r"r&s r#__eq__zInstructionExnTabEntry.__eq__9s}!34 JJ!'' ! &AEE! & qxx' & agg% & agg%  r%N) __name__ __module__ __qualname____annotations__intboolstrr$objectr*r%r#rr)s=    J K #     4  r%rceZdZUdZeed<eed<edzed<eed<dZedzed<dZ edzed<d Z e ed <dZ e d ed <dZe ded <dZedzed<dZedzed<defdZdede fdZdefdZddZy)rz$A mutable version of dis.InstructionopcodeopnameNargargvaloffset starts_lineFis_jump_targetz dis.Positions positionsr exn_tab_entryargreprrct|SNidr!s r#__hash__zInstruction.__hash__Ws $xr%otherc0t|t|k(Sr@rAr"rDs r#r*zInstruction.__eq__Zs$x2e9$$r%c<d|jd|jdS)NzInstruction(opname=z , offset=r)r6r9r!s r#r zInstruction.short_inst_repr]s$T[[M4;;-qIIr%cH|j|_|j|_yr@)r:r<rFs r#copy_positionszInstruction.copy_positions`s ,,r%)rDrrN)r+r,r-__doc__r/r.r1rr9r:r;r0r<r rr=rr>rCr2r*r rIr3r%r#rrEs. K K tO KFC$J"Kt" ND +/Ix(/&*FH] #*37M)D07GS4Z#%F%t%JJ)r%r irc t|j|j|j|j|j |j |j|jSr@) rr5r6r7r8r9 line_numberr;r<rNs r#convert_instructionrRgI HH HH EE HH HH MM   KK  r%rL c t|j|j|j|j|j |j |j|jSr@) rr5r6r7r8r9r:r;r<rQs r#rRrRurSr%c t|j|j|j|j|j |j |jdSr@)rr5r6r7r8r9r:r;rQs r#rRrRsE HH HH EE HH HH MM     r%ceZdZdefdZy) _NotProvidedrcy)NrYr3r!s r#r$z_NotProvided.__repr__sr%N)r+r,r-r1r$r3r%r#rYrYs#r%rYrL namec |dvS)N) LOAD_ATTR LOAD_GLOBALLOAD_SUPER_ATTRr3r]s r#inst_has_op_bitsrcsFFFr%c |dk(S)Nr`r3rbs r#rcrcs }$$r%cyr(r3rbs r#rcrcsr%)r7r8rr7r8rct|r| td|%d}n"|du|tuz|duz}|dkDr td|t|ts tdt t j|||||S)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)r5r6r7r8r)rc RuntimeErrorrYr)r/rdisopmap)r]r7r8rcnts r#create_instructionrks0  KL L ;C$6#=>&PTBTU 7R  z#s3@AA yytVF r%cJtjdk\rdnd}t||S)NrT JUMP_FORWARD JUMP_ABSOLUTErsys version_infork)rinsts r#create_jump_absoluterts# --8>oD d6 22r%c|jdvS)N)rmrn)r6ros r#is_jump_absolutervs === ==r%valcheckedcJ|rt|s Jd|td|S)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_CONSTr8)rrk)rwrxs r#create_load_constr|s/ $>(8&>>$ l3 77r%c>ddlm}td|tdgS)a Create instructions that trigger the bytecode debugger to stop. Usage: codegen.extend_output(create_breakpoint()) When the bytecode debugger is active, execution will pause at this point. At runtime, these instructions have no effect on program state. r BREAKPOINT_MARKERrzr{POP_TOP)bytecode_debuggerrrkr~s r#create_breakpointrs&5 <0AB9% r%cXtjdk\r tddStdS)NrTCOPYr r7DUP_TOPrpr3r%r#create_dup_toprs( 7"!&a00 i ((r%nc|dkrgStjdk\r&t|ddDcgc]}td|c}S|dkrtdgd|d z zgStd |gScc}w) 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 rTSWAPrROT_)TWOTHREEFOURROT_N)rqrrrangerk)rrNs r# create_rot_nrs| Av  7".inst_has_bit_set*s/Sz~~)))Sz~~!Q&&r%NcX|jJ|xjdzc_yrrrs r# set_inst_bitz#add_push_null..set_inst_bit.s)Sz~~))) c !r%rKr PUSH_NULLr[rr`rT) r)rlistr/r0rqrrrcr6rk)rrrrs @r# add_push_nullrsu"--$1?-...'c'd'#$ 7" E"I,, -6Fr6J  & L#/ <==E" L!  W $ E"I,, -6Fr6J   L1X__ -6Fq6I O L( 45=E L  W $ E!HOO ,5Ea5H O L( 45=E Lr%ct|tr|g}nt|tsJ|}tjdkr|Stjdk\rdnd}||j dk(r@||j J||j dzdk(r||xj dzc_|Stjdk\r|tdgz}|Stdg|z}|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. rTrKrrr`r r)r)rrrqrrr6r7rk)rrrs r#add_push_null_call_function_exrNs --$1?-... '!   G+"C SzM)Sz~~))) :>>A  " #JNNa NL 7"+K899 L$K01E9 Lr%nargs push_nullctjdk\rg}|rQ|jtdtjdk\r|dzn|dz}|j t |tjdkr|jtd||jtd ||Std |gS) 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), ) rTrrKr rr[PRECALLrCALL CALL_FUNCTION)rqrrappendrkextendr)rroutputrotss r#create_call_functionrlsR 7"  MM,[9 : # 0 0G ;519D MM,t, -   g % MM,YEB C (U;< E : ;;r% has_kwargsignore_314_kwargs_pushctjdk\rg}tjdk\r |s|s|jtdd}|rc|jtdtjdk\rt |dzn t |dz}|j t ||jtdt | |Stdt | gS) a( Assumes that in 3.14+, if has_kwargs=False, there is NOT a NULL on the TOS for the kwargs. This utility function will add a PUSH_NULL. If the caller has already pushed a NULL for the kwargs, then set ignore_314_kwargs_push=True so we don't push another NULL for the kwargs. rTrLrTrKrrLCALL_FUNCTION_EXr)rqrrrrkr/rr)rrrrrs r#create_call_function_exrs 7"    '* MM,[9 :J  MM,[9 : ##w.J!#_q(  MM,t, - ();ZQR 1s: G HHr%ctjdk\rtd|gStjdk\rtd|td|gStd|gS)Nr[rrrTr CALL_METHODrp)rs r#create_call_methodrs^ 7""6u566 7" ye 4 v5 1   }% 8 99r%c^tjdk\rtdd|Std|S)Nr[r_r )r7r8 LOAD_METHODr{rprbs r#create_load_methodrs, 7"!+1TBB mD 99r%cJtjdk\rdnd}t||S)NrT BEFORE_WITH SETUP_WITHrorp)rr6s r#create_setup_withrs#!..'9]|F fV 44r%c tjdk\rtd|gS|dk(rgS|dk(r tdgS|dk(rtdtdgS td |dz td td d ttdtd tdtd d tdtd t dg t dtdtd|dz S)NrTrrr rROT_TWOrL ROT_THREE BUILD_LISTrrzrr{ STORE_SUBSCRreverserrUNPACK_SEQUENCE)rqrrrkcreate_binary_subscrrrrs r# create_swaprs 7""6q122Av a"9-.. a";/1CI1NOO& *9%9%  A   9%  ,!a%8 r%rrstorecZtjdk\r.|r tdn t}t t |||gStjdk\r&|rdnd}t |t |t|gS|rdnd}t |t |tdd t|gS) zQ BINARY_SLICE and STORE_SLICE (if `set` is True) for all Python versions rrr[ STORE_SLICE BINARY_SLICE BINARY_SUBSCR BUILD_SLICErr)rqrrrkrr|slice)rrr subscr_inst inst_names r#create_binary_slicer s  7"27 ~ .=Q=S  eE3/ 0     W $%*M e $ c " y )  ',N e $ c " }! 4 y )   r%ctjdk\rtd|gS|dk(r tdgSgt|t t|dztdS)NrTrrr rr)rqrrrkrrrQs r# create_copyr)sw 7""6q122Av"9-..  Q   QU   Q  r%rcgttdtt|tj dk\rdndzt ddtdS)Nrzr{rTrr Fr)rrkprintrrqrrr)rs r#create_print_on_stackr=sb  ),uE F  U3#3#3w#>aAF G  a '  9%  r%valuecgttdttd|tddtdS)Nrzr{r Fr)rrkrr)rs r#create_print_valuerGsP  ),uE F <6  a '  9%  r%cXtjdkr tdStddS)Nrr BINARY_OPrrpr3r%r#rrPs( '!!/22 kr 22r%cbtjdk\r|dk(r tdStd|S)Nrrr3 BUILD_TUPLEr)rqrrr|rkrs r#create_build_tuplerWs. 7"qAv $$ m 33r% first_linenoctjdddk(sJg|dddtdtddffd d td tddffd }d tddffd }||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 Nr)rL r byteno_delta lineno_deltarc|dk7s|dk7retdt|d}tdt|d}|dk7s|dk7sJ||z}||z}j||dzf|dk7r^|dk7rdyy)Nri)maxminr)rr byte_offset line_offset linetables r#_updatez!linetable_writer.._updateksa<1#4a\3!78KdC c$:;K!#{a'7 77 K 'L K 'L   k;+=> ? a<1#4r% lineno_new byteno_newc4|z }|||z |yr@r3)rrrrbytenolinenors r#updatez linetable_writer..updatets+!F*  l+!F* r% total_bytesc|z yr@r3)rrrrs r#rzlinetable_writer..end|s f$l3r%)rqrrr/)rrrrrrrrs @@@@@r#linetable_writerr]s   BQ 7 ** *I FL F@c@@@3CD444 fc !!r%c|dk\sJ|dzg}|dz}|dkDr,|dxxdzcc<|j|dz|dz}|dkDr,|S)z~ 6-bit chunk encoding of an unsigned integer See https://github.com/python/cpython/blob/3.11/Objects/locations.md r?r@)r)rbs r# encode_varintrs` 6M6 RA!GA a% "  R a a% Hr%ctjdk\sJg|dtjdzdtddffd }|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 rTr<N inst_sizercr jnd}dtdtddffd }|d}n|z }||dkDr||d|dz}|dkDr|||y)Ndeltasizercd|cxkrdksJJd}r~jrjfjZjNj d|zdz jjz jdzjdzf}nj d|zdz |dkr | dzdz}n|dz}j t ||D]}j t |y)Nrr3r )r end_lineno col_offsetend_col_offsetrrr)rr other_varintsrrr<s r#rz5linetable_311_writer..update.._updates4}1}$}$}24 !((4!,,8!,,8!00<$$\D%81%<=",,y/?/??!,,q0!0014%M $$\D%81%<=19$f]a/EaKE  u!56&7A$$]1%567r%rr)rr/)r<rrrrrrs` r#rz$linetable_311_writer..updatesx-6))DJ 7s 7# 7$ 7@! )F2 #a- a(Q a- L) ,r%)rqrrrh Positionsr/)rrrrs @@r#linetable_311_writerrsT7*** , -cmmd2, -s, -t, -\&  r%c@eZdZUeed<eed<eed<eed<eed<y)ExceptionTableEntryrrrrrN)r+r,r-r/r.r0r3r%r#rrs J H K J Kr%rc|dk\sJ|dzg}|dz}|dkDr|j|dz|dz}|dkDr|jtt|dz D]}||xxdzcc<|S)zR Similar to `encode_varint`, but the 6-bit chunks are ordered in reverse. rrrr r)rrrlen)rrrNs r#encode_exception_table_varintrs 6M6 RA!GA a% R a a%IIK 3q6A:  !  Hr% bytes_iterclt|}|dz}|dzr|dz}t|}||dzz}|dzr|S)z5 Inverse of `encode_exception_table_varint`. rrr)next)rrrws r#decode_exception_table_varintrsN ZA b&C b&     q2v  b& Jr%tabc tt|dz D]j}||j||jkrH||j||dzjkr&||dzj||dzjkrjJy)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)rrrr)rrNs r#check_exception_tablers 3s8a<   FLLCFJJ &A SQZ---AE   CAJNN2  3 r%exntabc 6t|}g} t|dz}t|dz}||zdz }t|dz}t|}|dz }t|dz} |jt ||||| o#t $rt ||cYSwxYw)z Parse the exception table according to https://github.com/python/cpython/blob/3.11/Objects/exception_handling_notes.txt rr )iterrr0rr StopIterationr) r exntab_iterrrlengthrrdlrrs r#parse_exception_tabler"s v,K C 1+>BE2;?!CF&.1$C2;?!CF.{;B!GEaLE JJ*5#vueL M c" sA0A??BBcg}|D]}t|jdz}|dxxdzcc<|j||j|jz dz}|jt|dz|jt|jdz|j dz|j z}|jt|t|S)zd Inverse of parse_exception_table - encodes list of exception table entries into bytes. rrr )rrrrrrrbytes)rrentry first_entryr r!s r#assemble_exception_tabler(s A43EKK14DE A&  U[[(1, .v{;< .u||q/@ABkkQ%++ - .r234 8Or% instructions firstlinenocFg}tjdk\rt|\}}d}t|D]\}}|jdk(r=d}|dz }dD]0} ||| zjdk7s||| zj |_nnt |dz|z}d}||j |d}|jxsd} |j|j| dzftt |dzdz D]} |jdnt|\}}} |D]Z}|j||jt||jxsd} |j|j| dzf\| t|t|t|fS) z)Do the opposite of dis.get_instructions()rTr EXTENDED_ARGr r rrLrr)rr)rqrrr enumerater6r<instruction_sizer7rr5rrr:rr%) r)r*codelnotab update_linenonum_extrNrsrjr7_rs r#assembler6+sD 7" 4[ A  . $GAt{{n, 1 "A#AE*11^C)5a!e)<)F)F -T2a7'A  $..) 4G((-aC KKcDj1 2+D1Q6:; $ F# $! $&&6k%B" s  3D+d..D :((-aC KKcDj1 2  3 CI ;f %%r%offset_to_instr9cldD]/}|||zjtjk7s'|||zcSy)zU Get the instruction located at a given offset, accounting for EXTENDED_ARGs )rrrrNr5rhr,)r7r9rs r#_get_instruction_by_offsetr:QsD . &1* % , ,0@0@ @!&1*- -. r%c|Dcic]}|j|j|}}|D]V}|jtjvs|jtjvs>#BRBRR!#,/F  M  Mr%c ttjjttj}|D]}|j |vs|j tjvs/|j "|j j |jJ|j j|jkr9tjdkr tdd|jvst|tjdk\sd|jvst|t|t|}|D]/}|j |vs|j Jt|||j }|j tjvrJtjdkr,|jJt!|jdz |_nvtd|j |jJt%t!|j|jz t'|z |_|xj"dzc_|j|_d|j|_2y) zJFill in args for virtualized jump target after instructions may have movedNrTz*Got negative jump offset for Python < 3.11rBrCrz+Python 3.11+ should not have absolute jumpszto )setrhr<unionr=r5rr9rqrrrgr6rGupdate_offsetsrrIr/r7absr/r8r>)r)jumpsrsindexofrs r#devirtualize_jumpsrQs   " "3s{{#3 4E2 ;;% {{#++-KK+ **6 /0;;%% 3'''1*+WXX DKK/+D1''72zT[[7P+D1#2(< ,'G1 ;;% ;;* **+L'$++:NOF{{ckk)##g-"==444"6==1#45DH&'TUU}}0T[[5LLL  36Ft6LLMQ --DK 0DL+1r% exn_tab_bytescB t|}|Dcic]}tt|j|c} t j  dt | dtttff fd }|\}}|D]}|jJ|j|jkDr$|\}}|j|jkDr$|j|jk\shtj||_ ycc}w#t$rYywxYw)zDReplace exception table entries with pointers to make editing easierrrcvt}tkr8|jkr&dz tkr|jkr&dkDsJdz }tt |j t |t |j |j|j}||fS)Nr r) rrrrr:rrrr)r& end_offset inst_entryend_offset_idx exn_tab_iterr7offsetss r#stepz(virtualize_exception_table..steps&E W-'.2IUYY2V!#W-'.2IUYY2V"A% %% !!34J/*>5;;G*>:F*>5<<H  J*$ $r%N)r"r r/r9sortedkeysrtuplerrrrcopyr=r) rRr)exn_tabrsrZr&rVrWrXr7rYs @@@@r#virtualize_exception_tabler`s $M2G?KLtd3 ,d2LN^((*+GN=L   %e/1GGH % %,!Fz  ;D;;* **++ )$(F!z++ ){{ekk)%)YYz%:"  ;;MF    s$"D  A5DD0D DDc 0 i t|}|D] }|jst|||jjj}|Jt t |jjjt|jjzdz }|Jt|||jjj}|J||f}||jj|jjf}| vr ||k(sJ| |<t jd}dg g d fd } |D]} r& dd|dkr|  r dd|dkr ro dd|dcxkr|dcxkr ddksJJt dd} | |dkr( jt!| |ddz g d|d j| r |  r t#  S) zMCompute exception table in list format from instructions with exn_tab_entriesrc|d|d fSNrr r3ts r#z)compute_exception_table.. s1!u r%keyrcrQj}|dkr8jtt|d|dg||ddzyyy)zT Pop the key_stack and append an exception table entry if possible. r rrN)poprrr)rhexn_dictr_ key_stacknextis r#rjz$compute_exception_table..popse --/CA'CFE(:CFSXc]SA   r%rr )rN)rr=rIrr9r r/rr/rrrr[r\rrrr)r)rPrsrrrrhrw keys_sortedrjleftrkr_rlrms @@@@r#compute_exception_tablerpsh>@H,'G    *gd&8&8&>&>?f $ $$S$,,00778"4#5#5#9#9:;  ? "?+gd&8&8&?&?@f % %%#,C4--33T5G5G5M5MNCh}+++HSM- <.EFK E')I)+G # # IbM!,s1v5 EIbM!,s1v5 R=#s1vKQK9R=;KK KK KKuimA./Dc!f}'c!fqjS8IbM;RSFE    '" Nr%rPc*g}|D]}||j||jf}|r/|dd|dkr!|j|r|dd|dkr!|r(|dd|dcxkr|dcxkr |ddksJJ|j|y)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. rr rN)rrrjr)rrP entry_stackr&rhs r#!check_inst_exn_tab_entries_nestedrs4s*,K u{{#WUYY%78k"oa03q69 OO k"oa03q69 r?1%QO3q6O[_Q=OO OO OO3  r%ct|}i}|D]h}|js||jj||jjf}||vr|j||k(sJ|j||<jt |j dDcgc]}|| }}t |||D]M}t||j||jdzD]}tj|||_!Oycc}w)z~ Copies exception table entries to all instructions in an entry's range. Supports nested exception table entries. c|d|d fSrcr3rds r#rfz2propagate_inst_exn_table_entries..XsadQqTE]r%rgr N) rr=rrr[r\rsrr^)r)rPentriesrsrhsorted_entriesr&rNs r# propagate_inst_exn_table_entriesrxGs ,'G=?G.   **001**../Cg~))WS\999--GCL.!'w||~;R S N&ng> =wu{{+WUYY-?!-CD =A,0IIe,N>N2NqOLOOs"33old_inst new_instsc|jr:|jj|ur"t|dkDr|d|j_|ddD]7}tj|j|_|j|_9|dj |_|dj |_|dj|_|dj|_|dj|_ |g|ddzS)Nr rr) r=rrr^r<r5r6r7r8r)rrrss r#overwrite_instructionrs   " " & &( 2  NQ %.r]"!" ,!YYx'='=>!++, l))HOl))HOQ<##HLl))HOl))HO : !" %%r%ctjdksJddd}|D]G}|j|vs||j|_tj|j|_I|S)zFLOAD_METHOD puts a NULL on the stack which causes issues, so remove itrTr_r)rr)rqrrr6rhrir5)r)rewritesrss r#remove_load_call_methodrsg   g %% %*?KH1 ;;( ""4;;/DK))DKK0DK1 r%cg}|D]}d|jvrtdtd|jv}|j|_t j dkr(td|jvrdnd|j }ntd |j }td d ||g}|jt|||j|||ddy) N_NONEIS_OPNOTrr[rBPOP_JUMP_FORWARD_IF_TRUEPOP_JUMP_BACKWARD_IF_TRUEroPOP_JUMP_IF_TRUErzr{) r6rkr/r7r8rqrrrrrr)r)rrsis_opjump_op replace_instss r#remove_jump_if_nonersI# dkk !&wC8L4MNE 99EL'),% 338;; --? T#<=M   24G H   T "3#4 LOr%cXg}|D]}|j||jdvs$tjdk\r|jdk(r t }n%t |jj dd}|jr)|jj|ur||j_tj|j|_|j|_ tjd|_ d|_d|_d|_|j| ||ddy)N)rrrrSLICESUBSCRrr)rr6rqrrrrkrDr=rr^r<rhrir5r7r8)r)rrsrs r#remove_binary_store_slicersI* ;;9 97*t{{n/L24 01D1DWh1WX !!d&8&8&<&<&D)4""&(, $2D2D(EK %$(NNK !))M2DK'DKDHDK   [ )#*$ LOr%) LOAD_FASTr)LOAD_FAST_BORROWr) STORE_FASTr)rr)LOAD_FAST_LOAD_FAST!LOAD_FAST_BORROW_LOAD_FAST_BORROWSTORE_FAST_STORE_FASTSTORE_FAST_LOAD_FASTcg}|D]}|jtvr[t|j\}}|j\}}t||t||g}|j t ||p|j |||ddy)Nr{)r6 FUSED_INSTSr8rkrrr)r)rrsinst0inst1argval0argval1rs r#remove_fused_load_storersI # ;;+ %&t{{3LE5#{{ GW#59"59M   24G H   T " # LOr%cg}|D]k}d|jvrJtddt|j|jg}|jt ||[|j |m||ddy)NRETURNNOPGRAPH_BREAK_IF_LEAFr{)r6rkr8rrr)r)rrsrs r#$add_graph_break_if_leaf_instructionsrswI# t{{ ""51FG"4;;t{{CM   24G H   T "# LOr%cg}tj|D]$\}}|jdk(r|jdk(r|jj drt t |D]}||j|ur |||_||js.||jj|ur|||j_ ||jj|ur|||j_ ||jj|us|||j_|j|'|j|d||ddy)Nrrrr) itertoolspairwiser6r8 startswithrrrr=rrr)r)rrs next_instrNs r#'remove_graph_break_if_leaf_instructionsrsHI$--l;#i KK5  44  ++H53|,- I?))T1-6LO*?00$A44::dB>G Q55;#A4488D@.maybe_pop_nIs9q A&*++s/?/??  r%rirLr,rrirrr )r/r5rhr,r7rrkr)r)rrsaddedrs @r#fix_extended_argsrEsr "Fst  ;;#** *DH XX$((X- N MM,^RP Q MM,^RP Q MM,^QO P XX$((V+ N MM,^RP Q MM,^QO P XX$((T/ N MM,^QO P d!$ K#l+ +E A::LO Lr%rscddl}tjdk\r=d|jjj j |jdzzSy)NrrTrr )torchrqrr_C_dynamo eval_framepy_opcode_cachesr5)rsrs r#r/r/fsF 7"EHH$$//@@MPQQRR r%cTd}|D]!}|j|k(sJ|t|z }#yNrr9r/r)r9rss r# check_offsetsrns8 F){{f$$$"4(()r%c@d}|D]}||_|t|z }yrrrs r#rMrMus- F) "4(()r%argsc Ntttt|}|gt |zt |d|dDcgc]\}}t ||k7c}}gzDcgc]}djd|D}}ddj|zScc}}wcc}w)Nr c3$K|]}|d yw)03Nr3).0xs r# zdebug_bytes..s(qAb6(szbytes mismatch  )rrmaprrzipr/join)rindexarr7results r# debug_bytesr}s #c#tn% &E7 t*$'R$r($;   (C((F  & 1 11 = s B'B"ct|dd\}}|j|jk(s%Jt|j|jy)z.sr%T)safeN)transform_code_objectco_coder)r0doder5s r# debug_checksrs?#D*;$GGD! <<4<< 'PT\\4<<)PP 'r% code_optionsc|t|dD] \}}||us |cS|dxx|fz cc<t|ddz S)N co_constsr )r.r)rrwrNvs r#get_const_indexrsV,{341 !8H #' |K( )A --r%varname_from_oparg.c@  tdDcic]\}}|| c}} dtdtf fd }tjdkrO|JtdDcic]\}}|| c}} tddzDcic]\}}|| c}} njt |sJi}t jD]} ||}|||<dDcic]}||| c} ddzDcic]}||| c} ttD] dtf fd } jd k(r jtusJtjdk\rS jJ| jd zt!t jd zz _| j _ jd k(r jtusJtjdk\rT jJ| jd zt!t jd zz _N| j _m jdk(rn jJ jtusJ| jd zt!t jd zzd z _ jt"vrtjdk\sJt% jt&sJt jd k(sJt' fd jD}|ddz|d dzz _ j(t*vre|stjdk\r- j vr j _ j _ j(t,vr(|s(| j _G j(t.vr%|se j _ j(t0vs jt3 j}|dk\sJ| _ycc}}wcc}}wcc}}w#t$rY>wxYwcc}wcc}w)Nco_namesr]rc |}|S#t$r=tx}|<gd|d<tdtk(sJY|SwxYw)Nr)KeyErrorr)r]rrnamess r#get_name_indexz fix_vars..get_name_indexsu ?+C   ? #E *C%+'Hj)A'H4'HL $|J/0CJ> >>  ?s AAArTrrrc.jtuSr@)r8rY)rNr)sr#should_compute_argz$fix_vars..should_compute_args?))= =r%r`r rr_r[rarKc3:K|]}|vr|n|ywr@r3)rr] freenamesvarnamess r#rzfix_vars..s-#'("2 $Gsrr)r.r1r/rqrrcallablercount IndexErrorrrr0r6r8rYr7r rr)r]r5 HAS_LOCALHAS_NAMEHAS_FREE HAS_CONSTr) r)rrrr]rallnamesr arg_tuplerrNrrs `` @@@@r#fix_varsrsM )2,z2J(K L93T3Y LESS '!!)))/8m9T/UV)#tD#IV']+l=.II T #I *+++??$ C )#.!$  6B-5PQTD(4.(Q%]3l=6QQ  (4.  3|$ %C* >D > ? ! !] 2?))= ==7*#A**666'5l1o6L6L'MQR'Rl1o112Q6' Q#'5\!_5K5K&L Q# !_ # #{ 2?))= ==7*#A**666'5l1o6L6L'MQR'Rl1o112Q6' Q#'5\!_5K5K&L Q# !_ # #'8 8?&&2 22?))= == Q 6 671<\!_001A57 O  !_ # #{ 2##w. ..l1o44e< <<|A--.!3 33(O22I$-Q<1#419J"KLO  !_ # #y 0!#$$/$Q..h>+4LO4J4J*KLO'*2<?3I3I*JLO' !_ # #x /!#&4\!_5K5K&L Q# !_ # #x /!#&/ Q0F0F&G Q# !_ # #y 0A""*%lLO4J4JKaxx&) Q#GC*G MW   Q s/ S3$ S9 S? T T8 T TTc|D]t}|jtus|jtvs7|jtvs%|jt vs|jt vs_|jdvsnd|_vy)N)r`r_ra) r8rYr5rrrrr6r7)r)rss r#clear_instruction_argsrsc  KK| + y(;;(*;;(*;;)+ #RRDH r%cNdg}|jd|jgdtjdk\r|jd|jd|jdtjdk\r|jd|jd d g|S) N co_argcountco_posonlyargcount) co_kwonlyargcount co_nlocals co_stacksizeco_flagsrrrr co_filenameco_namerT co_qualnameco_firstlineno co_linetableco_exceptiontablerr)rrrqrr)r\s r# get_code_keysr&s ?DKK$%KK   7" M"KK !KK 7" '(KK    Kr%transformationsrrct}|Dcic]}|t||}}t|d|dk(sJt||}t ||||}t |||\}} | |fScc}w)Nrr)rgetattrrcleaned_instructionsrclean_and_assemble_instructions) r0rrr\krr) tracer_outputr5bytecodes r#rrKs ?D156AAwtQ''6L6 |M* +|L/I II I'd3L %#L,?M1,lSKAx ] ""7sA/r\ct|t|t|d|d<d}tjdk\r/t j |Dcgc]}|| c}}|j}t|||d}|r-t|t|tt|}|r-t|t||d\}}||d<||d<t||d <t!|d hz t!|j#d hz k(sJtjdk\rt%t'||d <|t j |Dcgc]}|| c}fScc}wcc}w) NrrrT)rTr rrrrr)rr|rrqrrtypesCodeType_varname_from_opargrrMrQr0rrr6rrKr\r(rp) r)r\rrrtmp_codedirtyrr1s r#rr_sy,L9$\2!$\-%@!AL 7">>T#BLO#BC%99 \<dictrrrrrr)r)rNcopiedremapr&s r#r%r%s "   HH HH EE HH HH MM   KK HH OO II F" \6* +EE$K  ?OO #5 #4ekk"eii ell#  AO  M?sB D&cttttj|}t |t |tjdk\r,t||jt|j|t|t||stjdkr t|tjdkr t!||tjdk\rGt#|tjdk\r t%|tjdk\r t'|t(j*r t-|tjdk\rt/|t1||S)NrTr[rK)rrrRrhget_instructionsrrrqrrr"rr`rr?r~rrrrrr &debug_force_graph_break_on_leaf_returnrrMrQr&s r#r$r$s/1E1Ed1KLML %, 7" t~~>"4#9#9<H\" %    g % #L 1   g % 4 .   w &  -7*),77*' 5  8 8 0 > 7"|$<( r% with_uuidc|dtt}|r*|dtjj ddz }|S)Nr5-)r_unique_id_counteruuiduuid4rD)r]r.rets r# unique_idr5sE F!D+,- .C 4::<.!))#s33 Jr%c*d}|j|zdkDS)N r)r )r0 co_generators r# is_generatorr9sL MML (A --r%fn varname_mapnoreturnnoprefixct|j}t||r,t|D]\}}|jdk(s||dzd}n|D]6}d|_d|_|s|j|vs#||j|_8|rtjdk\rvg}|D]m}|jdk(rKtjd|_ d|_|j||jtd]|j|o|}g}|D]#}|jdk(s|j|%t|dk(r|d|d ur|j!d |St|dkDr|d |d urOd |d _tjd |d _ d|d _t$|d _|j!d n|jtd |D]e}t'|d } | j|_| j|_ | j"|_| j|_| j(|_g|S) aGenerates bytecode from a template function `fn` for use in dynamo bytecode generation. For example, we can generate Python-version-independent bytecode for looping through a dictionary and copying the values to a new dictionary. def template(d1, d2): for k, v in d1.items(): d2[k] = v or a try block: def template(): try: dummy1 except: dummy2 raise dummy3 Args: fn: a function template to generate bytecode from varname_map: a mapping of `fn`'s varnames to new names. This map will be applied to the generated bytecode's varnames. For example, local variables in `fn` can be replaced with new names that are generated by `OutputGraph.new_var`. noreturn: remove all RETURN_* bytecodes and replace them with a jump to the end of the bytecode. NOTE: any items pushed to the stack for return WILL remain on the stack! Append a POP_TOP if you don't want that item to be present. noprefix: remove prefix bytecodes (all bytecode before the first RESUME, inclusive). RESUMEr Nr[ RETURN_CONSTrz RETURN_VALUErrr)r__code__rr.r6r:r<r8rqrrrhrir5rrkrrjr7rYrtr) r:r;r<r=rrNrsrreturns jump_insts r#bytecode_from_templaterEsNN ! -E5! ' GAt{{h&a!eg  3  4;;+5%dkk2DK 3   w &I +;;.0"%))L"9DK".DK$$T*$$%7%GH$$T* +E %D{{n,t$ % w<1 uRy!8 IIbM0 L/\A RyGBK'#(b  #&99U#3b  $b #/b   B /67  /1r; '.. '.. $=='.. '..  / Lr%)T)Fr@)NTT)rJr^ dataclassesrh functoolsrrqrr2collections.abcrrrrrtypingrr r r r bytecode_analysisrrrrutilsr output_graphr dataclassrrrrrRrYr1r0rcr/rkrtrvr|rrrrrrrrrrrrrrrrrrr]rrr rrrrrr%r"r(r6r(r:r?rKr=rFrGrIrQr`rprsrxr|r~rrrrrrrrrrrr/rrMrrhaslocalrhasnamerhasfreerhasconstrrrr lru_cacherrrr"rr%r$rr1r5r9rEr3r%r#rTs     KK55 $0T"  # 6T"))#)>w  s  ;      s  ;    s  ;   wGsGtG  %s%t% st%!% ' ' t' $J ' $  '  'V333 >[>T>838884 ,$) ) 0C0D-0.8k!228 +8vk!22 +<4<4<4<k9J434 ,(k):cd;&73k34#4+4 """" 49hSz4/0(C5$;2G GH""J  S  T#Y  w:!:! tCy(CMMD$8##>#DEE F:!zT"#  S  T#Y  hsm    t$78  T  %D1D,E,$':";$#&4 ,#&3#&5PUCV#&L k)* 47 4 P8K#8PTP   ,[ ,T , k):   21T+%621421j) ) (,[(9)  ) XJ{#J JZ $ % 04[#5E0F   &=4 3D==6C4 3DCC0Pd;&7PDP&&&*;&7& +&.${*;[@Q d;&7 D > D,= $ 06)Q97   ${*;  &  tK7H  T   ${:K PT 6##tK7H#T#LD$5#B ; 3 ) 5)$))+!6)4) 2u 2 2Qu~~Q$Q   s{{  s{{     .$sCx. .s .s .59l*{#l*sCx.l*!c*T1l* l*^k):t" !tCy!!R # ..# k DcN+X6J-KK#  #  5>>8$89 9: #($J{#$J+/9$JDHcN$J 4 enn ,-$JN+8K+@+#+RV+-u~~-T-d;FW-"h{&;"[@Q"J ', .. $ k>%Y__&CDS.u~~.$.-1 gcgc"T)gg g  + gr%