mct-nightly 1.11.0.20240321.357__py3-none-any.whl → 1.11.0.20240323.408__py3-none-any.whl

model_compression_toolkit/core/common/mixed_precision/kpi_tools/kpi_aggregation_methods.py

@@ -1,105 +0,0 @@
-# Copyright 2022 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.
-# ==============================================================================
-import copy
-from enum import Enum
-from functools import partial
-from typing import List, Any
-import numpy as np
-
-from pulp import lpSum
-
-
-def sum_kpi(kpi_vector: np.ndarray, set_constraints: bool = True) -> List[Any]:
-    """
-    Aggregates KPIs vector to a single KPI measure by summing all values.
-
-    Args:
-        kpi_vector: A vector with nodes' KPI values.
-        set_constraints: A flag for utilizing the method for KPI computation of a
-            given config not for LP formalization purposes.
-
-    Returns: A list with an lpSum object for lp problem definition with the vector's sum.
-
-    """
-    if not set_constraints:
-        return [0] if len(kpi_vector) == 0 else [sum(kpi_vector)]
-    return [lpSum(kpi_vector)]
-
-
-def max_kpi(kpi_vector: np.ndarray, set_constraints: bool = True) -> List[float]:
-    """
-    Aggregates KPIs vector to allow max constraint in the linear programming problem formalization.
-    In order to do so, we need to define a separate constraint on each value in the KPI vector,
-    to be bounded by the target KPI.
-
-    Args:
-        kpi_vector: A vector with nodes' KPI values.
-        set_constraints: A flag for utilizing the method for KPI computation of a
-            given config not for LP formalization purposes.
-
-    Returns: A list with the vector's values, to be used to define max constraint
-    in the linear programming problem formalization.
-
-    """
-    if not set_constraints:
-        return [0] if len(kpi_vector) == 0 else [max(kpi_vector)]
-    return [kpi for kpi in kpi_vector]
-
-
-def total_kpi(kpi_tensor: np.ndarray, set_constraints: bool = True) -> List[float]:
-    """
-    Aggregates KPIs vector to allow weights and activation total kpi constraint in the linear programming
-    problem formalization. In order to do so, we need to define a separate constraint on each activation value in
-    the KPI vector, combined with the sum weights kpi.
-    Note that the given kpi_tensor should contain weights and activation kpi values in each entry.
-
-    Args:
-        kpi_tensor: A tensor with nodes' KPI values for weights and activation.
-        set_constraints: A flag for utilizing the method for KPI computation of a
-            given config not for LP formalization purposes.
-
-    Returns: A list with lpSum objects, to be used to define total constraint
-    in the linear programming problem formalization.
-
-    """
-    if not set_constraints:
-        weights_kpi = sum([kpi[0] for kpi in kpi_tensor])
-        activation_kpi = max([kpi[1] for kpi in kpi_tensor])
-        return [weights_kpi + activation_kpi]
-
-    weights_kpi = lpSum([kpi[0] for kpi in kpi_tensor])
-    total_kpis = [weights_kpi + activation_kpi for _, activation_kpi in kpi_tensor]
-
-    return total_kpis
-
-
-class MpKpiAggregation(Enum):
-    """
-    Defines kpi aggregation functions that can be used to compute final KPI metric.
-    The enum values can be used to call a function on a set of arguments.
-
-     SUM - applies the sum_kpi function
-
-     MAX - applies the max_kpi function
-
-     TOTAL - applies the total_kpi function
-
-    """
-    SUM = partial(sum_kpi)
-    MAX = partial(max_kpi)
-    TOTAL = partial(total_kpi)
-
-    def __call__(self, *args):
-        return self.value(*args)

model_compression_toolkit/core/common/mixed_precision/kpi_tools/kpi_functions_mapping.py

@@ -1,26 +0,0 @@
-# Copyright 2022 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 model_compression_toolkit.core.common.mixed_precision.kpi_tools.kpi import KPITarget
-from model_compression_toolkit.core.common.mixed_precision.kpi_tools.kpi_aggregation_methods import MpKpiAggregation
-from model_compression_toolkit.core.common.mixed_precision.kpi_tools.kpi_methods import MpKpiMetric
-
-
-# When adding a KPITarget that we want to consider in our mp search,
-# a matching pair of kpi_tools computation function and a kpi_tools
-# aggregation function should be added to this dictionary
-kpi_functions_mapping = {KPITarget.WEIGHTS: (MpKpiMetric.WEIGHTS_SIZE, MpKpiAggregation.SUM),
-                         KPITarget.ACTIVATION: (MpKpiMetric.ACTIVATION_OUTPUT_SIZE, MpKpiAggregation.MAX),
-                         KPITarget.TOTAL: (MpKpiMetric.TOTAL_WEIGHTS_ACTIVATION_SIZE, MpKpiAggregation.TOTAL),
-                         KPITarget.BOPS: (MpKpiMetric.BOPS_COUNT, MpKpiAggregation.SUM)}