aboutsummaryrefslogtreecommitdiff
path: root/driver_library/python/src/ethosu_driver/__init__.py
diff options
context:
space:
mode:
authorMikael Olsson <mikael.olsson@arm.com>2023-06-12 15:00:55 +0200
committerMikael Olsson <mikael.olsson@arm.com>2023-08-09 15:11:11 +0200
commit308e7f1352bfdab8cc90be78a82b7ce4195301bc (patch)
treee64ebb166ff2cf63f5bb9e26ecc88314f859f8de /driver_library/python/src/ethosu_driver/__init__.py
parentf1cfe19a5fd6ccc07e6e86cbe5ab863f4b372418 (diff)
downloadethos-u-linux-driver-stack-308e7f1352bfdab8cc90be78a82b7ce4195301bc.tar.gz
Add kernel driver version check in driver library
The driver library will now check that it's compatible with the kernel driver in use by checking the kernel driver's version. The kernel driver version has also been made available to the library users and the Python wrapper has been updated accordingly. Change-Id: Ieae8c0bfc323f945038e7264eceeab90c833f76d Signed-off-by: Mikael Olsson <mikael.olsson@arm.com>
Diffstat (limited to 'driver_library/python/src/ethosu_driver/__init__.py')
-rw-r--r--driver_library/python/src/ethosu_driver/__init__.py5
1 files changed, 3 insertions, 2 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