DASK MG Helper functions#

Methods#

cugraph.dask.comms.comms.initialize([comms, ...])

Initialize a communicator for multi-node/multi-gpu communications.

cugraph.dask.comms.comms.destroy()

Shuts down initialized comms and cleans up resources.

cugraph.dask.comms.comms.is_initialized()

Returns True if comms was initialized, False otherwise.

cugraph.dask.comms.comms.get_comms()

Returns raft Comms instance

cugraph.dask.comms.comms.get_workers()

Returns the workers in the Comms instance, or None if Comms is not initialized.

cugraph.dask.comms.comms.get_session_id()

Returns the sessionId for finding sessionstate of workers, or None if Comms is not initialized.

cugraph.dask.comms.comms.get_2D_partition()

Returns a tuple representing the 2D partition information: (prows, pcols, partition_type)

cugraph.dask.comms.comms.get_default_handle()

Returns the default handle.

cugraph.dask.comms.comms.get_handle(sID[, ...])

Returns the handle from within the worker using the sessionstate.

cugraph.dask.comms.comms.get_worker_id(sID)

Returns the worker's sessionId from within the worker.

cugraph.dask.common.read_utils.get_chunksize(...)

Calculate the appropriate chunksize for dask_cudf.read_csv to get a number of partitions equal to the number of GPUs.