code: def test_compute_group():
      m = te.size_var("m")
      n = te.size_var("n")
      x = te.compute(
          (m, n), lambda i, j: tvm.tir.const(1, "float32"), name="x"
      )
      x1 = te.compute(x.shape, lambda *i: x(*i) + 1, name="x1")
      x2 = te.compute(x.shape, lambda *i: x1(*i) + 2, name="x2")
      s = te.create_schedule(x2.op)
      g = s.create_group(outputs=x1, inputs=x, include_inputs=True)
      assert s[x1].group == g
      assert s[x].group == g
      g.compute_at(s[x2], x2.op.axis[1])
      assert g.attach_stage == s[x2]
      assert g.num_child_stages == 2
  
imports: import tvm
  from tvm import te
label_desc: Use image node_id 2 for calling a global function with example usage: test_compute_group() without return types