code: def test_pass_simple():
      @T.prim_func
      def element_wise(
          A: T.Buffer((128, 128), "float32"),
          C: T.Buffer((128, 128), "float32"),
      ):
          B = T.alloc_buffer((128, 128), "float32")
          for i, j in T.grid(128, 128):
              with T.block("B"):
                  vi, vj = T.axis.remap("SS", [i, j])
                  B[vi, vj] = A[vi, vj] * 2.0
          for i, j in T.grid(128, 128):
              with T.block("C"):
                  # It's a opaque block , so it can use outside variables
                  C[i, j] = B[i, j] * 2.0
  
      assert tvm.tir.analysis.verify_well_formed(element_wise)
      assert tvm.tir.analysis.verify_well_formed(
          tvm.IRModule.from_expr(element_wise)
      )
  
file_path: tvm/tests/python/tir-analysis/test_tir_analysis_verify_well_formed.py
imports: import pytest
  import tvm
  import tvm.testing
  from tvm.script import ir, tir
label_desc: Use image node_id 1 for calling a global function with example usage: test_pass_simple() without return types