Function PYBIND11_MODULE

simple-knn/ext.cpp:15–17

#include "spatial.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.def("distCUDA2", &distCUDA2);
}