fix typos in weighted graclus cuda

b3c689bc · Haojie Lin · 65a43ac7 · b3c689bc
Commit b3c689bc authored Aug 29, 2019 by Haojie Lin
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

cuda/graclus.cpp cuda/graclus.cpp +1 -1

No files found.
--- 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) {