min cluster

dcd88f5a · rusty1s · 4d72a05e · dcd88f5a
Commit dcd88f5a authored Apr 27, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

aten/cpu/cluster.cpp aten/cpu/cluster.cpp +2 -1

No files found.
--- a/aten/cpu/cluster.cpp
+++ b/aten/cpu/cluster.cpp
@@ -36,7 +36,8 @@ at::Tensor graclus(at::Tensor row, at::Tensor col, int64_t num_nodes) {
      for (d_idx = 0; d_idx < deg_data[r]; d_idx++) {
        c = col_data[e_idx + d_idx];
        if (cluster_data[c] < 0) {
-          cluster_data[c] = r;
+          cluster_data[r] = std::min(r, c);
+          cluster_data[c] = std::min(r, c);
          break;
        }
      }