fix

ee08da1f · rusty1s · 288cfd44 · ee08da1f
Commit ee08da1f authored Nov 24, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/cpu/relabel_cpu.cpp csrc/cpu/relabel_cpu.cpp +1 -1

No files found.
--- a/csrc/cpu/relabel_cpu.cpp
+++ b/csrc/cpu/relabel_cpu.cpp
@@ -131,7 +131,7 @@ relabel_one_hop_cpu(torch::Tensor rowptr, torch::Tensor col,
    }
  }
-  if (bipartite)
+  if (!bipartite)
    out_rowptr = torch::cat(
        {out_rowptr, torch::full({(int64_t)n_ids.size()}, out_col.numel(),
                                 rowptr.options())});