slots for Image, SVM

inducer · Oct 5, 2024 · 02f6918 · 02f6918
1 parent f97160e
commit 02f6918
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/src/wrap_cl_part_2.cpp b/src/wrap_cl_part_2.cpp
@@ -32,7 +32,11 @@
 
 
 
-void *dummy_fn() {}
+void dummy_fn() {}
+PyType_Slot slots[] = {
+      {Py_tp_init, (void*) dummy_fn},
+      { 0, nullptr }
+  };
 
 namespace pyopencl {
 #if PYOPENCL_CL_VERSION >= 0x1020
@@ -100,7 +104,7 @@ void pyopencl_expose_part_2(py::module_ &m)
 
   {
     typedef image cls;
-    py::class_<cls, memory_object>(m, "Image", py::dynamic_attr())
+    py::class_<cls, memory_object>(m, "Image", py::dynamic_attr(), py::type_slots(slots))
       .def(
           "__init__",
           [](
@@ -348,7 +352,7 @@ void pyopencl_expose_part_2(py::module_ &m)
 
   {
     typedef svm_arg_wrapper cls;
-    py::class_<cls, svm_pointer>(m, "SVM", py::dynamic_attr())
+    py::class_<cls, svm_pointer>(m, "SVM", py::dynamic_attr(), py::type_slots(slots))
       .def(py::init<py::object>())
       ;
   }
@@ -536,11 +540,6 @@ void pyopencl_expose_part_2(py::module_ &m)
 
   // {{{ kernel
 
-  PyType_Slot slots[] = {
-      {Py_tp_init, (void*) dummy_fn},
-      { 0, nullptr }
-  };
-
   {
     typedef kernel cls;
     py::class_<cls>(m, "Kernel", py::dynamic_attr(), py::type_slots(slots))