o YÛ·i¾ã@s¢dZddlmZddlmZmZerddlmZddlm Z ddl m Z ddl m Z gd ¢Z dd dd„Z dd!dd„Z dd"dd„Ze dƒGdd„de ƒƒZdS)#z(Structural helper objects and functions.é)Ú annotations)Ú TYPE_CHECKINGÚAnyé)Ú AccessPath)Ú_ffi_api)ÚObject)Úregister_object)Ú StructuralKeyÚget_first_structural_mismatchÚstructural_equalÚstructural_hashFÚlhsrÚrhsÚ map_free_varsÚboolÚskip_tensor_contentÚreturncCót ||||¡S)a¯Check structural equality between two values. Structural equality compares the *shape/content structure* of two values instead of Python object identity. For container-like values, this means recursive comparison of elements/fields. For object types that provide structural equal hooks, those hooks are used. Parameters ---------- lhs Left-hand side value. rhs Right-hand side value. map_free_vars Whether free variables (variables without a definition site) can be mapped to each other during comparison. skip_tensor_content Whether to skip tensor data content when comparing tensors. Returns ------- result Whether the two values are structurally equal. Examples -------- .. code-block:: python import tvm_ffi assert tvm_ffi.structural_equal([1, 2, 3], [1, 2, 3]) assert not tvm_ffi.structural_equal([1, 2, 3], [1, 2, 4]) See Also -------- :py:func:`tvm_ffi.structural_hash` Hash function compatible with structural equality. :py:func:`tvm_ffi.get_first_structural_mismatch` Mismatch diagnostics with access paths. )rÚStructuralEqual©rrrr©rúH/home/ubuntu/vllm_env/lib/python3.10/site-packages/tvm_ffi/structural.pyr 's.r ÚvalueÚintcCst |||¡d@S)a—Compute structural hash of a value. This hash is designed to be consistent with :py:func:`structural_equal` under the same options. If two values are structurally equal, they should have the same structural hash. Parameters ---------- value Input value to hash. map_free_vars Whether free variables mapped to each other during hashing. skip_tensor_content Whether tensor data content is ignored when hashing tensors. Returns ------- hash_value Structural hash value. Examples -------- .. code-block:: python import tvm_ffi h0 = tvm_ffi.structural_hash([1, 2, 3]) h1 = tvm_ffi.structural_hash([1, 2, 3]) assert h0 == h1 Notes ----- Structural hash is intended for hash-table bucketing and fast pre-checks. Always use structural equality to confirm semantic equivalence. ìÿÿÿÿ)rÚStructuralHash)rrrrrrr Xs+r ú$tuple[AccessPath, AccessPath] | NonecCr)a„Like structural_equal(), but returns the AccessPath pair of the first detected mismatch. Parameters ---------- lhs The left operand. rhs The right operand. map_free_vars Whether free variables (i.e. variables without a definition site) should be mapped as equal to each other. skip_tensor_content Whether to skip the data content of tensor. Returns ------- mismatch: tuple[AccessPath, AccessPath] | None `None` if `lhs` and `rhs` are structurally equal. Otherwise, a tuple of two AccessPath objects that point to the first detected mismatch. )rÚGetFirstStructuralMismatchrrrrr †sr zffi.StructuralKeyc@sNeZdZUdZded<ded<erddd „Zdd d „Zdd d„Zddd„Z dS)r aÌHash-cached structural key wrapper. This wrapper can be used to hint that a dict uses structural equality and hash for the key. Examples -------- Use ``StructuralKey`` with Python dictionaries when you want key lookup by structural semantics: .. code-block:: python import tvm_ffi k0 = tvm_ffi.StructuralKey([1, 2, 3]) k1 = tvm_ffi.StructuralKey([1, 2, 3]) k2 = tvm_ffi.StructuralKey([1, 2, 4]) d = {k0: "value-a", k2: "value-b"} assert d[k1] == "value-a" # k1 matches k0 structurally assert d[k2] == "value-b" It can also be used directly with :py:class:`tvm_ffi.Map`: .. code-block:: python m = tvm_ffi.Map({k0: 1, k1: 2}) assert len(m) == 1 assert m[k0] == 2 See Also -------- :py:func:`tvm_ffi.structural_equal` Structural equality comparison. :py:func:`tvm_ffi.structural_hash` Structural hash computation. rÚkeyrÚhash_i64rrcCsdS)Nr©ÚselfrrrÚ__ffi_shallow_copy__Ñsz"StructuralKey.__ffi_shallow_copy__ÚNonecCs| tj|¡dS)z¦Create a structural key from ``key``. Parameters ---------- key The underlying value used for structural hash/equality. N)Ú__init_handle_by_constructor__rr )r"rrrrÚ__init__Õs zStructuralKey.__init__cCs |jd@S)zReturn cached structural hash.r)r r!rrrÚ__hash__às zStructuralKey.__hash__ÚotherrcCst|tƒo t ||¡S)zCompare by structural equality.)Ú isinstancer rÚStructuralKeyEqual)r"r(rrrÚ__eq__æszStructuralKey.__eq__N)rr)rrrr$)rr)r(rrr) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú__annotations__rr#r&r'r+rrrrr ¤s (   r N)FF) rrrrrrrrrr)rrrrrrrr) rrrrrrrrrr)r/Ú __future__rÚtypingrrÚ access_pathrÚrÚcorerÚregistryr Ú__all__r r r r rrrrÚs"      ÿ2 ÿ/ ÿ