code: def __get_2D_div(ngpus):
      prows = int(math.sqrt(ngpus))
      while ngpus % prows != 0:
          prows = prows - 1
      return prows, int(ngpus / prows)
  
file_path: cugraph/python/cugraph/cugraph/dask/comms/comms.py
imports: from cugraph.dask.common.read_utils import MissingUCXPy
  from pylibraft.common.handle import Handle
  from cugraph.dask.comms.comms_wrapper import init_subcomms
  from dask.distributed import default_client, get_worker
  from cugraph.dask.common import read_utils
  import math
label_desc: Use image node_id 1 for calling a global function with example usage: __get_2D_div(ngpus) and returns: prows,  int