code: def main_parallel_write(pool_params):
      args, tid, ctxt = pool_params
      handle = ctxt["handle"]
      start_time = time.time()
      ret = handle.pwrite(
          ctxt["buffer"], ctxt["file"], args.validate, True
      )
      assert ret != -1
      handle.wait()
      end_time = time.time()
      ctxt["elapsed_sec"] += end_time - start_time
  
      return ctxt
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/ops/csrc/aio/py_test/ds_aio_handle.py
imports: import torch
  import os
  import time
  from multiprocessing import Pool, Barrier
  from deepspeed.ops.aio import AsyncIOBuilder
  from test_ds_aio_utils import report_results, task_log, task_barrier
label_desc: Use image node_id 6 for calling a global function with example usage: main_parallel_write(pool_params) and returns: ctxt