torch-ext/torch_binding.cpp

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

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.doc() = "Kronecker Coefficients (Symmetric Group) CUDA kernel";
  m.def("compute_kronecker", &compute_kronecker, py::arg("ct"), py::arg("z_inv"));
}