aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/data_type.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/data_type.py')
-rw-r--r--ethosu/vela/data_type.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/data_type.py b/ethosu/vela/data_type.py
index 829cef38..5d0320b2 100644
--- a/ethosu/vela/data_type.py
+++ b/ethosu/vela/data_type.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright 2020-2021 Arm Limited and/or its affiliates <open-source-office@arm.com>
+# SPDX-FileCopyrightText: Copyright 2020-2021, 2023 Arm Limited and/or its affiliates <open-source-office@arm.com>
#
# SPDX-License-Identifier: Apache-2.0
#
@@ -110,7 +110,7 @@ class DataType:
BaseType.Complex: "c",
}
assert self.type in numpy_dtype_code, f"Failed to interpret {self} as a numpy dtype"
- return np.dtype(numpy_dtype_code[self.type] + str(self.size_in_bytes()))
+ return np.dtype(numpy_dtype_code[self.type] + str(self.size_in_bytes())).type
stem_name = {
BaseType.UnsignedInt: ("uint%s", True),