mct-nightly 2.1.0.20240529.431__py3-none-any.whl → 2.1.0.20240530.440__py3-none-any.whl

{mct_nightly-2.1.0.20240529.431.dist-info → mct_nightly-2.1.0.20240530.440.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 2.1.0.20240529.431
+Version: 2.1.0.20240530.440
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN

{mct_nightly-2.1.0.20240529.431.dist-info → mct_nightly-2.1.0.20240530.440.dist-info}/RECORD

@@ -1,4 +1,4 @@
-model_compression_toolkit/__init__.py,sha256=F0hTsPlZ83527863XchjFzothwgPnodI3rvD5OIysp0,1573
+model_compression_toolkit/__init__.py,sha256=Fu3xEc3yOO69H_m9_viK2IcJAgsYnhBYYNUGRoMKeUs,1573
 model_compression_toolkit/constants.py,sha256=b63Jk_bC7VXEX3Qn9TZ3wUvrNKD8Mkz8zIuayoyF5eU,3828
 model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
 model_compression_toolkit/logger.py,sha256=3DByV41XHRR3kLTJNbpaMmikL8icd9e1N-nkQAY9oDk,4567
@@ -417,7 +417,7 @@ model_compression_toolkit/target_platform_capabilities/immutable.py,sha256=YhROB
 model_compression_toolkit/target_platform_capabilities/target_platform/__init__.py,sha256=bD2xE101rRw7pGarGiBzYraeFF7FUnzyWtAVzwu5JT4,1724
 model_compression_toolkit/target_platform_capabilities/target_platform/current_tp_model.py,sha256=1Glr4qKDJfdk5TwM5fzZ12XzgbpQFioDOxb475905gk,2013
 model_compression_toolkit/target_platform_capabilities/target_platform/fusing.py,sha256=f3xBAI6ivPvEj4lw8cAvTKdIbs7CRdLAa_0LvhGw3Dg,3924
-model_compression_toolkit/target_platform_capabilities/target_platform/op_quantization_config.py,sha256=kk2c0XjeW3WbK0NihV1j_EffnnpfOXJEI4hJeu6amEc,14693
+model_compression_toolkit/target_platform_capabilities/target_platform/op_quantization_config.py,sha256=tpy19ZwFCqPWCFtTc76K_kv2dEzFqojdsnWhYxjdDUA,15015
 model_compression_toolkit/target_platform_capabilities/target_platform/operators.py,sha256=rRmrmPBY4rxCWVpEc6FxeOPUFh8MkfwgQsqD82U9a7w,3108
 model_compression_toolkit/target_platform_capabilities/target_platform/target_platform_model.py,sha256=55EQyP4McJBwsw9_l9eseEZ-V2ygQfdRXjjJLF6HdIM,9482
 model_compression_toolkit/target_platform_capabilities/target_platform/target_platform_model_component.py,sha256=TDbNQwmF7Id-FoIQZlR7ZOcz_nRb4XKBmDihAgKT0u8,1392
@@ -483,8 +483,8 @@ model_compression_toolkit/trainable_infrastructure/keras/quantize_wrapper.py,sha
 model_compression_toolkit/trainable_infrastructure/keras/quantizer_utils.py,sha256=MVwXNymmFRB2NXIBx4e2mdJ1RfoHxRPYRgjb1MQP5kY,1797
 model_compression_toolkit/trainable_infrastructure/pytorch/__init__.py,sha256=huHoBUcKNB6BnY6YaUCcFvdyBtBI172ZoUD8ZYeNc6o,696
 model_compression_toolkit/trainable_infrastructure/pytorch/base_pytorch_quantizer.py,sha256=MxylaVFPgN7zBiRBy6WV610EA4scLgRJFbMucKvvNDU,2896
-mct_nightly-2.1.0.20240529.431.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-mct_nightly-2.1.0.20240529.431.dist-info/METADATA,sha256=IK0APkxwHkLMatb9oW_-VOZ8-CXihqoqyA2TL2drWYg,19721
-mct_nightly-2.1.0.20240529.431.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-mct_nightly-2.1.0.20240529.431.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-2.1.0.20240529.431.dist-info/RECORD,,
+mct_nightly-2.1.0.20240530.440.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+mct_nightly-2.1.0.20240530.440.dist-info/METADATA,sha256=mm-nfq4JHKo4P57r3asIsMswm8EfDjVV7H_gC4WWyqQ,19721
+mct_nightly-2.1.0.20240530.440.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+mct_nightly-2.1.0.20240530.440.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-2.1.0.20240530.440.dist-info/RECORD,,

model_compression_toolkit/__init__.py

@@ -27,4 +27,4 @@ from model_compression_toolkit import data_generation
 from model_compression_toolkit import pruning
 from model_compression_toolkit.trainable_infrastructure.keras.load_model import keras_load_quantized_model
 
-__version__ = "2.1.0.20240529.000431"
+__version__ = "2.1.0.20240530.000440"

model_compression_toolkit/target_platform_capabilities/target_platform/op_quantization_config.py

@@ -197,7 +197,7 @@ class OpQuantizationConfig:
             self.simd_size == other.simd_size
 
 
-class QuantizationConfigOptions(object):
+class QuantizationConfigOptions:
     """
 
     Wrap a set of quantization configurations to consider during the quantization
@@ -215,19 +215,24 @@ class QuantizationConfigOptions(object):
         """
 
         assert isinstance(quantization_config_list,
-                          list), f'\'QuantizationConfigOptions\' options list must be a list, but received: {type(quantization_config_list)}.'
-        assert len(quantization_config_list) > 0, f'Options list can not be empty.'
+                          list), f"'QuantizationConfigOptions' options list must be a list, but received: {type(quantization_config_list)}."
         for cfg in quantization_config_list:
-            assert isinstance(cfg, OpQuantizationConfig), f'Each option must be an instance of \'OpQuantizationConfig\', but found an object of type: {type(cfg)}.'
+            assert isinstance(cfg, OpQuantizationConfig),\
+                f"Each option must be an instance of 'OpQuantizationConfig', but found an object of type: {type(cfg)}."
         self.quantization_config_list = quantization_config_list
         if len(quantization_config_list) > 1:
-            assert base_config is not None, f'For multiple configurations, a \'base_config\' is required for non-mixed-precision optimization.'
-            assert base_config in quantization_config_list, f"\'base_config\' must be included in the quantization config options list."
+            assert base_config is not None, \
+                f"For multiple configurations, a 'base_config' is required for non-mixed-precision optimization."
+            assert any([base_config is cfg for cfg in quantization_config_list]), \
+                f"'base_config' must be included in the quantization config options list."
+            # Enforce base_config to be a reference to an instance in quantization_config_list.
             self.base_config = base_config
         elif len(quantization_config_list) == 1:
+            assert base_config is None or base_config == quantization_config_list[0], "'base_config' should be included in 'quantization_config_list'"
+            # Set base_config to be a reference to the first instance in quantization_config_list.
             self.base_config = quantization_config_list[0]
         else:
-            Logger.critical("\'QuantizationConfigOptions\' requires at least one \'OpQuantizationConfig\'; the provided list is empty.")
+            raise AssertionError("'QuantizationConfigOptions' requires at least one 'OpQuantizationConfig'. The provided list is empty.")
 
     def __eq__(self, other):
         """