code: def test_annotate_entry_func_single_primfunc():
      @tvm.script.ir_module
      class MockModule:
          @T.prim_func(private=True)
          def func1(A: T.Buffer((16,), "float32")):
              for i in T.serial(16):
                  if i == 5:
                      if i == 5:
                          A[i] = 0.0
  
      mod = MockModule
      assert mod
      assert mod["func1"].attrs is None
      after = tvm.tir.transform.AnnotateEntryFunc()(mod)
      assert (
          after["func1"].attrs
          and "tir.is_entry_func" in after["func1"].attrs
          and after["func1"].attrs["tir.is_entry_func"]
      )
  
imports: import pytest
  import tvm
  from tvm.script import tir, ir
  import tvm.testing
label_desc: Use image node_id 1 for calling a global function with example usage: test_annotate_entry_func_single_primfunc() without return types