min cluster

rusty1s · rusty1s · commit dcd88f5a2382 · 2018-04-27T08:20:55.000+02:00
diff --git 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;
         }
       }

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,8 @@ at::Tensor graclus(at::Tensor row, at::Tensor col, int64_t num_nodes) {`
`36`	`36`	`for (d_idx = 0; d_idx < deg_data[r]; d_idx++) {`
`37`	`37`	`c = col_data[e_idx + d_idx];`
`38`	`38`	`if (cluster_data[c] < 0) {`
`39`		`- cluster_data[c] = r;`
	`39`	`+ cluster_data[r] = std::min(r, c);`
	`40`	`+ cluster_data[c] = std::min(r, c);`
`40`	`41`	`break;`
`41`	`42`	`}`
`42`	`43`	`}`