From 7d2d79ab1d5b00c8bb38299cc7dac1023d4741fe Mon Sep 17 00:00:00 2001 From: Tom Vercauteren Date: Thu, 26 Sep 2024 22:20:07 +0100 Subject: [PATCH] flake8 fix --- torchsparsegradutils/indexed_matmul.py | 1 - 1 file changed, 1 deletion(-) diff --git a/torchsparsegradutils/indexed_matmul.py b/torchsparsegradutils/indexed_matmul.py index daa1a0d..298fd6b 100644 --- a/torchsparsegradutils/indexed_matmul.py +++ b/torchsparsegradutils/indexed_matmul.py @@ -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