__init__.py Source File

Mobilint SDK qb Compiler v1.1 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 OptqConfig,

11 ModConfig,

12 LlmConfig,

13 EquivalentTransformationConfig,

14 SearchWeightScaleConfig,

15 RuntimeOptions,

16 SaveSampleConfig,

17 SCHEMA_VERSION,

18)

19from .presets import Preset, get_preset, list_presets

20from .enums import (

21 Device,

22 InferenceScheme,

23 LutClusteringMethod,

24 Method,

25 QuantMethod,

26 QuantMode,

27 QuantOutput,

28 SearchType,

29)

30

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

32from . import compat # noqa: F401

33

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

35# For new code, use CalibrationConfig and BitConfig directly

36QuantizationConfig = CalibrationConfig # Alias for backward compatibility

37

38

39# Factory function for backward compatibility

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

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

42

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

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

45 """

46 return CompileConfig(**kwargs)

47

48

49__all__ = [

50 # Main config

51 "CompileConfig",

52 "get_compile_config",

53 # Sub-configs

54 "Uint8InputConfig",

55 "PreprocessingConfig",

56 "ResourceManagementConfig",

57 "CalibrationConfig",

58 "BitConfig",

59 "OptqConfig",

60 "ModConfig",

61 "LlmConfig",

62 "EquivalentTransformationConfig",

63 "SearchWeightScaleConfig",

64 "RuntimeOptions",

65 "SaveSampleConfig",

66 # Legacy alias

67 "QuantizationConfig",

68 # Presets

69 "Preset",

70 "get_preset",

71 "list_presets",

72 # Enums

73 "Device",

74 "InferenceScheme",

75 "LutClusteringMethod",

76 "Method",

77 "QuantMethod",

78 "QuantMode",

79 "QuantOutput",

80 "SearchType",

81 # Version

82 "SCHEMA_VERSION",

83]

qbcompiler.configs.models.CompileConfig

Unified compilation configuration for Mobilint MXQ compilation.

Definition models.py:1435

qbcompiler.configs.get_compile_config

CompileConfig get_compile_config(**kwargs)

Create a CompileConfig with the given parameters.

Definition __init__.py:40

---

Generated by  1.14.0