code: def split_gw(gw, partition):
      gw_list = []
      for i in range(partition):
          gw_list.append(PartitionWrapper(gw, i, partition))
      return gw_list
  
file_path: PGL/ogb_examples/nodeproppred/ogbn-proteins/unimp_cross_feat/module/transformer_gat_pgl.py
imports: import paddle.fluid
  import paddle.fluid.layers
  from pgl import graph_wrapper
  from pgl.utils import paddle_helper
  import math
label_desc: Use image node_id 2 for calling a global function with example usage: split_gw(gw, partition) and returns: gw_list