no so files

.gitignore

@@ -6,3 +6,4 @@ dist/
 .eggs/
 *.egg-info/
 .coverage
+*.so

aten/cpu/cluster.cpp

+#include <torch/torch.h>
+
+#include "graclus.cpp"
+#include "grid.cpp"
+
+PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
+  m.def("graclus", &graclus, "Graclus (CPU)");
+  m.def("grid", &grid, "Grid (CPU)");
+}

aten/cpu/graclus.cpp

@@ -65,7 +65,3 @@ at::Tensor graclus(at::Tensor row, at::Tensor col, int num_nodes) {
 
   return cluster;
 }
-
-PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
-  m.def("graclus", &graclus, "Graclus (CPU)");
-}

aten/cpu/grid.cpp

@@ -19,5 +19,3 @@ at::Tensor grid(at::Tensor pos, at::Tensor size, at::Tensor start,
 
   return cluster;
 }
-
-PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { m.def("grid", &grid, "Grid (CPU)"); }