[GR-34614] Fixes for PyTorch

PullRequest: graalpython/2021

Author: msimacek

graalpython/com.oracle.graal.python.cext/src/capi.c

@@ -172,7 +172,6 @@ declare_type(PyList_Type, list, PyListObject);
 declare_type(PyComplex_Type, complex, PyComplexObject);
 declare_type(PyModule_Type, module, PyModuleObject);
 declare_type(PyModuleDef_Type, moduledef, PyModuleDef);
-declare_type(PyCapsule_Type, PyCapsule, PyCapsule);
 declare_type(PyMemoryView_Type, memoryview, PyMemoryViewObject);
 declare_type(PySet_Type, set, PySetObject);
 declare_type(PyFloat_Type, float, PyFloatObject);
@@ -892,6 +891,13 @@ void register_native_slots(PyTypeObject* managed_class, PyGetSetDef* getsets, Py
     }
 }
 
+PyObject* truffle_create_datetime_capsule(void *object) {
+    if (PyType_Ready(&PyCapsule_Type) < 0) {
+        return NULL;
+    }
+    return PyCapsule_New(object, "datetime.datetime_CAPI", NULL);
+}
+
 int truffle_subclass_check(PyObject* type) {
     return PyType_FastSubclass(Py_TYPE(type), Py_TPFLAGS_TYPE_SUBCLASS);
 }

graalpython/com.oracle.graal.python.cext/src/capi.h

@@ -66,15 +66,6 @@ typedef struct {
     PyObject *md_name;  /* for logging purposes after md_dict is cleared */
 } PyModuleObject;
 
-// taken from CPython "Objects/capsule.c"
-typedef struct {
-    PyObject_HEAD
-    void *pointer;
-    const char *name;
-    void *context;
-    PyCapsule_Destructor destructor;
-} PyCapsule;
-
 typedef struct {
     PyObject_VAR_HEAD
     int readonly;