P/j:% ddlZddlmZmZmZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZmZddlmZmZmZd d lmZmZerdd lmZed ed ej8dej:dej<diZGddeZ GddeZ!y)N)DictList TYPE_CHECKING) GetItemSource) variables) unimplemented UserError UserErrorType) GuardBuilder install_guard)common_constant_typesistypenp)typestrVariableTracker)InstructionTranslatorz!List types must use ListVariable.z&Dict types must use ConstDictVariable.z%Tensor types must use TensorVariable.zvSymInts must use SymNodeVariable. If the underlying value is static, we will create a ConstantVariable and specialize.z SymInts must use SymNodeVariableceZdZedefdZdfd ZdZdefdZ dZ dZ e dZ d d d efd Zed ZdZddZ dfd Zd d deddfdZxZS)ConstantVariablereturnc |jdd}tj|}|s/tj D]\}}t ||sJ||ret |t tfrOg}t|D])\}}|jtj|+tj|fi|S|rt |ttfrg}t|D]e\}}|r t||nd} | r(t!| j#t$j&|jtj|| gtj(j+t-||fi|St|fi|S)Nsource)r)getr is_literal_type_to_assert_reasonitems isinstanceset frozenset enumerateappendcreater SetVariablelisttuplerr make_guardr CONSTANT_MATCHBaseListVariablecls_fortype) valuekwargsrrdisallowed_typereasonrix item_sources c/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/_dynamo/variables/constant.pyr#zConstantVariable.create&slHd+%007 +A+G+G+I F'%e_=EvE= F *US),<=E!%( 91 -44Q78 9((9&9 9 JutUm<E!%( 1:@mFA6d !+"8"89T9T"UV $++*, --55d5kB5SFS S000c `t|di|tj|s/tj D]\}}t ||sJ|t |ttfrJdt0t |tjr|j|_ y||_ y)NzFConstantVariable(list) is banned - please create a ListVariable(items)) super__init__rrrrrr%r&rnumberitemr,)selfr,r-r.r/ __class__s r3r8zConstantVariable.__init__Ds "6"**51+A+G+G+I F'%e_=EvE= F D%=  T S T  >j :DJDJr4c|jSNr,r;s r3as_proxyzConstantVariable.as_proxyR zzr4ctdt|jjdt|jdS)NzConstantVariable(z: ))r+r,__name__reprr@s r3__str__zConstantVariable.__str__Us2"4 #3#<#<"=RTZZ@P?QQRSSr4c|jSr>r?r@s r3as_python_constantz#ConstantVariable.as_python_constantXrBr4cy)NTr6r@s r3is_python_constantz#ConstantVariable.is_python_constant[sr4c&|jdS)zz Need this when adding a BaseListVariable and a ConstantVariable together. Happens in detectron2. N)tx)unpack_var_sequencer@s r3rzConstantVariable.items^s ''4'00r4rMrargcbtj|j|jSr>)rr#r,rI)r;rMrOs r3 getitem_constzConstantVariable.getitem_constfs*&& JJs--/ 0  r4ct|tvryt|tttt t jfvrtd|DSy)NTc3FK|]}tj|ywr>)rr).0r1s r3 z.ConstantVariable.is_literal..qsC!'2215Cs!F) r+rr%r&rr torchSizeall)objs r3rzConstantVariable.is_literalksA 9- - 9uc9ejjA ACsCC Cr4c |jDcgc]}tj|c}Scc}w#t$r }t|d}~wwxYwr>)rIrr# TypeErrorNotImplementedError)r;rMr1es r3rNz$ConstantVariable.unpack_var_sequencetsH -8<8O8O8QR1$++A.R RR -%1 , -s!8388 A AA ct|jtrttj ddt |j|}t|rt|S)Nz]Can't access members of type(obj) for a generated custom object. Please use __class__ insteadtype_reflection_method) case_name) rr,r+r r ANTI_PATTERNgetattrcallabler\r;rMnamemembers r3 const_getattrzConstantVariable.const_getattrzsS djj$ '**/2   T* F % % r4rc4 ddlm|dk(rQt|jtr7t j tjj||g||S|dk(rt|jtr~t|dk(rt|dk(sJ|dj|} |Dcgc]}|j}}tj|jj|St%fd|Dr@t j|j'|jj#||||S |Dcgc]}|j} }|j)D cic]\} } | | j} } } t+|jtrQ|tj,j/vr1t1|j|} tj| | i| St+|jt2t4fr|s0|s.tjt1|j|St7t8|r$t|dk(r|dj;r| d}t1t8|}t+|t<j>t<j@t<jBfr=|jDjGd||j|fi}j|||Stj||j|St+|jtHr6|dk(r1t1|j|} tj| | i| S|d k(r,|s*|s(tjt|jS|d k(r[t|dk(rM|dj;r:|rJ|dj}||jv}tj|StKd tM|jd |ycc}w#t$rt |E||||cYSwxYwcc}wcc} } w#t$rt |E||||cYSwxYw) Nr)SymNodeVariableformatjoinrc36K|]}t|ywr>)r)rTr1ris r3rUz/ConstantVariable.call_method..sc&kQ&6 667<?Q!..0?J?BH,,.Q$!QAq3355QLQ djj# &43<<3D3D3F+FTZZ.F#**6:+N+NO O  UCL 1F'../Hwtzz4/H/JKK$'ING..0'] Xt, MII22'djj*-ErE+11"eZHH+222djj*3MNN  E *tx/?TZZ.F#**6:+N+NO O 9 df#**3tzz?; ; ^ #D Q47;U;U;W :!W//1Ftzz)F#**62 2*74::+>*?qGHiK& Cw*2tT6BB C@Q" ?7&r4v> > ?sT1Q 5Q /Q Q9Q.,Q9Q3Q9Q Q+*Q+. Q99RRreclt|j|}tjj |Sr>)r}r,rrr#)r;rMrers r3 call_hasattrzConstantVariable.call_hasattrs)T*))0088r4rNrMr)rzList[VariableTracker]r-zDict[str, VariableTracker]rr)rE __module__ __qualname__ staticmethodrr#r8rArsrGrIrKpropertyrrQrrNrgrwr __classcell__r<s@r3rr%s1?11: TT11 7 o - EI& EI - EI  EIN969c9FW9r4rcPeZdZdfd ZedZdZdefdZdZ d dZ xZ S) EnumVariablerc 2t|di|||_y)Nr6)r7r8r,)r;r,r-r<s r3r8zEnumVariable.__init__s "6" r4ct|tjr8t|D]*}|j|j k(s!||fi|cSt dy)Nz5Enum variable is constructed with non constant values)rrrr%r,rIr )clscls_typevalue_vtoptionsrfs r3r#zEnumVariable.createsU h : : ;x. 2<<8#>#>#@@v111 2 MNr4c|jSr>r?r@s r3rAzEnumVariable.as_proxyrBr4c4dt|jdS)Nz EnumVariable(rD)r+r,r@s r3rGzEnumVariable.__str__stDJJ/022r4c|jSr>r?r@s r3rIzEnumVariable.as_python_constantrBr4cTt|j|}t|rt|Sr>)rbr,rcr\rds r3rgzEnumVariable.const_getattrs%T* F % % r4rr) rErrr8 classmethodr#rArsrGrIrgrrs@r3rrs9OO33r4r)"r~typingrrrrVtorch._dynamo.sourcerrexcr r r guardsr r utilsrrrbaserrtorch._dynamo.symbolic_convertrr%dictTensorrrrrrr6r4r3rs,, .99055*D - 2 LL9 LL[ NN6"k9k9\?r4