From aabda424e7782b4dc120d77384b7bcb9db68cfe9 Mon Sep 17 00:00:00 2001 From: Alex Barghi <105237337+alexbarghi-nv@users.noreply.github.com> Date: Fri, 9 Sep 2022 11:47:44 -0400 Subject: [PATCH] Small fix to the MG PyG Test to Account for Current Sampling Behavior (#2666) This PR will correct an error that shows up occasionally in the MG tests, especially when running on an odd number of GPUs. This error is related to issue #2665 and resolves it temporarily while the C++ team works on a permanent fix. Authors: - Alex Barghi (https://github.com/alexbarghi-nv) Approvers: - Rick Ratzel (https://github.com/rlratzel) URL: https://github.com/rapidsai/cugraph/pull/2666 --- .../cugraph/cugraph/tests/mg/test_mg_pyg_extensions.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/python/cugraph/cugraph/tests/mg/test_mg_pyg_extensions.py b/python/cugraph/cugraph/tests/mg/test_mg_pyg_extensions.py index f146250114c..436b7b193fc 100644 --- a/python/cugraph/cugraph/tests/mg/test_mg_pyg_extensions.py +++ b/python/cugraph/cugraph/tests/mg/test_mg_pyg_extensions.py @@ -430,7 +430,10 @@ def test_neighbor_sample(single_vertex_graph): noi_groups, row_dict, col_dict, _ = graph_store.neighbor_sample( index=cupy.array([0, 1, 2, 3, 4], dtype='int32'), - num_neighbors=[-1], + # FIXME The following line should be num_neighbors=[-1] but + # there is currently a bug in MG uniform_neighbor_sample. + # Once this bug is fixed, this line should be changed. + num_neighbors=[8], replace=True, directed=True, edge_types=[ @@ -487,7 +490,10 @@ def test_neighbor_sample_multi_vertex( noi_groups, row_dict, col_dict, _ = graph_store.neighbor_sample( index=cupy.array([0, 1, 2, 3, 4], dtype='int32'), - num_neighbors=[-1], + # FIXME The following line should be num_neighbors=[-1] but + # there is currently a bug in MG uniform_neighbor_sample. + # Once this bug is fixed, this line should be changed. + num_neighbors=[8], replace=True, directed=True, edge_types=[