Skip to content

GitLab

Commit c2350bb6 authored by rusty1s's avatar rusty1s
Browse files

bugfix

parent cca4e78e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
benchmark/ptr2ind.py
View file @ c2350bb6
...@@ -62,7 +62,7 @@ def bucketize(matrix): ...@@ -62,7 +62,7 @@ def bucketize(matrix):
def convert_coo_to_csr(matrix): def convert_coo_to_csr(matrix):
row_indices = matrix.indices()[0] row_indices = matrix.indices()[0]
return torch._convert_coo_to_csr(row_indices, matrix.size(0) + 1) return torch._convert_coo_to_csr(row_indices, matrix.size(0))
for device in ['cpu', 'cuda']: for device in ['cpu', 'cuda']:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment