code: def create_dataframe(client):
      n_rows = 25_000_000
      df = cudf.DataFrame(
          {
              "src": cp.arange(0, n_rows, dtype=cp.int32),
              "dst": cp.arange(0, n_rows, dtype=cp.int32),
              "eids": cp.ones(n_rows, cp.int32),
          }
      )
      ddf = dask_cudf.from_cudf(
          df, npartitions=len(client.scheduler_info()["workers"])
      ).persist()
      client.rebalance(ddf)
      del df
      _ = wait(ddf)
      return ddf
  
file_path: cugraph/benchmarks/shared/build_cugraph_ucx/test_client_bandwidth.py
imports: from dask_cuda import LocalCUDACluster
  from dask.distributed import Client, wait
  import cupy
  import numpy
  import cudf
  import dask_cudf
  import rmm
  from time import perf_counter_ns
label_desc: Use image node_id 2 for calling a global function with example usage: create_dataframe(client) and returns: ddf