Skip to content

GitLab

"cacheflow/git@developer.sourcefind.cn:norm/vllm.git" did not exist on "aedba6d5ec4e1eaad10745f71970c10b601f9dc1"
Commit ae35b8a5 authored by rusty1s's avatar rusty1s
Browse files

remove dim_size

parent cdd1cb4d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
torch_sparse/storage.py
View file @ ae35b8a5
...@@ -269,8 +269,7 @@ class SparseStorage(object): ...@@ -269,8 +269,7 @@ class SparseStorage(object):
value = self.value value = self.value
if self.has_value(): if self.has_value():
idx = mask.cumsum(0) - 1 idx = mask.cumsum(0) - 1
dim_size = idx[-1].item() + 1 value = segment_csr(idx, value, reduce=reduce)
value = segment_csr(idx, value, dim_size=dim_size, reduce=reduce)
value = value[0] if isinstance(value, tuple) else value value = value[0] if isinstance(value, tuple) else value
return self.__class__(index, value, self.sparse_size(), is_sorted=True) return self.__class__(index, value, self.sparse_size(), is_sorted=True)
......
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