def test_mg_edge_betweenness_centrality(
graph_file,
directed,
subset_size,
normalized,
weight,
subset_seed,
result_dtype,
mg_device_count,
):
prepare_test()
skip_if_not_enough_devices(mg_device_count)
with MGContext(number_of_devices=mg_device_count, p2p=True):
sorted_df = calc_edge_betweenness_centrality(
graph_file,
directed=directed,
normalized=normalized,
k=subset_size,
weight=weight,
seed=subset_seed,
result_dtype=result_dtype,
multi_gpu_batch=True,
)
compare_scores(
sorted_df,
first_key="cu_bc",
second_key="ref_bc",
epsilon=DEFAULT_EPSILON,
)
def test_mg_edge_betweenness_centrality(graph_file, directed, subset_size,
normalized, weight, subset_seed,
result_dtype, dask_client):
sorted_df = calc_edge_betweenness_centrality(
graph_file,
directed=directed,
normalized=normalized,
k=subset_size,
weight=weight,
seed=subset_seed,
result_dtype=result_dtype,
multi_gpu_batch=True,
)
compare_scores(
sorted_df,
first_key="cu_bc",
second_key="ref_bc",
epsilon=DEFAULT_EPSILON,
)