[TVMScript] Connect assert_structural_equal with new TVMScript printer

[cyx-6]

This PR refactors the output of assert_structural_equal. Different from the directly printing mismatching nodes, in the old version, the improved one will print the whole scripts, with mismatching nodes underlined. And we print the ObjectPath to the mismatching nodes for further better debug. For example, we have following functions

@T.prim_func
def func1(a: T.handle, b: T.handle):
  A = T.match_buffer(a, (128, 128))
  B = T.match_buffer(b, (128, 128))

@T.prim_func
def func2(a: T.handle, b: T.handle):
  A = T.match_buffer(a, (128, 128))
  B = T.match_buffer(b, (128, 256))

the log of assert_structural_equal(func1, func2) will be like

ValueError: StructuralEqual check failed, caused by lhs at <root>.buffer_map[b].shape[1].value:
# from tvm.script import tir as T

@T.prim_func
def main(a: T.handle, b: T.handle):
  A = T.match_buffer(a, (128, 128))
  B = T.match_buffer(b, (128, 128))
                              ^^^
  T.evaluate(0)
and rhs at <root>.buffer_map[b].shape[1].value:
# from tvm.script import tir as T

@T.prim_func
def main(a: T.handle, b: T.handle):
  A = T.match_buffer(a, (128, 128))
  B = T.match_buffer(b, (128, 256))
                              ^^^
  T.evaluate(0)

instead of

ValueError: StructuralEqual check failed, caused by lhs:
128
and rhs:
256

which is not readable sometimes.

[tvm-bot]

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

• cc @junrushao, @shingjan _{See #10317 for details}

_{Generated by tvm-bot}

[junrushao]

Thanks @cyx-6 for the PR! A side note: you may note the PR description in the commit message :-)