init (#263)

03f2caa9 · Manan Shah · GitHub · 27633e2e · 03f2caa9
Unverified Commit 03f2caa9 authored Jul 27, 2022 by Manan Shah Committed by GitHub Jul 28, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 6 deletions

csrc/cpu/hgt_sample_cpu.cpp csrc/cpu/hgt_sample_cpu.cpp +4 -6

No files found.
--- a/csrc/cpu/hgt_sample_cpu.cpp
+++ b/csrc/cpu/hgt_sample_cpu.cpp
@@ -224,12 +224,10 @@ hgt_sample_cpu(const c10::Dict<rel_t, torch::Tensor> &colptr_dict,
        }
      }
    }
-    if (rows.size() > 0) {
+    out_row_dict.insert(rel_type, from_vector<int64_t>(rows));
-      out_row_dict.insert(rel_type, from_vector<int64_t>(rows));
+    out_col_dict.insert(rel_type, from_vector<int64_t>(cols));
-      out_col_dict.insert(rel_type, from_vector<int64_t>(cols));
+    out_edge_dict.insert(rel_type, from_vector<int64_t>(edges));
-      out_edge_dict.insert(rel_type, from_vector<int64_t>(edges));
+}
-    }
-  }
  // Generate tensor-valued output node dictionary (line 20):
  for (const auto &kv : nodes_dict) {