revert

rusty1s · rusty1s · commit a418e03c4cc4 · 2022-04-22T18:04:11.000Z
diff --git a/csrc/cpu/neighbor_sample_cpu.cpp b/csrc/cpu/neighbor_sample_cpu.cpp
@@ -174,7 +174,7 @@ hetero_sample(const vector<node_t> &node_types,
     // root_time[i] stores the timestamp of the computation tree root
     // of the node samples[i]
     if (temporal) {
-      torch::Tensor &node_time = node_time_dict.at(node_type);
+      torch::Tensor node_time = node_time_dict.at(node_type);
       node_time_data = node_time.data_ptr<int64_t>();
     }
 

Original file line number	Diff line number	Diff line change
`@@ -174,7 +174,7 @@ hetero_sample(const vector<node_t> &node_types,`
`174`	`174`	`// root_time[i] stores the timestamp of the computation tree root`
`175`	`175`	`// of the node samples[i]`
`176`	`176`	`if (temporal) {`
`177`		`- torch::Tensor &node_time = node_time_dict.at(node_type);`
	`177`	`+ torch::Tensor node_time = node_time_dict.at(node_type);`
`178`	`178`	`node_time_data = node_time.data_ptr<int64_t>();`
`179`	`179`	`}`
`180`	`180`