mct-nightly 2.2.0.20241212.528__tar.gz → 2.2.0.20241213.540__tar.gz

{mct-nightly-2.2.0.20241212.528 → mct-nightly-2.2.0.20241213.540}/PKG-INFO

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

{mct-nightly-2.2.0.20241212.528 → mct-nightly-2.2.0.20241213.540}/mct_nightly.egg-info/PKG-INFO

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

{mct-nightly-2.2.0.20241212.528 → mct-nightly-2.2.0.20241213.540}/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.2.0.20241212.000528"
+__version__ = "2.2.0.20241213.000540"

{mct-nightly-2.2.0.20241212.528 → mct-nightly-2.2.0.20241213.540}/model_compression_toolkit/core/common/graph/base_node.py

@@ -24,6 +24,7 @@ from model_compression_toolkit.core.common.quantization.node_quantization_config
 from model_compression_toolkit.logger import Logger
 from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import QuantizationConfigOptions, \
     OpQuantizationConfig
+from model_compression_toolkit.target_platform_capabilities.schema.schema_functions import max_input_activation_n_bits
 from model_compression_toolkit.target_platform_capabilities.target_platform import TargetPlatformCapabilities, LayerFilterParams
 
 
@@ -584,7 +585,7 @@ class BaseNode:
         _node_qc_options = node_qc_options.quantization_config_list
         if len(next_nodes):
             next_nodes_qc_options = [_node.get_qco(tpc) for _node in next_nodes]
