提交 1a5c6c84 编写于 作者: R rusty1s

update

上级 0e18448b
......@@ -59,6 +59,8 @@ class SparseStorage(object):
sparse_sizes = (int(M), int(N))
else:
assert len(sparse_sizes) == 2
if row is not None:
assert row.max().item() < sparse_sizes[0]
assert col.max().item() < sparse_sizes[1]
if row is not None:
......@@ -67,7 +69,6 @@ class SparseStorage(object):
assert row.dim() == 1
assert row.numel() == col.numel()
row = row.contiguous()
assert row.max().item() < sparse_sizes[0]
if rowptr is not None:
assert rowptr.dtype == torch.long
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册