__init__.py Source File

__init__.py Source File#

Mobilint SDK qb Compiler: __init__.py Source File
Mobilint SDK qb Compiler v1.2
MCS002-EN
__init__.py
1"""Auto-generated config module from config_schema.yaml."""
2
3from .models import (
4 CompileConfig,
5 Uint8InputConfig,
6 PreprocessingConfig,
7 ResourceManagementConfig,
8 CalibrationConfig,
9 BitConfig,
10 OptqConfig,
11 ModConfig,
12 LlmConfig,
13 MoeConfig,
14 EquivalentTransformationConfig,
15 SearchWeightScaleConfig,
16 LoadScaleConfig,
17 RuntimeOptions,
18 SaveSampleConfig,
19 SCHEMA_VERSION,
20)
21from .presets import Preset, get_preset, list_presets
22from .enums import (
23 Device,
24 InferenceScheme,
25 LutClusteringMethod,
26 Method,
27 MoeExpertSelectionMode,
28 QuantMethod,
29 QuantMode,
30 QuantOutput,
31 SearchType,
32)
33
34# Import compat to patch models with to_cpp_dict() methods
35from . import compat # noqa: F401
36
37# Legacy compatibility: QuantizationConfig is now split into CalibrationConfig + BitConfig
38# For new code, use CalibrationConfig and BitConfig directly
39QuantizationConfig = CalibrationConfig # Alias for backward compatibility
40
41
42# Factory function for backward compatibility
43def get_compile_config(**kwargs) -> CompileConfig:
44 """Create a CompileConfig with the given parameters.
45
46 This is a compatibility wrapper. For new code, use CompileConfig directly
47 or CompileConfig.from_preset() for preset-based configuration.
48 """
49 return CompileConfig(**kwargs)
50
51
52__all__ = [
53 # Main config
54 "CompileConfig",
55 "get_compile_config",
56 # Sub-configs
57 "Uint8InputConfig",
58 "PreprocessingConfig",
59 "ResourceManagementConfig",
60 "CalibrationConfig",
61 "BitConfig",
62 "OptqConfig",
63 "ModConfig",
64 "LlmConfig",
65 "MoeConfig",
66 "EquivalentTransformationConfig",
67 "SearchWeightScaleConfig",
68 "LoadScaleConfig",
69 "RuntimeOptions",
70 "SaveSampleConfig",
71 # Legacy alias
72 "QuantizationConfig",
73 # Presets
74 "Preset",
75 "get_preset",
76 "list_presets",
77 # Enums
78 "Device",
79 "InferenceScheme",
80 "LutClusteringMethod",
81 "Method",
82 "MoeExpertSelectionMode",
83 "QuantMethod",
84 "QuantMode",
85 "QuantOutput",
86 "SearchType",
87 # Version
88 "SCHEMA_VERSION",
89]
Unified compilation configuration for Mobilint MXQ compilation.
Definition models.py:1580
CompileConfig get_compile_config(**kwargs)
Create a CompileConfig with the given parameters.
Definition __init__.py:43