Skip to content

fixed nnz value#17

Merged
migalkin merged 1 commit intomainfrom
rspmm_fix
Mar 30, 2024
Merged

fixed nnz value#17
migalkin merged 1 commit intomainfrom
rspmm_fix

Conversation

@migalkin
Copy link
Copy Markdown
Collaborator

Fixes #16

The nnz value was read from the edge_index incorrectly which led to incomplete aggregation for the very last node in the adjacency matrix.
This PR fixes both CPU and CUDA kernels

@migalkin migalkin merged commit c414f83 into main Mar 30, 2024
@migalkin migalkin deleted the rspmm_fix branch March 30, 2024 02:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Why is nnz set as edge_index.size(0) in rspmm.cpp?

1 participant