Skip to content

Commit

Permalink
flake8 fix
Browse files Browse the repository at this point in the history
  • Loading branch information
tvercaut committed Sep 26, 2024
1 parent 5a1693c commit 7d2d79a
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion torchsparsegradutils/indexed_matmul.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@ def segment_mm(a, b, seglen_a):
if not a.shape[1] == D1 or not seglen_a.shape[0] == R:
raise ValueError("Incompatible size for inputs")

torchdevice = a.device
segidx_a = torch.cumsum(seglen_a[:-1], dim=0)

# Ideally the conversions below to nested tensor would be handled natively
Expand Down

0 comments on commit 7d2d79a

Please sign in to comment.