aboutsummaryrefslogtreecommitdiff
path: root/verif/conformance/tosa_profiles.py
diff options
context:
space:
mode:
Diffstat (limited to 'verif/conformance/tosa_profiles.py')
-rw-r--r--verif/conformance/tosa_profiles.py34
1 files changed, 34 insertions, 0 deletions
diff --git a/verif/conformance/tosa_profiles.py b/verif/conformance/tosa_profiles.py
new file mode 100644
index 0000000..9aa1ceb
--- /dev/null
+++ b/verif/conformance/tosa_profiles.py
@@ -0,0 +1,34 @@
+# Copyright (c) 2024, ARM Limited.
+# SPDX-License-Identifier: Apache-2.0
+
+
+class TosaProfiles:
+ TosaBI = "tosa-bi"
+ TosaMI = "tosa-mi"
+ TosaExtInt16 = "tosa-ext-int16"
+ TosaExtInt4 = "tosa-ext-int4"
+ TosaExtBF16 = "tosa-ext-bf16"
+ TosaExtFP8E4M3 = "tosa-ext-fp8e4m3"
+ TosaExtFP8E5M2 = "tosa-ext-fp8e5m2"
+ TosaExtFFT = "tosa-ext-fft"
+ TosaExtVariable = "tosa-ext-variable"
+
+ @staticmethod
+ def profiles():
+ return [TosaProfiles.TosaBI, TosaProfiles.TosaMI]
+
+ @staticmethod
+ def extensions():
+ return [
+ TosaProfiles.TosaExtInt16,
+ TosaProfiles.TosaExtInt4,
+ TosaProfiles.TosaExtBF16,
+ TosaProfiles.TosaExtFP8E4M3,
+ TosaProfiles.TosaExtFP8E5M2,
+ TosaProfiles.TosaExtFFT,
+ TosaProfiles.TosaExtVariable,
+ ]
+
+ @staticmethod
+ def all():
+ return TosaProfiles.profiles() + TosaProfiles.extensions()