code: def get_scale_by_two_schedule():
      mod = tvm.IRModule.from_expr(
          scale_by_two.with_attr("global_symbol", "main")
      )
      sch = tir.Schedule(mod, debug_mask="all")
      block_c = sch.get_block("C")
      (flat,) = sch.get_loops(block_c)
      outer, _, _, _ = sch.split(flat, factors=[8, 4, 2, 128])
      cache_block = sch.cache_read(
          block_c, 0, storage_scope="global.vtcm"
      )
      sch.compute_at(cache_block, outer)
      return sch
  
file_path: tvm/tests/python/contrib/test_hexagon/test_vtcm.py
imports: import pytest
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from .infrastructure import get_hexagon_target
label_desc: Use image node_id 2 for calling a global function with example usage: get_scale_by_two_schedule() and returns: sch