Merge pull request #32 from hjlin0515/master

rusty1s · web-flow · commit bd3ae685cdd0 · 2019-08-29T06:23:11.000+02:00
fix typos in weighted graclus cuda
diff --git a/cuda/graclus.cpp b/cuda/graclus.cpp
@@ -18,7 +18,7 @@ at::Tensor weighted_graclus(at::Tensor row, at::Tensor col, at::Tensor weight,
   CHECK_CUDA(row);
   CHECK_CUDA(col);
   CHECK_CUDA(weight);
-  return graclus_cuda(row, col, num_nodes);
+  return weighted_graclus_cuda(row, col, weight, num_nodes);
 }
 
 PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,7 @@ at::Tensor weighted_graclus(at::Tensor row, at::Tensor col, at::Tensor weight,`
`18`	`18`	`CHECK_CUDA(row);`
`19`	`19`	`CHECK_CUDA(col);`
`20`	`20`	`CHECK_CUDA(weight);`
`21`		`- return graclus_cuda(row, col, num_nodes);`
	`21`	`+ return weighted_graclus_cuda(row, col, weight, num_nodes);`
`22`	`22`	`}`
`23`	`23`
`24`	`24`	`PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {`