dgl.to_cugraph¶

dgl.to_cugraph(g)[source]¶

Convert a DGL graph to a cugraph.Graph and return.

Parameters: g (DGLGraph) – A homogeneous graph.
Returns: The converted cugraph graph.
Return type: cugraph.Graph

Notes

The function only supports GPU graph input.

Examples

The following example uses PyTorch backend.

>>> import dgl
>>> import cugraph
>>> import torch

>>> g = dgl.graph((torch.tensor([1, 2]), torch.tensor([1, 3]))).to('cuda')
>>> cugraph_g = g.to_cugraph()
>>> cugraph_g.edges()
    src  dst
0    2    3
1    1    1