removed branch

67d0cd48 · rusty1s · 157a8fe1 · 67d0cd48
Commit 67d0cd48 authored Mar 29, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 9 deletions

torch_cluster/src/serial_cpu.c torch_cluster/src/serial_cpu.c +4 -9

No files found.
--- a/torch_cluster/src/serial_cpu.c
+++ b/torch_cluster/src/serial_cpu.c
@@ -12,16 +12,11 @@
  while(e < THLongTensor_nElement(row)) { \
    row_value = row_data[e]; \
    if (output_data[row_value] < 0) { \
-      col_value = -1; \
+      col_value = row_value; \
      SELECT \
-      if (col_value < 0) { \
+      v = row_value < col_value ? row_value : col_value; \
-        output_data[row_value] = row_value; \
+      output_data[row_value] = v; \
-      } \
+      output_data[col_value] = v; \
-      else { \
-        v = row_value < col_value ? row_value : col_value; \
-        output_data[row_value] = v; \
-        output_data[col_value] = v; \
-      } \
    } \
    e += degree_data[row_value]; \
  } \