BjI<dZddlmZddlZddlmZmZmZmZddl m Z ddl Z ddl m Z ddlmZdd lmZmZdd lmZmZmZmZd d lmZmZerdd lmZddlmZd dlm Z GddeZ!e!dZ"e!dZ#e!dZ$GddeZ%dS)z Constant variable tracking in Dynamo. This module is fundamental to Dynamo's ability to track and propagate constant values during compilation, ensuring proper handling of Python literals and maintaining type safety through the compilation process. ) annotationsN)AnyLiteraloverload TYPE_CHECKING)override) GetItemSource) variables)raise_observed_exception unimplemented)common_constant_typesistypenpraise_args_mismatch)ValueMutationNewVariableTracker)Sequence)InstructionTranslator)UserFunctionVariablec  eZdZdZeeeeee de e dfZ e e dAdZe e dBd Ze e dCd Ze dCdZdDfd ZdEdZdFdZdEdZdGdZdHdZdIdZdHdZedJdZdKdZe dLd"Ze dMdNd%ZdOd'ZdPd(ZdPd)ZdPd*ZdQd,Z dRfd/ Z!dSfd6 Z"e#dTd7Z$dGd8Z%dUd:Z&dVd<Z'dWd=Z(dXfd> Z)dXd?Z*dXd@Z+xZ,S)YConstantVariablea Variable tracker for Python literals and basic immutable types, with automatic routing support for collection types (lists, tuples, sets, etc.). The create() method intelligently constructs appropriate variable types for nested collections. N.valueNonereturncdSNrs c/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/torch/_dynamo/variables/constant.pycreatezConstantVariable.create214boolcdSrrr s r!r"zConstantVariable.create6r#r$rkwargsrc dSrr)rr's r!r"zConstantVariable.create:s>Acr$c P|s!|xtSxdurtSdurtS|d}t |t rd|D}t j|fi|St |trd|D}t j |fi|St |tr?|j |j |j f}td|D}t j|fi|St |t tfrg}t#|D]H\}}|rt%||nd}|t(||It jt1||fi|St)|fi|S) aV Create a `ConstantVariable` based on the given value, and supports automatic routing for collection types like `tuple` (in which case we'd create `ConstantVariable` for the leaf items). NOTE: the caller must install the proper guards if needed; most often the guard will be `CONSTANT_MATCH`. NTFsourcecBg|]}t|Srrr".0xs r! z+ConstantVariable.create..W'???A%,,Q//???r$cBg|]}t|Srr,r-s r!r0z+ConstantVariable.create..Zr1r$c3JK|]}t|VdSrr,)r.args r! z*ConstantVariable.create..^s1#W#WS$4$;$;C$@$@#W#W#W#W#W#Wr$)r*)CONSTANT_VARIABLE_NONECONSTANT_VARIABLE_TRUECONSTANT_VARIABLE_FALSEget isinstancesetr SetVariable frozensetFrozensetVariableslicestartstopsteptuple SliceVariablelist enumerater appendrr"BaseListVariablecls_fortype) rr'r*items slice_argsslice_args_varsir/ item_sources r!r"zConstantVariable.create>s 3TT11TTTT11UU22H%% eS ! ! T?????E(99&99 9 y ) ) T?????E.u???? ? u % % T+uz5:>J##W#WJ#W#W#WWWO*?EEfEE E e} - - TE!%((  1:@JmFA666d  $++*, -55d5kkBB5SSFSS S00000r$c *tjdi|t|sJdt |dt 5t |t jr||_ dS||_ dS)Nz7 Cannot construct `ConstantVariable` for value of type a`. This failure likely due to PyTorch-internal use of `ConstantVariable` on non-literal python values, please try using `VariableTracker.build` instead. If you believe it's a necessary and legitimate use case (the value is immutable and can't easily be represented with another `VariableTracker` class), please add its type to `common_constant_types`. r) super__init__ris_base_literalrJrr:numberitemrselfrr' __class__s r!rRzConstantVariable.__init__ns""6"""//66  97;E{{999  6 >j ::>DJJJDJJJr$c|jSrr rWs r!as_proxyzConstantVariable.as_proxy~ zr$strcfdt|jjdt|jdS)NzConstantVariable(z: ))rJr__name__reprrZs r!__repr__zConstantVariable.__repr__s2S4 #3#3#<SSTZ@P@PSSSSr$c|jSrr rZs r!as_python_constantz#ConstantVariable.as_python_constantr\r$ Literal[True]cdSNTrrZs r!is_python_constantz#ConstantVariable.is_python_constanttr$cDt|jttfSr)r:rintr%rZs r!is_symnode_likez ConstantVariable.is_symnode_likes$*sDk222r$valuesc|j|vSrr )rWrms r!is_constant_matchz"ConstantVariable.is_constant_matchszV##r$c|jduSrr rZs r!is_constant_nonez!ConstantVariable.is_constant_noneszT!!r$list[VariableTracker]c.|dS)zz Need this when adding a BaseListVariable and a ConstantVariable together. Happens in detectron2. N)tx)unpack_var_sequencerZs r!rKzConstantVariable.itemss ''4'000r$rtrr4cpt|j|Sr)rr"rrd)rWrtr4s r! getitem_constzConstantVariable.getitem_consts2 && Js--// 0   r$objobjectc.t|tvSr)rJr)rxs r!rSz ConstantVariable.is_base_literalsCyy111r$cachedict[int, object] | Nonec(it|vrdSt|tttt t jfvr-|t|<tfd|DSt |S)NTc3NK|]}t|V dSr)r is_literal)r.r/r{s r!r5z.ConstantVariable.is_literal..s4JJ'221e<<JJJJJJr$) idrJrErCr;r=torchSizeallrrS)rxr{s `r!rzConstantVariable.is_literals =E c77e  4 99uc9ejA A A E"S''NJJJJcJJJJJ J//444r$InstructionTranslator | Nonect d|DS#t$r }t|d}~wwxYw)NcBg|]}t|Srr,r-s r!r0z8ConstantVariable.unpack_var_sequence..s'RRR1$++A..RRRr$)rd TypeErrorNotImplementedErrorrWrtes r!ruz$ConstantVariable.unpack_var_sequencesN -RR8O8O8Q8QRRR R - - -%1 , -s 727c tt|jS#t$r<}t t ||t|jYd}~dSd}~wwxYw)z9Generic len for any constant value (sequence or mapping).argsN) rr"lenrrr rJrErrs r!len_implzConstantVariable.len_impls| E#**3tz??;; ; E E E $T!WWbtAF|| D D D D D D D D D D Es+. A41A//A4c,||S)z6Sequence length - delegates to len_impl for constants.rrWrts r! sq_lengthzConstantVariable.sq_length}}R   r$c,||S)z5Mapping length - delegates to len_impl for constants.rrs r! mp_lengthzConstantVariable.mp_lengthrr$namect|j|stt||gt |j|}t |rt |S)Nr)hasattrrr AttributeErrorgetattrcallabler)rWrtrmembers r! const_getattrzConstantVariable.const_getattrsZtz4(( F $^Rtf E E E ET** F   &% % r$rdict[str, VariableTracker]c ddlm|dkrPt|jtr6t jtj||g||S|dkrt|jtr|st|dkr2t||dt|dt|d|d |} d |D}t |j|S#t$r't!||||cYSwxYw|d krLt|jtr2dd lm}|||t+ St-fd |DrNt j |||j||||S d|D}d|D} n4#t$r't!||||cYSwxYwt3|jtry|tjvrkt7|j|} t | |i| S#t8$r(} t;t=| |Yd} ~ nd} ~ wwxYwt3|jt>t@frtC|j|r|s|s t t7|j|S#tDtFf$r;} t;t=| |tI| j%Yd} ~ nd} ~ wwxYwtCtL|r)t|dkr|d'r|d} t7tL|}t3| tPj)tPj*tPj+fr;|j,-d||j| fi} ||| S t ||j| S#t8$r;} t;t=| |tI| j%Yd} ~ nd} ~ wwxYwnt3|jt\r;|dkr5t7|j|} t | |i| St=|jt^urx|t^jvrjt7|j|} t | |i| S#t8$r'} t;t=| |Yd} ~ nd} ~ wwxYw|dkrt|dkr|d'r t ta|j|d1S#t8$r;} t;t=| |tI| j%Yd} ~ nd} ~ wwxYw|dkrt|dkr|d'r|rJ|d1} ||jv}t |S#td$r;} t;t=| |tI| j%Yd} ~ nd} ~ wwxYwt!||||S)NrSymNodeVariableformatjoinz1 args and 0 kwargsz args and z kwargsrc6g|]}|Srrdr-s r!r0z0ConstantVariable.call_method..s$JJJQ1133JJJr$__iter__)ListIteratorVariable) mutation_typec38K|]}t|VdSr)r:)r.r/rs r!r5z/ConstantVariable.call_method..s-<.s$???Q!..00???r$c>i|]\}}||Srr)r.kvs r! z0ConstantVariable.call_method..s*QQQ$!QAq3355QQQr$r call_functiondecode __round__ __contains__)3tensorrrrr]r BuiltinVariablerrrrforce_unpack_var_sequencerr"rrrQ call_methodlistsrruranyr[rKr:__dict__r Exceptionr rJfloatrkr OverflowError ValueErrorrEroperatorrhrSymBoolSymFloatSymIntoutput create_proxybytescomplexroundrdr)rWrtrrr' arg_unpacked arg_constr const_args const_kwargsmethodrexc add_targetopproxysearchresultrrXs @r!rzConstantVariable.call_methods ,+++++ 8  tz3 7 7 ,SZ88FFTMDM6 V^^tz3 7 7^ Ta#)4yy@@CKK@@@   7<> > > > ? dj# & &, 643<+?+?TZ..F 6'..vvz/R\/R/RSSS 6 6 6(a"55555555 6  UCL 1 1& 6gdj$6O6O& 6 F +223L74:t3L3L3N3NOOO%z2,S !#(^^$'' QIINNG..00#(] Xt,, M QI22'dj*-ErE+11"eZHHHQ/66rr$*j7Q7QRRR$QQQ0a"4<<PPPPPPPPPQ  E * * 6tx/?/?TZ..F#**66:+N+N+NOO O $*   ( (TW5E-E-ETZ..F 6'..vvz/R\/R/RSSS 6 6 6(a"55555555 6 ;  3t99>>d1g6P6P6R6R> I'..$*d1g&@&@&B&BCC I I I(a"4<<HHHHHHHHH I ^ # #D Q47;U;U;W;W  :!W//11F I4:-'..v666 I I I(a"4<<HHHHHHHHH Iww""2tT6:::s"=D .EE*H;;.I,+I,-K K?K::K??4M44O1N;;O(R:: S?1S::S?V77 W(W##W(AY$$ Z).1Z$$Z)>"\!! ]&+1]!!]& tree_map_fnrmap_fnrestSequence[VariableTracker]tree_map_kwargsc |j|d}|L t|}n_#t$r||||||cYSwxYwt t |dddd}|d }|r|||g|iS|D]1} | s||||||cS2| St|jtttttt t"jfr|||g|iSt'|||||S)N none_is_leaffn __module__optree)rr9r%rdr_tree_map_fallbackr startswithrrqcloner:rkrrr]rrdtyperQ call_tree_map) rWrtrrrrnone_is_leaf_varrtree_map_moduleotherrXs r!rzConstantVariable.call_tree_map@s : .22>BB + #'(8(K(K(M(M#N#NLL*22#' #*Kt44lB## $3#=#=h#G#GG  $++B  rBBB!E 1133#66'" +   zz||# dj3tWc5%+"V W W ?''TMDM2>> >ww$$         s!A#A)(A)cjt|j|}tj|Sr)rrr rr")rWrtrrs r!call_obj_hasattrz!ConstantVariable.call_obj_hasattrts,T**)00888r$cdSrgrrZs r!is_python_hashablez#ConstantVariable.is_python_hashable{rir$rkc*t|jSrhashrrZs r!get_python_hashz ConstantVariable.get_python_hash~DJr$rcddlm}t||r*||kSt|t o)||kS)Nrr)rrr:rd evaluate_exprr)rWrrs r!is_python_equalz ConstantVariable.is_python_equals++++++ e_ - - F**,,0C0C0E0EE E uo . . H''))U-E-E-G-GG r$c|jSrr rZs r!get_real_python_backed_valuez-ConstantVariable.get_real_python_backed_valuer\r$ct|jttfr1tt j|jSt |Sr) r:rrkr%rr"rindexrQ nb_index_impl)rWrtrXs r!rzConstantVariable.nb_index_implsV dj3+ . . G#**8>$*+E+EFF Fww$$R(((r$cZtt|jSr)rr"rkrrs r! nb_int_implzConstantVariable.nb_int_impls  &&s4:777r$cZtt|jSr)rr"rrrs r! nb_float_implzConstantVariable.nb_float_impls" &&uTZ'8'8999r$)rrrr)rr%rr)rrr'rrr)rrr'rrr)rr)rr])rrerr%)rmrrr%)rrr)rtrr4rrr)rxryrr%r)rxryr{r|rr%)rtrrrr)rtrrr)rtrrr]rr) rtrrr]rrrr'rrr) rtrrrrrrrrrrr)rtrrr]rrrrkrryrr%)rry)rtrrr)-r`r __qualname____doc__rkrr]rr%rJr _cpython_typer staticmethodr"rRr[rbrdrhrlrorqpropertyrKrwrSrrurrrrrrrrrrrrrrr __classcell__rXs@r!rr sY%eT44::wS RM 444\X4 444\X4 AAA\XA-1-1-1\-1^ TTTT3333$$$$""""111X1    222\2 5 5 5 5\ 5----EEEE!!!!!!!!m;m;m;m;m;m;^2 2 2 2 2 2 h999X9         ))))))8888::::::::r$rTFc`eZdZdZeZdfd Zdd Zdd Zdd Z ddZ ddZ ddZ ddZ xZS)FakeIdVariablea6A compile-time-only id value that can be used as a dict key but cannot be reconstructed across graph breaks. When dynamo evaluates ``id(x)`` on a variable tracker that has no corresponding runtime object (e.g. a ``ConstDictVariable`` created during tracing), we mint a fake integer id. This variable holds that id and supports the minimal interface needed to participate as a dict key (hashing and equality). It intentionally blocks reconstruction so that a graph break does not silently bake a stale id into the resumed bytecode. rrkr'rrrc Htjdi|||_dS)Nr)rQrRrrVs r!rRzFakeIdVariable.__init__s+""6""" r$c|jSrr rZs r!rdz!FakeIdVariable.as_python_constantr\r$r%cdSNFrrZs r!rhz!FakeIdVariable.is_python_constantsur$rJctSr)rkrZs r! python_typezFakeIdVariable.python_types r$cdSrgrrZs r!rz!FakeIdVariable.is_python_hashablerir$c*t|jSrrrZs r!rzFakeIdVariable.get_python_hashrr$rrycxt|ttfr|j|kSdSr )r:rrrrd)rWrs r!rzFakeIdVariable.is_python_equals8 en.>? @ @ <:!9!9!;!;; ;ur$codegencRtdt|jddgdS)Nz Reconstruction of FakeIdVariablezdA fake id produced by id() on a compile-time container cannot be reconstructed across a graph break.zr!s#"""""888888888888&&&&&& ......99999999JJJJJJJJJJJJ333333330((((((DDDDDD//////H:H:H:H:H:H:H:H:V *)$//))$//**5112 2 2 2 2 _2 2 2 2 2 r$