init (#263)

mananshah99 · web-flow · commit 03f2caa9c0a1 · 2022-07-28T02:03:37.000+02:00
diff --git 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_col_dict.insert(rel_type, from_vector<int64_t>(cols));
-      out_edge_dict.insert(rel_type, from_vector<int64_t>(edges));
-    }
-  }
+    out_row_dict.insert(rel_type, from_vector<int64_t>(rows));
+    out_col_dict.insert(rel_type, from_vector<int64_t>(cols));
+    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) {

Original file line number	Diff line number	Diff line change
`@@ -224,12 +224,10 @@ hgt_sample_cpu(const c10::Dict<rel_t, torch::Tensor> &colptr_dict,`
`224`	`224`	`}`
`225`	`225`	`}`
`226`	`226`	`}`
`227`		`- if (rows.size() > 0) {`
`228`		`- out_row_dict.insert(rel_type, from_vector<int64_t>(rows));`
`229`		`- out_col_dict.insert(rel_type, from_vector<int64_t>(cols));`
`230`		`- out_edge_dict.insert(rel_type, from_vector<int64_t>(edges));`
`231`		`- }`
`232`		`- }`
	`227`	`+ out_row_dict.insert(rel_type, from_vector<int64_t>(rows));`
	`228`	`+ out_col_dict.insert(rel_type, from_vector<int64_t>(cols));`
	`229`	`+ out_edge_dict.insert(rel_type, from_vector<int64_t>(edges));`
	`230`	`+}`
`233`	`231`
`234`	`232`	`// Generate tensor-valued output node dictionary (line 20):`
`235`	`233`	`for (const auto &kv : nodes_dict) {`