def test_mg_betweenness_centrality( graph_file, directed, subset_size, normalized, weight, endpoints, subset_seed, result_dtype, dask_client, ): sorted_df = calc_betweenness_centrality( graph_file, directed=directed, normalized=normalized, k=subset_size, weight=weight, endpoints=endpoints, 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_betweenness_centrality( graph_file, directed, subset_size, normalized, weight, endpoints, 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_betweenness_centrality( graph_file, directed=directed, normalized=normalized, k=subset_size, weight=weight, endpoints=endpoints, 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, )