code: def get_schedule(args, read_op):
      schedule = {}
      if read_op:
          schedule["pre"] = pre_basic_read
          schedule["post"] = post_basic
          schedule["main"] = main_basic_read
      else:
          schedule["pre"] = pre_basic_write
          schedule["post"] = post_basic
          schedule["main"] = main_basic_write
  
      return schedule
  
file_path: DeepSpeed/deepspeed/ops/csrc/aio/py_test/ds_aio_basic.py
imports: import torch
  import os
  import time
  from multiprocessing import Pool, Barrier
  from test_ds_aio_utils import report_results, task_log, task_barrier
  from deepspeed.accelerator import get_accelerator
  from deepspeed.ops.op_builder import AsyncIOBuilder
label_desc: Use image node_id 7 for calling a global function with example usage: get_schedule(args, read_op) and returns: schedule