# 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()