inverse if

0b790779 · rusty1s · f6aaacc1 · 0b790779
Commit 0b790779 authored Mar 21, 2019 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

torch_sparse/transpose.py torch_sparse/transpose.py +3 -3

No files found.
--- a/torch_sparse/transpose.py
+++ b/torch_sparse/transpose.py
@@ -35,10 +35,10 @@ def transpose_matrix(index, value, m, n):

    assert value.dim() == 1

-    if not index.is_cuda:
+    if index.is_cuda:
+        return transpose(index, value, m, n)
+    else:
        mat = to_scipy(index, value, m, n).tocsc()
        (col, row), value = from_scipy(mat)
        index = torch.stack([row, col], dim=0)
        return index, value
-    else:
-        return transpose(index, value, m, n)