class-numbers-cuda / torch-ext /torch_binding.cpp

Add torch library bindings (guard main, add torch wrapper functions)

a51a509 verified 6 months ago

334 Bytes

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

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.doc() = "Class Numbers of Real Quadratic Fields CUDA kernel";
  m.def("compute_class_numbers", &compute_class_numbers_torch,
        "Compute class numbers h(d) for fundamental discriminants",
        py::arg("discriminants"));
}

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

	PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
	m.doc() = "Class Numbers of Real Quadratic Fields CUDA kernel";
	m.def("compute_class_numbers", &compute_class_numbers_torch,
	"Compute class numbers h(d) for fundamental discriminants",
	py::arg("discriminants"));
	}