From 3ba1a02bf29b91e9d7b3cca08110ca60f7a1bcfb Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 5 Sep 2022 20:27:34 -0500
Subject: [PATCH] Expose {SVMAllocation,PooledSVM}._queue
---
src/wrap_cl.hpp | 9 +++++++++
src/wrap_cl_part_2.cpp | 11 +++++++++++
src/wrap_mempool.cpp | 20 ++++++++++++++++++++
3 files changed, 40 insertions(+)
diff --git a/src/wrap_cl.hpp b/src/wrap_cl.hpp
index d013519d..dc6a3684 100644
--- a/src/wrap_cl.hpp
+++ b/src/wrap_cl.hpp
@@ -3783,6 +3783,15 @@ namespace pyopencl
m_queue.reset();
}
+
+ // only use for testing/diagnostic/debugging purposes!
+ cl_command_queue queue() const
+ {
+ if (m_queue.is_valid())
+ return m_queue.data();
+ else
+ return nullptr;
+ }
};
// }}}
diff --git a/src/wrap_cl_part_2.cpp b/src/wrap_cl_part_2.cpp
index 33cc6ce3..80560bd7 100644
--- a/src/wrap_cl_part_2.cpp
+++ b/src/wrap_cl_part_2.cpp
@@ -408,6 +408,17 @@ void pyopencl_expose_part_2(py::module &m)
.def("bind_to_queue", &cls::bind_to_queue,
py::arg("queue"))
.DEF_SIMPLE_METHOD(unbind_from_queue)
+
+ // only for diagnostic/debugging/testing purposes!
+ .def_property_readonly("_queue",
+ [](cls const &self) -> py::object
+ {
+ cl_command_queue queue = self.queue();
+ if (queue)
+ return py::cast(new command_queue(queue, true));
+ else
+ return py::none();
+ })
;
}
diff --git a/src/wrap_mempool.cpp b/src/wrap_mempool.cpp
index 46845594..36f6b410 100644
--- a/src/wrap_mempool.cpp
+++ b/src/wrap_mempool.cpp
@@ -488,6 +488,15 @@ namespace pyopencl {
m_ptr.queue.reset();
}
+
+ // only use for testing/diagnostic/debugging purposes!
+ cl_command_queue queue() const
+ {
+ if (m_ptr.queue.is_valid())
+ return m_ptr.queue.data();
+ else
+ return nullptr;
+ }
};
// }}}
@@ -677,6 +686,17 @@ void pyopencl_expose_mempool(py::module &m)
.def("__hash__", [](cls &self) { return (intptr_t) self.svm_ptr(); })
.DEF_SIMPLE_METHOD(bind_to_queue)
.DEF_SIMPLE_METHOD(unbind_from_queue)
+
+ // only for diagnostic/debugging/testing purposes!
+ .def_property_readonly("_queue",
+ [](cls const &self) -> py::object
+ {
+ cl_command_queue queue = self.queue();
+ if (queue)
+ return py::cast(new pyopencl::command_queue(queue, true));
+ else
+ return py::none();
+ })
;
}
--
GitLab