code: def dataloading_benchmark(
      g, train_idx, fanouts, batch_sizes, use_uva
  ):
      """
      Run the dataloading benchmark.
      Args:
          g: DGLGraph
          train_idx: Tensor containing the training indices.
          fanouts: List of fanouts to use for the dataloader.
          batch_sizes: List of batch sizes to use for the dataloader.
          use_uva: Whether to use unified virtual address space.
      """
      time_ls = []
      for fanout in fanouts:
          for batch_size in batch_sizes:
              dataloader = create_dataloader(
                  g,
                  train_idx,
                  batch_size=batch_size,
                  fanouts=fanout,
                  use_uva=use_uva,
              )
              dataloading_st = time.time()
              for input_nodes, output_nodes, blocks in dataloader:
                  pass
              dataloading_et = time.time()
              dataloading_time = dataloading_et - dataloading_st
              time_d = {
                  "fanout": fanout,
                  "batch_size": batch_size,
                  "dataloading_time_per_epoch": dataloading_time,
                  "dataloading_time_per_batch": dataloading_time
                  / len(dataloader),
                  "num_edges": g.num_edges(),
                  "num_batches": len(dataloader),
              }
              time_ls.append(time_d)
  
              print("Dataloading completed")
              print(f"Fanout = {fanout}, batch_size = {batch_size}")
              print(
                  f"Time taken {dataloading_time:.2f} ",
                  f"seconds for num batches {len(dataloader)}",
                  flush=True,
              )
              print("==============================================")
      return time_ls
  
file_path: cugraph/benchmarks/cugraph-dgl/python-script/dgl_dataloading_benchmark/dgl_benchmark.py
imports: import dgl
  import torch
  import pandas
  import os
  import time
  import json
  import random
  import numpy
  from argparse import ArgumentParser
label_desc: Use image node_id 5 for calling a global function with example usage: dataloading_benchmark(g, train_idx, fanouts, batch_sizes, use_uva) and returns: time_ls