Merge pull request #32 from hjlin0515/master

fix typos in weighted graclus cuda

Merge pull request #32 from hjlin0515/master
fix typos in weighted graclus cuda
bd3ae685 · Matthias Fey · GitHub · 65a43ac7 · b3c689bc · bd3ae685
Unverified Commit bd3ae685 authored Aug 29, 2019 by Matthias Fey Committed by GitHub Aug 29, 2019
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) {