commit e87446c9ce7f6cbefcd7bbaff324eebdca10687e
author Mikael Olsson <mikael.olsson@arm.com> Fri Dec 15 17:17:06 2023 +0100
committer Mikael Olsson <mikael.olsson@arm.com> Wed Dec 20 12:05:37 2023 +0100
tree 2c3911d181afc7ece79398a1b6afaf15483192ba
parent 5087ba6f9821cae6e5a2843b33368771adfa687d

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>

driver_library/python/src/ethosu_driver/swig/driver.i

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 @@
     Returns:
         list: PMU event data
     ") getPmuCounters;
-    const std::vector<uint32_t> getPmuCounters();
+    const std::vector<uint64_t> getPmuCounters();
 
     %feature("docstring",
     "

driver_library/python/swig_generate.py

diff --git a/driver_library/python/swig_generate.py b/driver_library/python/swig_generate.py
index bdd43a3..8394d96 100755
--- a/driver_library/python/swig_generate.py
+++ b/driver_library/python/swig_generate.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright 2021-2022 Arm Limited and/or its affiliates <open-source-office@arm.com>
+# SPDX-FileCopyrightText: Copyright 2021-2023 Arm Limited and/or its affiliates <open-source-office@arm.com>
 # SPDX-License-Identifier: Apache-2.0
 """
 This script executes SWIG commands to generate C++ library wrappers.
@@ -13,6 +13,7 @@
     print('Generating wrappers for {}'.format(name))
     subprocess.check_output("swig -v -c++ -python" +
                             " -Wall" +
+                            " -DSWIGWORDSIZE64 " + # Force 64-bit word size for uint64_t vector to work
                             " -o {}/src/ethosu_driver/_generated/{}_wrap.cpp ".format(__current_dir, name) +
                             "-outdir {}/src/ethosu_driver/_generated ".format(__current_dir) +
                             "{} ".format(extr_includes) +