__init__.py Source File

Mobilint SDK qb Compiler v1.4 MCS002-EN

• src
• qbcompiler
• configs

__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 HessianQuantConfig,

11 BiasCorrectionConfig,

12 ModConfig,

13 LlmConfig,

14 MoeConfig,

15 EquivalentTransformationConfig,

16 SearchWeightScaleConfig,

17 LoadScaleConfig,

18 RuntimeOptions,

19 SaveSampleConfig,

20 ExtraOutputConfig,

21 GroupWiseConfig,

22 SCHEMA_VERSION,

23)

24from .presets import Preset, get_preset, list_presets

25from .enums import (

26 Device,

27 HessianDtype,

28 InferenceScheme,

29 Method,

30 MoeExpertSelectionMode,

31 QuantMethod,

32 QuantMode,

33 QuantOutput,

34 SearchType,

35 Solver,

36)

37

38# Import compat to patch models with to_cpp_dict() methods

39from . import compat # noqa: F401

40

41# Legacy compatibility: QuantizationConfig is now split into CalibrationConfig + BitConfig

42# For new code, use CalibrationConfig and BitConfig directly

43QuantizationConfig = CalibrationConfig # Alias for backward compatibility

44

45

46# Factory function for backward compatibility

47def get_compile_config(**kwargs) -> CompileConfig:

48 """Create a CompileConfig with the given parameters.

49

50 This is a compatibility wrapper. For new code, use CompileConfig directly

51 or CompileConfig.from_preset() for preset-based configuration.

52 """

53 return CompileConfig(**kwargs)

54

55

56__all__ = [

57 # Main config

58 "CompileConfig",

59 "get_compile_config",

60 # Sub-configs

61 "Uint8InputConfig",

62 "PreprocessingConfig",

63 "ResourceManagementConfig",

64 "CalibrationConfig",

65 "BitConfig",

66 "HessianQuantConfig",

67 "BiasCorrectionConfig",

68 "ModConfig",

69 "LlmConfig",

70 "MoeConfig",

71 "EquivalentTransformationConfig",

72 "SearchWeightScaleConfig",

73 "LoadScaleConfig",

74 "RuntimeOptions",

75 "SaveSampleConfig",

76 "ExtraOutputConfig",

77 "GroupWiseConfig",

78 # Legacy alias

79 "QuantizationConfig",

80 # Presets

81 "Preset",

82 "get_preset",

83 "list_presets",

84 # Enums

85 "Device",

86 "HessianDtype",

87 "InferenceScheme",

88 "Method",

89 "MoeExpertSelectionMode",

90 "QuantMethod",

91 "QuantMode",

92 "QuantOutput",

93 "SearchType",

94 "Solver",

95 # Version

96 "SCHEMA_VERSION",

97]

qbcompiler.configs.models.CompileConfig

Unified compilation configuration for Mobilint MXQ compilation.

Definition models.py:1898

qbcompiler.configs.get_compile_config

CompileConfig get_compile_config(**kwargs)

Create a CompileConfig with the given parameters.

Definition __init__.py:47

---

Generated by  1.14.0