def anyGraphWithTransposedAdjListComputed(request):
"""
Create a Graph (or DiGraph) obj based on the param, compute the transposed
adjacency list and return it.
"""
setFixtureParamNames(request, ["dataset", "managed_mem", "pool_allocator"])
csvFileName = request.param[0]
reinitRMM(request.param[1], request.param[2])
G = createGraph(csvFileName)
G.view_transposed_adj_list()
return G
def graphWithAdjListComputed(request):
"""
Create a Graph obj from the CSV file in param, compute the adjacency list
and return it.
"""
setFixtureParamNames(request, ["dataset", "managed_mem", "pool_allocator"])
csvFileName = request.param[0]
reinitRMM(request.param[1], request.param[2])
G = createGraph(csvFileName, cugraph.structure.graph_classes.Graph)
G.view_adj_list()
return G
def edgelistCreated(request):
"""
Returns a new edgelist created from a CSV, which is specified as part of
the parameterization for this fixture.
"""
# Since parameterized fixtures do not assign param names to param values,
# manually call the helper to do so. Ensure the order of the name list
# passed to it matches if there are >1 params.
# If the request only contains n params, only the first n names are set.
setFixtureParamNames(request, ["dataset", "managed_mem", "pool_allocator"])
csvFileName = request.param[0]
reinitRMM(request.param[1], request.param[2])
return utils.read_csv_file(csvFileName)
def daskGraphFromDataset(request, client_connection):
"""
Returns a new dask dataframe created from the dataset file param.
"""
# Since parameterized fixtures do not assign param names to param values,
# manually call the helper to do so.
setFixtureParamNames(request, ["dataset"])
dataset = request.param
chunksize = dcg.get_chunksize(dataset)
ddf = dask_cudf.read_csv(dataset,
chunksize=chunksize,
delimiter=' ',
names=['src', 'dst', 'value'],
dtype=['int32', 'int32', 'float32'])
dg = cugraph.DiGraph()
dg.from_dask_cudf_edgelist(ddf, 'src', 'dst')
return dg
def daskGraphFromDataset(request, dask_client):
"""
Returns a new dask dataframe created from the dataset file param.
"""
# Since parameterized fixtures do not assign param names to param values,
# manually call the helper to do so.
setFixtureParamNames(request, ["dataset"])
dataset = request.param
chunksize = dcg.get_chunksize(dataset)
ddf = dask_cudf.read_csv(
dataset,
chunksize=chunksize,
delimiter=" ",
names=["src", "dst", "value"],
dtype=["int32", "int32", "float32"],
)
dg = cugraph.DiGraph()
dg.from_dask_cudf_edgelist(ddf, "src", "dst")
return dg
