diff options
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() |