^jdZddlmZddlZddlmZddlmZddlm Z dd l m Z erdd l m Z  dd e d d ddfdZde defdZGddZGddZy)a) Hashability utilities for PyTorch Dynamo variable tracking. This module provides the HashableTracker wrapper class and associated utilities for making VariableTracker instances usable as dictionary keys and set elements during symbolic execution. Used by both ConstDictVariable and SetVariable. ) TYPE_CHECKINGN) variables)raise_observed_exception)specialize_symnode)VariableTracker)InstructionTranslatorBaseargtxz InstructionTranslatorBase | Nonereturncr|ddlm}|j} |j}|A|j 5ddlm}ddlm }|dd|d|jd g|j | |jnt|j}tt|d |d g y#t$rd}YwxYw)NrInstructionTranslatorr)graph_break_hints) unimplementedzHashability mismatchzraise_unhashable zDynamo thinks zr is unhashable, but its __hash__ is not None. This likely indicates a missing or incorrect is_hashable() override.)gb_typecontext explanationhintszunhashable type: '')args)torch._dynamo.symbolic_convertr current_tx python_type Exception__hash__rexcr__name__ DYNAMO_BUGtyper TypeError)r r rarg_typerr type_names k/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/_dynamo/variables/hashable.pyraise_unhashabler's zH " - - /??$ 1 1 =('*'u-():):(;<342%001  &.%9!!tCy?Q?QI   1 - ) sB(( B65B6xct|tjr!|js|j ry|j S)NT) isinstancerLazyVariableTracker is_realizedis_hashable_lazy is_hashable)r(s r&r.r.>s8 1i334     ==?c@eZdZdZdZdeddfdZdefdZdede fd Z y) RawHasha5Wraps a pre-computed hash value to bypass int.__hash__'s modular reduction. When building a tuple/frozenset of per-item hashes, using bare ints would apply long_hash (mod sys.hash_info.modulus), corrupting the values. Wrapping in RawHash makes tuplehash/frozenset_hash see the original hash. hr3r Nc||_yNr2)selfr3s r&__init__zRawHash.__init__Us r/c|jSr5r2r6s r&rzRawHash.__hash__Xs vv r/othercXt|txr|j|jk(Sr5)r*r1r3)r6r:s r&__eq__zRawHash.__eq__[s!%)?dff.??r/) r __module__ __qualname____doc__ __slots__intr7robjectboolr<r/r&r1r1KsCI#$#@F@t@r/r1cdeZdZdZeZdeddfdZededefdZ de fdZ dede fd Z y) HashableTrackerz Class that wraps a VariableTracker and makes it hashable. Note that it's fine to put VTs into dictionaries and sets, but doing so does not take into account aliasing. vtr NcBt|}t|tjrF|j s6|j r&t |j|_||_ yddl m }ddl m }|j}|||\|_}||_ y)Nrrr)generic_hash_impl)rr*rr+r,r-hashoriginal_value_hashrGrrobject_protocolrIr)r6rGrrIr _s r&r7zHashableTracker.__init__hs  # r988 9NN$##%b//12DJDG  I6 " - - /)"b1  Ar/c@ddlm}ddlm}t |t j rH|js8t |jtjr|jSt ||s |jSg}|jD]}|jr |j|j3t ||rt!|dd}t!|dd}| t!|ddnd}t |t"r|j%dnd} t!| dd} t | tj&r3| j)dk(r |j| j+|jcStj|S) Nr) SizeVariable)TensorVariableproxynodemeta example_valueconstant)listsrPtensorrQr*rr+r,rKtorchSize_MISSINGitemsis_python_constantappendas_python_constantgetattrdictgetTensornumelitem) clsrGrPrQr\rerRrSrTrUrVs r&_maybe_constant_torch_sizez*HashableTracker._maybe_constant_torch_sizesM'* r988 9NN$2,,. ;$$& &"l+<< HH D&&( T4467$/gt4ufd36:6FwtVT2D1;D$1GDHH_-T#=*dCh 5(..:Ja:OLL1<< % (zz%  r/c|jSr5)rLr9s r&rzHashableTracker.__hash__s zzr/r:cNt|tsy|j|jury|j|j}|j|j}||jur||jur||k(S|jj |jS)a Checks equality between two HashableTracker instances. Delegates to the VariableTracker's is_python_equal method to compare the underlying variable trackers for Python-level equality. Args: other: Another HashableTracker instance to compare with Returns: True if the underlying variable trackers are Python-equal, False otherwise FT)r*rFrGrgr[is_python_equal)r6r: self_constantother_constants r&r<zHashableTracker.__eq__s%1 77ehh 77@ 88B  -. 2U N2 2ww&&uxx00r/)r r=r>r?rBr[r r7 classmethodrgrArrCr<rDr/r&rFrF_se xH?t4#!O#!#!#!J#1F1t1r/rFr5)r?typingrrYrrrrutilsrbaser rr r'rCr.r1rFrDr/r&rqsv! *&!HDH$ $@$ $N ? t @@(c1c1r/