Function assert_structural_equal

python/tvm/ir/base.py:165–203

Assert lhs and rhs are structurally equal to each other. Parameters ---------- lhs : Object The left operand. rhs : Object The left operand. map_free_vars : bool Whether or not shall we map free vars that does not bound to any definitions as equ

(lhs, rhs, map_free_vars=False)

def assert_structural_equal(lhs, rhs, map_free_vars=False):
    """Assert lhs and rhs are structurally equal to each other.

    Parameters
    ----------
    lhs : Object
        The left operand.

    rhs : Object
        The left operand.

    map_free_vars : bool
        Whether or not shall we map free vars that does
        not bound to any definitions as equal to each other.

    Raises
    ------
    ValueError : if assertion does not hold.

    See Also
    --------
    tvm_ffi.structural_equal
    """
    first_mismatch = tvm_ffi.get_first_structural_mismatch(lhs, rhs, map_free_vars)
    if first_mismatch is not None:
        from tvm.runtime.script_printer import (  # pylint: disable=import-outside-toplevel
            PrinterConfig,
            _script,
        )

        lhs_path, rhs_path = first_mismatch
        lhs_script = _script(lhs, PrinterConfig(syntax_sugar=False, path_to_underline=[lhs_path]))
        rhs_script = _script(rhs, PrinterConfig(syntax_sugar=False, path_to_underline=[rhs_path]))
        raise ValueError(
            f"StructuralEqual check failed, caused by lhs at {lhs_path}:\n"
            f"{lhs_script}\n"
            f"and rhs at {rhs_path}:\n"
            f"{rhs_script}"
        )


def deprecated(