code: def _check_used_memory_annotations(
      mod, expected_annotations, expected_io_annotation
  ):
      mod = relay.transform.InferType()(mod)
      mod = relay.transform.ToANormalForm()(mod)
      mod = relay.transform.InferType()(mod)
      mod = AnnotateUsedMemory()(mod)
  
      CheckUsedMemoryAnnotation(
          expected_annotations, expected_io_annotation
      )(mod["main"])
  
imports: import pytest
  import tvm
  from tvm import relay
  from tvm.relay.expr_functor import ExprVisitor
label_desc: Use image node_id 2 for calling a global function with example usage: _check_used_memory_annotations(mod, expected_annotations, expected_io_annotation) without return types