1#include <torch/csrc/dynamo/init.h>
2
3#include <torch/csrc/Exceptions.h>
4#include <torch/csrc/dynamo/eval_frame.h>
5#include <torch/csrc/dynamo/guards.h>
6
7static struct PyModuleDef _module =
8 {PyModuleDef_HEAD_INIT, "torch._C._dynamo", "", -1, NULL};
9
10namespace torch {
11namespace dynamo {
12
13void initDynamoBindings(PyObject* torch) {
14 PyObject* dynamo = PyModule_Create(&_module);
15 if (dynamo == NULL || PyModule_AddObject(torch, "_dynamo", dynamo) != 0) {
16 throw python_error();
17 }
18
19 PyObject* eval_frame = torch_c_dynamo_eval_frame_init();
20 if (eval_frame == NULL ||
21 PyModule_AddObject(dynamo, "eval_frame", eval_frame) != 0) {
22 throw python_error();
23 }
24
25 PyObject* guards = torch_c_dynamo_guards_init();
26 if (guards == NULL || PyModule_AddObject(dynamo, "guards", guards) != 0) {
27 throw python_error();
28 }
29}
30
31} // namespace dynamo
32} // namespace torch
33