__init__.py Source File

Mobilint SDK qb Compiler v1.3 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 LayerBiasCorrectionConfig,

12 ModConfig,

13 LlmConfig,

14 MoeConfig,

15 EquivalentTransformationConfig,

16 SearchWeightScaleConfig,

17 LoadScaleConfig,

18 RuntimeOptions,

19 SaveSampleConfig,

20 GroupWiseConfig,

21 SCHEMA_VERSION,

22)

23from .presets import Preset, get_preset, list_presets

24from .enums import (

25 AccumulationDevice,

26 Device,

27 HessianDtype,

28 InferenceScheme,

29 Method,

30 MoeExpertSelectionMode,

31 QuantMethod,

32 QuantMode,

33 QuantOutput,

34 SearchType,

35)

36

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

38from . import compat # noqa: F401

39

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

41# For new code, use CalibrationConfig and BitConfig directly

42QuantizationConfig = CalibrationConfig # Alias for backward compatibility

43

44

45# Factory function for backward compatibility

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

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

48

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

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

51 """

52 return CompileConfig(**kwargs)

53

54

55__all__ = [

56 # Main config

57 "CompileConfig",

58 "get_compile_config",

59 # Sub-configs

60 "Uint8InputConfig",

61 "PreprocessingConfig",

62 "ResourceManagementConfig",

63 "CalibrationConfig",

64 "BitConfig",

65 "HessianQuantConfig",

66 "LayerBiasCorrectionConfig",

67 "ModConfig",

68 "LlmConfig",

69 "MoeConfig",

70 "EquivalentTransformationConfig",

71 "SearchWeightScaleConfig",

72 "LoadScaleConfig",

73 "RuntimeOptions",

74 "SaveSampleConfig",

75 "GroupWiseConfig",

76 # Legacy alias

77 "QuantizationConfig",

78 # Presets

79 "Preset",

80 "get_preset",

81 "list_presets",

82 # Enums

83 "AccumulationDevice",

84 "Device",

85 "HessianDtype",

86 "InferenceScheme",

87 "Method",

88 "MoeExpertSelectionMode",

89 "QuantMethod",

90 "QuantMode",

91 "QuantOutput",

92 "SearchType",

93 # Version

94 "SCHEMA_VERSION",

95]

qbcompiler.configs.models.CompileConfig

Unified compilation configuration for Mobilint MXQ compilation.

Definition models.py:1848

qbcompiler.configs.get_compile_config

CompileConfig get_compile_config(**kwargs)

Create a CompileConfig with the given parameters.

Definition __init__.py:46

---

Generated by  1.14.0