diff options
Diffstat (limited to 'driver_library/python/src/ethosu_driver/swig/driver.i')
-rw-r--r-- | driver_library/python/src/ethosu_driver/swig/driver.i | 10 |
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", " |