mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
82e353fffc
As PyTorch can now use C++17 language features Pull Request resolved: https://github.com/pytorch/pytorch/pull/110618 Approved by: https://github.com/soulitzer
38 lines
1.1 KiB
C++
38 lines
1.1 KiB
C++
#include <torch/csrc/autograd/generated/python_functions.h>
|
|
|
|
// ${generated_comment}
|
|
|
|
#include <Python.h>
|
|
#include <ATen/ATen.h>
|
|
|
|
#include <c10/core/SymNodeImpl.h>
|
|
#include "torch/csrc/autograd/generated/Functions.h"
|
|
#include "torch/csrc/autograd/python_cpp_function.h"
|
|
#include <torch/csrc/autograd/python_variable.h>
|
|
#include <torch/csrc/autograd/saved_variable.h>
|
|
#include <torch/csrc/utils/pybind.h>
|
|
#include <pybind11/pybind11.h>
|
|
#include <torch/csrc/utils/pybind.h>
|
|
|
|
// NOTE: See [Sharded File] comment in VariableType
|
|
|
|
namespace torch::autograd::generated {
|
|
|
|
template<typename C>
|
|
static void addClass(PyObject* module, PyTypeObject& type, const char* name,
|
|
PyGetSetDef* function_properties=NULL, PyMethodDef* function_methods=NULL)
|
|
{
|
|
_initFunctionPyTypeObject(type, name, function_properties, function_methods);
|
|
Py_INCREF(&type);
|
|
PyModule_AddObject(module, name, (PyObject*)&type);
|
|
registerCppFunction(typeid(C), &type);
|
|
}
|
|
|
|
${py_function_props_and_getters}
|
|
|
|
void initialize_autogenerated_functions${shard_id}(PyObject* module) {
|
|
${py_function_initializers}
|
|
}
|
|
|
|
} // namespace torch::autograd::generated
|