aboutsummaryrefslogtreecommitdiff
path: root/driver_library/python/src/ethosu_driver/swig/driver.i
diff options
context:
space:
mode:
authorMikael Olsson <mikael.olsson@arm.com>2023-12-15 17:17:06 +0100
committerMikael Olsson <mikael.olsson@arm.com>2023-12-20 12:05:37 +0100
commite87446c9ce7f6cbefcd7bbaff324eebdca10687e (patch)
tree2c3911d181afc7ece79398a1b6afaf15483192ba /driver_library/python/src/ethosu_driver/swig/driver.i
parent5087ba6f9821cae6e5a2843b33368771adfa687d (diff)
downloadethos-u-linux-driver-stack-e87446c9ce7f6cbefcd7bbaff324eebdca10687e.tar.gz
Change PMU event counter values to use 64-bit
The PMU event counter value is an accumulation of 32-bit values during the inference and to ensure the total value fits in the rpmsg message and UAPI, the variable holding the value has been changed to 64-bit. The driver library, Python wrapper and inference runner have been changed accordingly to support the 64-bit values. Change-Id: I09a8e45eb75800c8a787f83abff5a3693148cc15 Signed-off-by: Mikael Olsson <mikael.olsson@arm.com>
Diffstat (limited to 'driver_library/python/src/ethosu_driver/swig/driver.i')
-rw-r--r--driver_library/python/src/ethosu_driver/swig/driver.i10
1 files changed, 9 insertions, 1 deletions
diff --git a/driver_library/python/src/ethosu_driver/swig/driver.i b/driver_library/python/src/ethosu_driver/swig/driver.i
index 6e0ad25..a8db7c1 100644
--- a/driver_library/python/src/ethosu_driver/swig/driver.i
+++ b/driver_library/python/src/ethosu_driver/swig/driver.i
@@ -29,6 +29,14 @@
%shared_ptr(EthosU::Buffer);
%shared_ptr(EthosU::Network);
+%typemap(out) (std::vector<uint64_t>) {
+ PyObject *list = PyList_New($1.size());
+ for (size_t i=0; i < $1.size(); ++i) {
+ PyList_SET_ITEM(list, i, PyLong_FromUnsignedLong($1.at(i)));
+ }
+ $result = list;
+}
+
namespace std {
%template(UintVector) vector<unsigned int>;
%template(SizeTVector) vector<size_t>;
@@ -508,7 +516,7 @@ public:
Returns:
list: PMU event data
") getPmuCounters;
- const std::vector<uint32_t> getPmuCounters();
+ const std::vector<uint64_t> getPmuCounters();
%feature("docstring",
"