torch-ext/torch_binding.cpp

#include <torch/extension.h>
#include "torch_binding.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.doc() = "Minkowski Question-Mark Singularity Spectrum CUDA kernel";
  m.def("compute_spectrum", &compute_spectrum,
        "Compute multifractal spectrum tau(q), alpha(q), f(alpha)",
        py::arg("A_max"), py::arg("N"));
}