diff options
author | Won Jeon <won.jeon@arm.com> | 2024-03-11 22:17:13 +0000 |
---|---|---|
committer | Won Jeon <won.jeon@arm.com> | 2024-04-18 18:31:24 +0000 |
commit | 3eec59b22128462eef668dbfb770c6924aeee40f (patch) | |
tree | 2d5e906705c12ce11b5e4ae2e4a65323b71db3af /verif/conformance/tosa_profiles.py | |
parent | b7f5eed148f515bc07e23ec8903d05dfe35dc8d7 (diff) | |
download | reference_model-3eec59b22128462eef668dbfb770c6924aeee40f.tar.gz |
Add extensions & profiles support to conformance generator
Support existing conformance profiles to ease transition
New combined config tosa_ext_profile_ops_info.json that supports
extension selection
Signed-off-by: Won Jeon <won.jeon@arm.com>
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ic04696a01d619d191b9c8abc4ef7f4e8b86c52ca
Diffstat (limited to 'verif/conformance/tosa_profiles.py')
-rw-r--r-- | verif/conformance/tosa_profiles.py | 34 |
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() |