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

Update API for node2vec and biased random walks #4841

Open
wants to merge 64 commits into
base: branch-25.02
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
64 commits
Select commit Hold shift + click to select a range
86b2038
add support for rng state
jnke2016 Dec 28, 2024
38690a6
update test to take rng state parameter
jnke2016 Dec 28, 2024
fd5b387
add support for rng state
jnke2016 Dec 28, 2024
9d56b5f
deprecate old API
jnke2016 Dec 28, 2024
615837e
add new API for node2vec random walks
jnke2016 Dec 28, 2024
21a76bb
add mg node2vec random walks to the python API
jnke2016 Dec 28, 2024
86a13d3
update docstrings
jnke2016 Dec 28, 2024
4c8744f
enable mg node2vec_random walks
jnke2016 Dec 28, 2024
4da1c7e
update argument list in function call
jnke2016 Dec 28, 2024
6984645
support optional weights
jnke2016 Dec 28, 2024
d04588a
update docstring and deprecate arguments
jnke2016 Dec 31, 2024
cb6a294
add new API for uniform_random_walks
jnke2016 Dec 31, 2024
7936026
deprecate method
jnke2016 Dec 31, 2024
7a5056f
update copyrights
jnke2016 Dec 31, 2024
e2e4694
add uniform random walks
jnke2016 Dec 31, 2024
877265b
add new API for node2vec random walks
jnke2016 Dec 31, 2024
bb77237
deprecate legacy implementation
jnke2016 Dec 31, 2024
618fe76
add random state argumment and update copyright
jnke2016 Dec 31, 2024
a1d004c
update header file to take as input a random state
jnke2016 Dec 31, 2024
bea2a2f
add support for rng state as input
jnke2016 Dec 31, 2024
755acc7
update tests to support rng state as input
jnke2016 Dec 31, 2024
ef00fa5
add biased random walks to the PLC API
jnke2016 Jan 1, 2025
ae4833c
add biased random walks to the python API
jnke2016 Jan 1, 2025
8314291
update docstrings and init file
jnke2016 Jan 1, 2025
1603bcd
fix typo
jnke2016 Jan 1, 2025
0a03b29
update copyright
jnke2016 Jan 1, 2025
88e405d
add mg implementation of biased and uniform random walks
jnke2016 Jan 1, 2025
c8265e7
update docstrings
jnke2016 Jan 2, 2025
0a4d29b
deprecate legacy implementation
jnke2016 Jan 2, 2025
4e0eff9
remove unused import
jnke2016 Jan 2, 2025
7c85269
update MG C tests
jnke2016 Jan 10, 2025
067d53b
remove unused variable and update the number of arrays passed at the …
jnke2016 Jan 10, 2025
b94a6ea
update copyright and remove debug print
jnke2016 Jan 11, 2025
e83722e
fix renumbering bug
jnke2016 Jan 11, 2025
2c1a034
enable MG tests and fix bugs
jnke2016 Jan 11, 2025
d521dfc
fix style
jnke2016 Jan 11, 2025
6418b96
remove unsued import
jnke2016 Jan 11, 2025
af3b31f
add type annotations
jnke2016 Jan 13, 2025
f0e3b0f
fix style
jnke2016 Jan 13, 2025
74648d4
deprecated old test suite
jnke2016 Jan 13, 2025
70b8d8b
add sg tests for uniform random walks
jnke2016 Jan 13, 2025
9b270c7
update copyright
jnke2016 Jan 13, 2025
9e850d2
update tests
jnke2016 Jan 14, 2025
10c471a
add support of multi column seeds
jnke2016 Jan 14, 2025
b94bad2
add support of multi column seeds for 'select_random_vertices'
jnke2016 Jan 14, 2025
af68c3f
add multi column tests
jnke2016 Jan 14, 2025
0c8f85e
add support of multi column seeds
jnke2016 Jan 14, 2025
37d3a47
add test for biased random walks
jnke2016 Jan 14, 2025
e7a5952
add mg ECG
jnke2016 Jan 15, 2025
dfe8dd2
update MG ecg implementation
jnke2016 Jan 16, 2025
5015e30
add mg ecg tests
jnke2016 Jan 16, 2025
92bfb57
pass different random seeds to each GPU
jnke2016 Jan 16, 2025
1282d52
add mg tests for biased random walks
jnke2016 Jan 16, 2025
79d7b5c
add node2vec mg tests
jnke2016 Jan 16, 2025
b22d4c8
add sg node2vec tests
jnke2016 Jan 16, 2025
c1f6ff0
deprecate tests
jnke2016 Jan 16, 2025
75e1253
fix style
jnke2016 Jan 16, 2025
f12bded
update copyright
jnke2016 Jan 16, 2025
b7effbb
Merge remote-tracking branch 'upstream/branch-25.02' into branch-25.0…
jnke2016 Jan 16, 2025
b7960b7
update pytest ini for random walks
jnke2016 Jan 18, 2025
ba015c4
fix style and update copyright
jnke2016 Jan 18, 2025
90bb3a0
Merge remote-tracking branch 'upstream/branch-25.02' into branch-25.0…
jnke2016 Jan 18, 2025
e4affce
update branch
jnke2016 Jan 20, 2025
2934ab2
Merge remote-tracking branch 'upstream/branch-25.02' into branch-25.0…
jnke2016 Jan 22, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
update header file to take as input a random state
jnke2016 committed Dec 31, 2024
commit a1d004c454db169a9a375dc4f43dfe1d2d059061
6 changes: 5 additions & 1 deletion cpp/include/cugraph_c/sampling_algorithms.h
Original file line number Diff line number Diff line change
@@ -41,6 +41,7 @@ typedef struct {
* @brief Compute uniform random walks
*
* @param [in] handle Handle for accessing resources
* @param [in,out] rng_state State of the random number generator, updated with each call
* @param [in] graph Pointer to graph. NOTE: Graph might be modified if the storage
* needs to be transposed
* @param [in] start_vertices Array of source vertices
@@ -52,6 +53,7 @@ typedef struct {
*/
cugraph_error_code_t cugraph_uniform_random_walks(
const cugraph_resource_handle_t* handle,
cugraph_rng_state_t* rng_state,
cugraph_graph_t* graph,
const cugraph_type_erased_device_array_view_t* start_vertices,
size_t max_length,
@@ -62,6 +64,7 @@ cugraph_error_code_t cugraph_uniform_random_walks(
* @brief Compute biased random walks
*
* @param [in] handle Handle for accessing resources
* @param [in,out] rng_state State of the random number generator, updated with each call
* @param [in] graph Pointer to graph. NOTE: Graph might be modified if the storage
* needs to be transposed
* @param [in] start_vertices Array of source vertices
@@ -73,6 +76,7 @@ cugraph_error_code_t cugraph_uniform_random_walks(
*/
cugraph_error_code_t cugraph_biased_random_walks(
const cugraph_resource_handle_t* handle,
cugraph_rng_state_t* rng_state,
cugraph_graph_t* graph,
const cugraph_type_erased_device_array_view_t* start_vertices,
size_t max_length,
@@ -83,7 +87,7 @@ cugraph_error_code_t cugraph_biased_random_walks(
* @brief Compute random walks using the node2vec framework.
*
* @param [in] handle Handle for accessing resources
* @param [in,out] rng_state State of the random number generator, updated with each call
* @param [in,out] rng_state State of the random number generator, updated with each call
* @param [in] graph Pointer to graph. NOTE: Graph might be modified if the storage
* needs to be transposed
* @param [in] start_vertices Array of source vertices