code: def run_passes(sch, args):
      mod = schedule_to_module(sch, args)
      return tvm.transform.Sequential(
          [
              tvm.tir.transform.StorageFlatten(64),
              tvm.tir.transform.Simplify(),
              tvm.tir.transform.VectorizeLoop(),
              tvm.tir.transform.StorageRewrite(),
              tvm.tir.transform.MergeDynamicSharedMemoryAllocations(),
          ]
      )(mod)
  
imports: import numpy
  import tvm
  import tvm.testing
  from tvm import te
  from tvm.driver.build_module import schedule_to_module
  from tvm.topi.math import cast
  from tvm.script import tir
label_desc: Use image node_id 1 for calling a global function with example usage: run_passes(sch, args) and returns: tvm