diff options
Diffstat (limited to 'driver_library/python/src')
| -rw-r--r-- | driver_library/python/src/ethosu_driver/__init__.py | 5 | ||||
| -rw-r--r-- | driver_library/python/src/ethosu_driver/swig/driver.i | 15 |
2 files changed, 16 insertions, 4 deletions
diff --git a/driver_library/python/src/ethosu_driver/__init__.py b/driver_library/python/src/ethosu_driver/__init__.py index ee6ea1f..a804c13 100644 --- a/driver_library/python/src/ethosu_driver/__init__.py +++ b/driver_library/python/src/ethosu_driver/__init__.py @@ -1,6 +1,7 @@ -# 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 -from ._generated.driver import Device, Inference, Network, Buffer +from ._generated.driver import Device, Inference, Network, Buffer, \ + MAX_SUPPORTED_KERNEL_DRIVER_MAJOR_VERSION, MIN_SUPPORTED_KERNEL_DRIVER_MAJOR_VERSION from ._utilities import open_device, load_model, populate_buffers, \ allocate_buffers, get_results, InferenceRunner diff --git a/driver_library/python/src/ethosu_driver/swig/driver.i b/driver_library/python/src/ethosu_driver/swig/driver.i index 4cd8bdf..558c22e 100644 --- a/driver_library/python/src/ethosu_driver/swig/driver.i +++ b/driver_library/python/src/ethosu_driver/swig/driver.i @@ -1,5 +1,5 @@ // -// SPDX-FileCopyrightText: Copyright 2020, 2022 Arm Limited and/or its affiliates <open-source-office@arm.com> +// SPDX-FileCopyrightText: Copyright 2020, 2022-2023 Arm Limited and/or its affiliates <open-source-office@arm.com> // SPDX-License-Identifier: Apache-2.0 // %module driver @@ -29,7 +29,6 @@ %shared_ptr(EthosU::Buffer); %shared_ptr(EthosU::Network); - namespace std { %template(UintVector) vector<unsigned int>; %template(SizeTVector) vector<size_t>; @@ -39,6 +38,9 @@ namespace std { namespace EthosU { +constexpr uint32_t MAX_SUPPORTED_KERNEL_DRIVER_MAJOR_VERSION; +constexpr uint32_t MIN_SUPPORTED_KERNEL_DRIVER_MAJOR_VERSION; + %feature("docstring", " Semantic Version : major.minor.patch @@ -166,6 +168,15 @@ public: Capabilities: Return capabilities of device. ") capabilities; Capabilities capabilities() const; + + %feature("docstring", + " + Returns kernel driver version information. + + Returns: + SemanticVersion: kernel driver version. + ") getDriverVersion; + const SemanticVersion &getDriverVersion() const; }; %extend Device { |