1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
#include <torch/csrc/python_headers.h>
#include <ATen/Utils.h>
#include <functional>
static PyObject* THPWrapperClass = nullptr;
struct THPWrapper {
PyObject_HEAD
void *data;
void (*destructor)(void*);
};
PyObject * THPWrapper_New(void *data, void (*destructor)(void*))
{
PyObject *args = PyTuple_New(0);
if (!args) {
return nullptr;
}
PyObject *result = PyObject_Call(THPWrapperClass, args, nullptr);
if (result) {
THPWrapper* wrapper = (THPWrapper*) result;
wrapper->data = data;
wrapper->destructor = destructor;
}
Py_DECREF(args);
return result;
}
bool THPWrapper_check(PyObject * obj)
{
return (PyObject*)Py_TYPE(obj) == THPWrapperClass;
}
void * THPWrapper_get(PyObject * obj)
{
return ((THPWrapper*)obj)->data;
}
static PyObject * THPWrapper_pynew(PyTypeObject *type, PyObject *args, PyObject *kwargs)
{
PyObject* self = type->tp_alloc(type, 0);
THPWrapper* wrapper = (THPWrapper*) self;
wrapper->data = nullptr;
wrapper->destructor = nullptr;
return self;
}
static void THPWrapper_dealloc(THPWrapper* self)
{
self->destructor(self->data);
Py_TYPE(self)->tp_free((PyObject*)self);
}
PyTypeObject THPWrapperType = {
PyVarObject_HEAD_INIT(nullptr, 0)
"torch._C._PtrWrapper", /* tp_name */
sizeof(THPWrapper), /* tp_basicsize */
0, /* tp_itemsize */
(destructor)THPWrapper_dealloc, /* tp_dealloc */
0, /* tp_vectorcall_offset */
nullptr, /* tp_getattr */
nullptr, /* tp_setattr */
nullptr, /* tp_reserved */
nullptr, /* tp_repr */
nullptr, /* tp_as_number */
nullptr, /* tp_as_sequence */
nullptr, /* tp_as_mapping */
nullptr, /* tp_hash */
nullptr, /* tp_call */
nullptr, /* tp_str */
nullptr, /* tp_getattro */
nullptr, /* tp_setattro */
nullptr, /* tp_as_buffer */
Py_TPFLAGS_DEFAULT, /* tp_flags */
nullptr, /* tp_doc */
nullptr, /* tp_traverse */
nullptr, /* tp_clear */
nullptr, /* tp_richcompare */
0, /* tp_weaklistoffset */
nullptr, /* tp_iter */
nullptr, /* tp_iternext */
nullptr, /* tp_methods */
nullptr, /* tp_members */
nullptr, /* tp_getset */
nullptr, /* tp_base */
nullptr, /* tp_dict */
nullptr, /* tp_descr_get */
nullptr, /* tp_descr_set */
0, /* tp_dictoffset */
nullptr, /* tp_init */
nullptr, /* tp_alloc */
THPWrapper_pynew, /* tp_new */
};
bool THPWrapper_init(PyObject *module)
{
THPWrapperClass = (PyObject*)&THPWrapperType;
if (PyType_Ready(&THPWrapperType) < 0)
return false;
Py_INCREF(&THPWrapperType);
return true;
}
|
