tjdZddlmZddlZddlmZddlmZddlm Z dd l m Z erdd l m Z  dd e d d ddfdZde defdZGddZGddZdS)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 | Nonereturncn|ddlm}|} |}n#t$rd}YnwxYw|5|j.ddlm}ddlm }|dd|d|j d g|j  ||j nt|j }tt|d |d g dS)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 ^/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/_dynamo/variables/hashable.pyraise_unhashabler's? zHHHHHH " - - / /??$$   1 =(((((('''''' *---4):4442%01     &.%9!!tCyy?QI - - - - s3 AAxct|tjr*|s|rdS|S)NT) isinstancerLazyVariableTracker is_realizedis_hashable_lazy is_hashable)r(s r&r.r.>sT 1i344      t ==??cBeZdZdZdZdeddfdZdefdZdede fd Z dS) 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||_dSNr2)selfr3s r&__init__zRawHash.__init__Us r/c|jSr5r2r6s r&rzRawHash.__hash__Xs v r/othercLt|to|j|jkSr5)r*r1r3)r6r:s r&__eq__zRawHash.__eq__[s %))?df.??r/) r __module__ __qualname____doc__ __slots__intr7robjectboolr<r/r&r1r1KsI#$#@F@t@@@@@@r/r1creZdZdZeZdeddfdZededefdZ de fdZ dede fd Z dS) 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 Nc|t|}t|tjrW|sC|r/t ||_||_ dSddl m }ddl m }|}|||\|_}||_ dS)Nrrr)generic_hash_impl)rr*rr+r,r-hashoriginal_value_hashrGrrobject_protocolrIr)r6rGrrIr _s r&r7zHashableTracker.__init__hs  # # r98 9 9 NN$$ ##%%  b//1122DJDG F IHHHHH666666 " - - / /))"b11  Ar/cddlm}ddlm}t |t jrT|s@t |tj r|St ||s|j Sg}|j D]%}| r(||?t ||rt!|dd}t!|dd}|t!|ddnd}t |t"r|dnd} t!| dd} t | tjrA| dkr)|| |j cStj |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_sizes''''''****** r98 9 9 'NN$$ '2,,.. ;; ' $$&& &"l++ < H  D&&((  T4466777$// gt44ufd336:6FwtVT222D1;D$1G1GQDHH_---T#=*dCCh 55(..:J:Ja:O:OLL111<   z%   r/c|jSr5)rLr9s r&rzHashableTracker.__hash__s zr/r:c&t|tsdS|j|jurdS||j}||j}||jur||jur||kS|j|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%11 5 7eh  477@@ 88BB  - -. 2U2U N2 2w&&ux000r/)r r=r>r?rBr[r r7 classmethodrgrArrCr<rDr/r&rFrF_s vxxH?t4#!O#!#!#!#![#!J#1F1t111111r/rFr5)r?typingrrYrrrrutilsrbaser rr r'rCr.r1rFrDr/r&rqsX! ******&&&&&&!!!!!!IHHHHHHDH$$ $@$ $$$$N ? t    @@@@@@@@(c1c1c1c1c1c1c1c1c1c1r/