-            next_nodes_supported_input_bitwidth = min([op_cfg.max_input_activation_n_bits
+            next_nodes_supported_input_bitwidth = min([max_input_activation_n_bits(op_cfg)
                                                        for qc_opts in next_nodes_qc_options
                                                        for op_cfg in qc_opts.quantization_config_list])
 
@@ -595,7 +596,7 @@ class BaseNode:
                 Logger.critical(f"Graph doesn't match TPC bit configurations: {self} -> {next_nodes}.")  # pragma: no cover
 
             # Verify base config match
-            if any([node_qc_options.base_config.activation_n_bits > qc_opt.base_config.max_input_activation_n_bits
+            if any([node_qc_options.base_config.activation_n_bits > max_input_activation_n_bits(qc_opt.base_config)
                     for qc_opt in next_nodes_qc_options]):
                 # base_config activation bits doesn't match next node supported input bit-width -> replace with
                 # a qco from quantization_config_list with maximum activation bit-width.

{mct-nightly-2.2.0.20241212.528 → mct-nightly-2.2.0.20241213.540}/model_compression_toolkit/core/common/quantization/set_node_quantization_config.py

@@ -32,6 +32,7 @@ from model_compression_toolkit.core.common.quantization.quantization_params_fn_s
     get_activation_quantization_params_fn, get_weights_quantization_params_fn
 from model_compression_toolkit.core.common.quantization.quantization_fn_selection import \
     get_weights_quantization_fn
+from model_compression_toolkit.target_platform_capabilities.schema.schema_functions import max_input_activation_n_bits
 from model_compression_toolkit.target_platform_capabilities.target_platform.targetplatform2framework import TargetPlatformCapabilities
 from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import OpQuantizationConfig, \
     QuantizationConfigOptions
@@ -117,7 +118,7 @@ def filter_node_qco_by_graph(node: BaseNode,
 
     if len(next_nodes):
         next_nodes_qc_options = [_node.get_qco(tpc) for _node in next_nodes]
-        next_nodes_supported_input_bitwidth = min([op_cfg.max_input_activation_n_bits
+        next_nodes_supported_input_bitwidth = min([max_input_activation_n_bits(op_cfg)
                                                    for qc_opts in next_nodes_qc_options
                                                    for op_cfg in qc_opts.quantization_config_list])
 
@@ -128,7 +129,7 @@ def filter_node_qco_by_graph(node: BaseNode,
             Logger.critical(f"Graph doesn't match TPC bit configurations: {node} -> {next_nodes}.")
 
         # Verify base config match
-        if any([node_qc_options.base_config.activation_n_bits > qc_opt.base_config.max_input_activation_n_bits
+        if any([node_qc_options.base_config.activation_n_bits > max_input_activation_n_bits(qc_opt.base_config)
                 for qc_opt in next_nodes_qc_options]):
             # base_config activation bits doesn't match next node supported input bit-width -> replace with
             # a qco from quantization_config_list with maximum activation bit-width.

mct-nightly-2.2.0.20241213.540/model_compression_toolkit/target_platform_capabilities/schema/schema_functions.py

@@ -0,0 +1,106 @@
+# Copyright 2024 Sony Semiconductor Israel, Inc. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+from logging import Logger
+from typing import Optional
+
+from model_compression_toolkit.target_platform_capabilities.schema.mct_current_schema import OpQuantizationConfig, \
+    TargetPlatformModel, QuantizationConfigOptions, OperatorsSetBase
+
+
+def max_input_activation_n_bits(op_quantization_config: OpQuantizationConfig) -> int:
+    """
+    Get the maximum supported input bit-width.
+
+    Args:
+        op_quantization_config (OpQuantizationConfig):  The configuration object from which to retrieve the maximum supported input bit-width.
+
+    Returns:
+        int: Maximum supported input bit-width.
+    """
+    return max(op_quantization_config.supported_input_activation_n_bits)
+
+
+def get_config_options_by_operators_set(tp_model: TargetPlatformModel,
+                                        operators_set_name: str) -> QuantizationConfigOptions:
+    """
+    Get the QuantizationConfigOptions of an OperatorsSet by its name.
+
+    Args:
+        tp_model (TargetPlatformModel): The target platform model containing the operator sets and their configurations.
+        operators_set_name (str): The name of the OperatorsSet whose quantization configuration options are to be retrieved.
+
+    Returns:
+        QuantizationConfigOptions: The quantization configuration options associated with the specified OperatorsSet,
+        or the default quantization configuration options if the OperatorsSet is not found.
+    """
+    for op_set in tp_model.operator_set:
+        if operators_set_name == op_set.name:
+            return op_set.qc_options
+    return tp_model.default_qco
+
+
+def get_default_op_quantization_config(tp_model: TargetPlatformModel) -> OpQuantizationConfig:
+    """
+    Get the default OpQuantizationConfig of the TargetPlatformModel.
+
+    Args:
+        tp_model (TargetPlatformModel): The target platform model containing the default quantization configuration.
+
+    Returns:
+        OpQuantizationConfig: The default quantization configuration.
+
+    Raises:
+        AssertionError: If the default quantization configuration list contains more than one configuration option.
+    """
+    assert len(tp_model.default_qco.quantization_config_list) == 1, \
+        f"Default quantization configuration options must contain only one option, " \
+        f"but found {len(tp_model.default_qco.quantization_config_list)} configurations." # pragma: no cover
+    return tp_model.default_qco.quantization_config_list[0]
+
+
+def is_opset_in_model(tp_model: TargetPlatformModel, opset_name: str) -> bool:
+    """
+    Check whether an OperatorsSet is defined in the model.
+
+    Args:
+        tp_model (TargetPlatformModel): The target platform model containing the list of operator sets.
+        opset_name (str): The name of the OperatorsSet to check for existence.
+
+    Returns:
+        bool: True if an OperatorsSet with the given name exists in the target platform model,
+              otherwise False.
+    """
+    return opset_name in [x.name for x in tp_model.operator_set]
+
+
+def get_opset_by_name(tp_model: TargetPlatformModel, opset_name: str) -> Optional[OperatorsSetBase]:
+    """
+    Get an OperatorsSet object from the model by its name.
+
+    Args:
+        tp_model (TargetPlatformModel): The target platform model containing the list of operator sets.
+        opset_name (str): The name of the OperatorsSet to be retrieved.
+
+    Returns:
+        Optional[OperatorsSetBase]: The OperatorsSet object with the specified name if found.
+        If no operator set with the specified name is found, None is returned.
+
+    Raises:
+        A critical log message if multiple operator sets with the same name are found.
+    """
+    opset_list = [x for x in tp_model.operator_set if x.name == opset_name]
+    if len(opset_list) > 1:
+        Logger.critical(f"Found more than one OperatorsSet in TargetPlatformModel with the name {opset_name}.") # pragma: no cover
+    return opset_list[0] if opset_list else None