#include <pybind11/pybind11.h>
#include <torch/extension.h>

namespace ascendc_ops {
at::Tensor run_add_custom(const at::Tensor &x, const at::Tensor &y);
at::Tensor run_trig_custom(const at::Tensor &x, const at::Tensor &out_sin, const at::Tensor &out_cos);
}

// expose Ascend custom ops to Python
PYBIND11_MODULE(custom_ops_lib, m)
{
    m.def("run_add_custom", &ascendc_ops::run_add_custom, "");
    m.def("run_trig_custom", &ascendc_ops::run_trig_custom, "");
}