code: def test_bound_group_schedule():
      m = te.var("m")
      n = te.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)
      g.compute_at(s[x2], x2.op.axis[0])
      assert s[x1].group == g
      assert s[x].group == g
      s = s.normalize()
      bounds = tvm.te.schedule.InferBound(s)
      assert bounds[x.op.axis[0]].extent.value == 1
      assert bounds[x.op.axis[1]].extent == n
  
imports: import tvm
  import tvm.testing
  from tvm import te
label_desc: Use image node_id 15 for calling a global function with example usage: test_bound_group_schedule() without return types