code: def test_highlight_script():
      @tvm.script.ir_module
      class Module:
          @T.prim_func
          def main(  # type: ignore
              a: T.handle,
              b: T.handle,
              c: T.handle,
          ) -> None:  # pylint: disable=no-self-argument
              T.func_attr(
                  {"global_symbol": "main", "tir.noalias": True}
              )
              A = T.match_buffer(a, [16, 128, 128])
              B = T.match_buffer(b, [16, 128, 128])
              C = T.match_buffer(c, [16, 128, 128])
              for n, i, j, k in T.grid(16, 128, 128, 128):
                  with T.block("matmul"):
                      vn, vi, vj, vk = T.axis.remap(
                          "SSSR", [n, i, j, k]
                      )
                      with T.init():
                          C[vn, vi, vj] = 0.0  # type: ignore
                      C[vn, vi, vj] = (
                          C[vn, vi, vj] + A[vn, vi, vk] * B[vn, vj, vk]
                      )
  
      Module.show()
      Module["main"].show()
      Module["main"].show(style="light")
      Module["main"].show(style="dark")
      Module["main"].show(style="ansi")
  
file_path: tvm/tests/python/tvmscript/test_tvmscript_printer_highlight.py
imports: import pytest
  import tvm
  import tvm.testing
  from tvm import relay
  from tvm.script import tir
  from tvm.script.highlight import cprint
label_desc: Use image node_id 1 for calling a global function with example usage: test_highlight_script() without return types