Skip to content

Commit

Permalink
Update MG Benchmark List (#4516)
Browse files Browse the repository at this point in the history
This PR partially addresses #4504

---

### Changes

1. Remove `force_atlas2` from the list of benchmarks being run inside `benchmarks/cugraph/pytest-based/bench_algos.py` . This algorithm is not designed to be used in an MG workflow. It was being skipped previously, but it would make more sense to remove it from the list of algos.

2. Mark `tests/community/test_triangle_count_mg.py` to be skipped in the nightly MNMG tests while a bug is being investigated. This will prevent the slowdown of subsequent tests in the meantime. The test will be re-introduced once there is a fix.

Authors:
  - Ralph Liu (https://github.com/nv-rliu)

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)

URL: #4516
  • Loading branch information
nv-rliu authored Jul 3, 2024
1 parent 2b1f452 commit c1f16fa
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 7 deletions.
8 changes: 1 addition & 7 deletions benchmarks/cugraph/pytest-based/bench_algos.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2020-2023, NVIDIA CORPORATION.
# Copyright (c) 2020-2024, NVIDIA CORPORATION.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
Expand Down Expand Up @@ -315,12 +315,6 @@ def bench_bfs(gpubenchmark, graph):
gpubenchmark(bfs, graph, start)


def bench_force_atlas2(gpubenchmark, graph):
if is_graph_distributed(graph):
pytest.skip("distributed graphs are not supported")
gpubenchmark(cugraph.force_atlas2, graph, max_iter=50)


def bench_sssp(gpubenchmark, graph):
sssp = dask_cugraph.sssp if is_graph_distributed(graph) else cugraph.sssp
start = graph.edgelist.edgelist_df["src"][0]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,9 @@
from cugraph.datasets import karate, dolphins


pytestmark = pytest.mark.skip(reason="Skipping until MG bug is resolved")


# =============================================================================
# Pytest Setup / Teardown - called for each test function
# =============================================================================
Expand Down

0 comments on commit c1f16fa

Please sign in to comment.