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 | |
7 | static struct PyModuleDef _module = |
8 | {PyModuleDef_HEAD_INIT, "torch._C._dynamo" , "" , -1, NULL}; |
9 | |
10 | namespace torch { |
11 | namespace dynamo { |
12 | |
13 | void 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 | |