code: def setup_common_pool():
      rmm.reinitialize(initial_pool_size=5e9, pool_allocator=True)
      torch.cuda.memory.change_current_allocator(rmm_torch_allocator)
  
file_path: cugraph/benchmarks/cugraph-dgl/scale-benchmarks/cugraph_dgl_benchmark.py
imports: import os
  import json
  import pandas
  import os
  import time
  from rmm.allocators.torch import rmm_torch_allocator
  import rmm
  import torch
  from cugraph_dgl.dataloading import HomogenousBulkSamplerDataset
  from model import run_1_epoch
  from argparse import ArgumentParser
  from load_graph_feats import load_node_labels, load_node_features
label_desc: Use image node_id 2 for calling a global function with example usage: setup_common_pool() without return types