diff options
Diffstat (limited to 'ethosu/vela/api.py')
| -rw-r--r-- | ethosu/vela/api.py | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/ethosu/vela/api.py b/ethosu/vela/api.py index 589a283..7125e88 100644 --- a/ethosu/vela/api.py +++ b/ethosu/vela/api.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2020-2023 Arm Limited and/or its affiliates <open-source-office@arm.com> +# SPDX-FileCopyrightText: Copyright 2020-2024 Arm Limited and/or its affiliates <open-source-office@arm.com> # # SPDX-License-Identifier: Apache-2.0 # @@ -27,7 +27,7 @@ import numpy API_VERSION_MAJOR = 1 -API_VERSION_MINOR = 4 +API_VERSION_MINOR = 5 API_VERSION = f"{API_VERSION_MAJOR}.{API_VERSION_MINOR}" @@ -273,6 +273,16 @@ class NpuKernel: self.dilation_y = dilation_y +class NpuAccumulatorType(Enum): + """ + Accumulator dtype of NPU operation + """ + + Default = auto() + Int32 = auto() + Int40 = auto() + + class NpuOperationType(Enum): """ Type of NPU operation @@ -343,6 +353,7 @@ class NpuBlockOperation(NpuOperation): self.fused_quantize: bool = False # IFM upscaling to be applied self.ifm_upscale: NpuResamplingMode = NpuResamplingMode.NONE + self.accumulator_type: NpuAccumulatorType = NpuAccumulatorType.Default class NpuConv2DOperation(NpuBlockOperation): |