aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/api.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/api.py')
-rw-r--r--ethosu/vela/api.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/ethosu/vela/api.py b/ethosu/vela/api.py
index 06de0d9a..0799ab1c 100644
--- a/ethosu/vela/api.py
+++ b/ethosu/vela/api.py
@@ -23,6 +23,10 @@ from typing import NamedTuple
from typing import Optional
from typing import Tuple
+API_version_major = 1
+API_version_minor = 0
+api_version = f"{API_version_major}.{API_version_minor}"
+
class NpuElementWiseOp(Enum):
"""
@@ -367,3 +371,13 @@ class NpuElementWiseOperation(NpuBlockOperation):
self.reversed_operands: bool = False
# Set to a tuple (scale, shift) for explicit rescale, else to None
self.rescale: Optional[Tuple] = None
+
+
+def npu_get_API_version():
+ """
+ Public facing API to get the API version
+ :return: int, the 16 most significant bits, corresponding to major version
+ the 16 least significant bits, corresponding to minor version
+ """
+ version = (API_version_major << 16) | (API_version_minor & 0xFFFF)
+ return version