enums.py Source File

enums.py Source File#

Mobilint SDK qb Compiler: enums.py Source File
Mobilint SDK qb Compiler v1.3
MCS002-EN
"""Auto-generated enum definitions from config_schema.yaml."""
 
from enum import Enum
 
 
class AccumulationDevice(str, Enum):
    auto = "auto"
    cpu = "cpu"
    gpu = "gpu"
 
    def __str__(self) -> str:
        return self.value
 
 
class Device(str, Enum):
    gpu = "gpu"
    cpu = "cpu"
 
    def __str__(self) -> str:
        return self.value
 
 
class HessianDtype(str, Enum):
    fp32 = "fp32"
    bf16 = "bf16"
 
    def __str__(self) -> str:
        return self.value
 
 
class InferenceScheme(str, Enum):
    single = "single"
    multi = "multi"
    _global = "global"
    global4 = "global4"
    global8 = "global8"
    all = "all"
 
    def __str__(self) -> str:
        return self.value
 
 
class Method(str, Enum):
    DeleteFloat = "DeleteFloat"
    SaveFloat = "SaveFloat"
    MoveFloat = "MoveFloat"
    KeepFloat = "KeepFloat"
    KeepAll = "KeepAll"
 
    def __str__(self) -> str:
        return self.value
 
 
class MoeExpertSelectionMode(str, Enum):
    TopK = "TopK"
    All = "All"
    Threshold = "Threshold"
 
    def __str__(self) -> str:
        return self.value
 
 
class QuantMethod(str, Enum):
    WChALayer = "WChALayer"
    WChAMulti = "WChAMulti"
    WChALayerZeropoint = "WChALayerZeropoint"
    WChAMultiZeropoint = "WChAMultiZeropoint"
 
    def __str__(self) -> str:
        return self.value
 
 
class QuantMode(str, Enum):
    Max = "Max"
    MaxPercentile = "MaxPercentile"
    Histogram = "Histogram"
 
    def __str__(self) -> str:
        return self.value
 
 
class QuantOutput(str, Enum):
    Layer = "Layer"
    Ch = "Ch"
    Sigmoid = "Sigmoid"
 
    def __str__(self) -> str:
        return self.value
 
 
class SearchType(str, Enum):
    Percentile = "Percentile"
    MSE = "MSE"
    KL = "KL"
 
    def __str__(self) -> str:
        return self.value