blob: 9aa1cebf32fb9b98509ea0556459632bab832da5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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()
|
