Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor test utility functions and update SG & MG core_number tests to test with edge masking #4253

Merged
merged 11 commits into from
Mar 19, 2024

Conversation

seunghwak
Copy link
Contributor

We previously had test_utilities.hpp having all different sorts of function declarations and property_generator.cuh (which is necessary to create an edge mask) was only for primitives testing.

This PR divides test_utilities.hpp to multiple files (check_utilities.hpp, conversion_utilities.hpp, csv_file_utilities.hpp, matrix_market_file_utilities.hpp, and misc_utilties.hpp) renamed/moved tests/prims/property_generator.cuh to tests/utilities/property_generator_utilities.hpp (now this file can be included from .cpp files).

Updated SG & MG core_number tests to use property generator to create an edge mask.

@seunghwak seunghwak self-assigned this Mar 15, 2024
@seunghwak seunghwak added improvement Improvement / enhancement to an existing function DO NOT MERGE Hold off on merging; see PR for details non-breaking Non-breaking change and removed cuGraph CMake labels Mar 15, 2024
@seunghwak seunghwak marked this pull request as ready for review March 15, 2024 21:14
@seunghwak seunghwak requested review from a team as code owners March 15, 2024 21:14
@seunghwak seunghwak removed the DO NOT MERGE Hold off on merging; see PR for details label Mar 15, 2024
Copy link
Contributor

@naimnv naimnv left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@ChuckHastings
Copy link
Collaborator

/merge

@rapids-bot rapids-bot bot merged commit d6fa29e into rapidsai:branch-24.04 Mar 19, 2024
137 checks passed
rapids-bot bot pushed a commit that referenced this pull request Apr 8, 2024
…gle counting & K-hop neighbor tests to include edge masked graph view inputs (#4261)

This PR pulls updates from #4253
Better be reviewed after #4253 is merged.

* Update triangle count to internally use edge masking (this is cheaper than creating a new graph in both memory footprint and compute) to
  * include only the edges with endpoints reachable within two hops (if the list of vertices to compute triangle counts is provided)
  * Exclude self-loops
  * Exclude vertices that do not belong to 2-Core
  * Note that we are still creating a new graph when we are creating a degree ordered directed graph (as this graph is now significantly smaller than the original graph and edge masking has non-negligible overhead and computing triangle counts on the final degree ordered directed graph is computationally intensive).
* Update SG/MG Triangle counting & K-hop neighbors C++ tests to include edge masked graph view inputs

Authors:
  - Seunghwa Kang (https://github.com/seunghwak)

Approvers:
  - Joseph Nke (https://github.com/jnke2016)
  - Naim (https://github.com/naimnv)
  - Chuck Hastings (https://github.com/ChuckHastings)

URL: #4261
@seunghwak seunghwak deleted the enh_mask_test_utils branch May 22, 2024 17:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CMake cuGraph improvement Improvement / enhancement to an existing function non-breaking Non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants