code: def test_alloc_in_loop(self):
      a, tmp, b = (
          te.BufHandle(name, [1], torch.float32)
          for name in ["a", "tmp", "b"]
      )
      body = te.Block(
          [tmp.store([0], a.load([0])), b.store([0], tmp.load([0]))]
      )
      for _ in range(4):
          i = te.VarHandle("i", torch.int32)
          body = te.For.make(i, 0, 100, body)
      nest = te.LoopNest(body, [b])
      nest.prepare_for_codegen()
      f = te.construct_codegen("llvm", nest.simplify(), [a, b])
      ta, tb = (torch.ones(1) for _ in range(2))
      f.call([ta.data_ptr(), tb.data_ptr()])
  
imports: import torch
  import numpy
  import torch._C._te
  from torch.testing._internal.common_utils import run_tests
  from torch.testing._internal.jit_utils import JitTestCase
  import unittest
label_desc: Use image node_id 16 for calling the TestTensorExprPyBind obj's underlying member method code with example usage: obj.test_alloc_in_loop() without return types