code: def multi_device(reader, dev_count):
      """multi device"""
      if dev_count == 1:
          for batch in reader:
              yield batch
      else:
          batches = []
          for batch in reader:
              batches.append(batch)
              if len(batches) == dev_count:
                  yield batches
                  batches = []
  
file_path: PGL/ogb_examples/linkproppred/ogbl-ppa/monitor/train_monitor.py
imports: import tqdm
  import json
  import numpy
  import sys
  import os
  import paddle.fluid
  from pgl.utils.log_writer import LogWriter
  from ogb.linkproppred import Evaluator
  from ogb.linkproppred import LinkPropPredDataset
label_desc: Use image node_id 1 for calling a global function with example usage: multi_device(reader, dev_count) without return types