Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cluster.cpp 200 Bytes
Newer Older
rusty1s's avatar
no so files  
rusty1s committed
1
2
3
4
5
6
7
8
9 
#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)");
}