mct-nightly 2.4.0.20250623.629__py3-none-any.whl → 2.4.0.20250624.605__py3-none-any.whl

{mct_nightly-2.4.0.20250623.629.dist-info → mct_nightly-2.4.0.20250624.605.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mct-nightly
-Version: 2.4.0.20250623.629
+Version: 2.4.0.20250624.605
 Summary: A Model Compression Toolkit for neural networks
 Author-email: ssi-dnn-dev@sony.com
 Classifier: Programming Language :: Python :: 3

{mct_nightly-2.4.0.20250623.629.dist-info → mct_nightly-2.4.0.20250624.605.dist-info}/RECORD

@@ -1,5 +1,5 @@
-mct_nightly-2.4.0.20250623.629.dist-info/licenses/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-model_compression_toolkit/__init__.py,sha256=yFCUSkGN3NdOEAAWhK0anOjvFT994h3jzn3LL-4Uc1Y,1557
+mct_nightly-2.4.0.20250624.605.dist-info/licenses/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+model_compression_toolkit/__init__.py,sha256=ptUqfuUK307Is3CAZ51Qlt2sDHnhvAP-BmXvOVr1vbw,1557
 model_compression_toolkit/constants.py,sha256=KNgiNLpsMgSYyXMNEbHXd4bFNerQc1D6HH3vpbUq_Gs,4086
 model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
 model_compression_toolkit/logger.py,sha256=L3q7tn3Uht0i_7phnlOWMR2Te2zvzrt2HOz9vYEInts,4529
@@ -69,7 +69,7 @@ model_compression_toolkit/core/common/mixed_precision/mixed_precision_quantizati
 model_compression_toolkit/core/common/mixed_precision/mixed_precision_ru_helper.py,sha256=MMb7qTwk_141-mxz1xch3lMb5F6eQjBf_uILcqXs1wE,4887
 model_compression_toolkit/core/common/mixed_precision/mixed_precision_search_facade.py,sha256=XnSNyG6ZLrIW4Y4_t-ggFvzBjag2RNejfiwbGYfk_Rg,6155
 model_compression_toolkit/core/common/mixed_precision/mixed_precision_search_manager.py,sha256=xdJ8v7M6De1S_F-kJwQFVxzDkPKRJe9sX9nQPPpfrZU,28326
-model_compression_toolkit/core/common/mixed_precision/solution_refinement_procedure.py,sha256=MY8df-c_kITEr_7hOctaxhdiq29hSTA0La9Qo0oTJJY,9678
+model_compression_toolkit/core/common/mixed_precision/solution_refinement_procedure.py,sha256=-4RjoWjCUstJes8b45z33ZYy9vTwI7n953LOWTvhvwE,9840
 model_compression_toolkit/core/common/mixed_precision/resource_utilization_tools/__init__.py,sha256=Rf1RcYmelmdZmBV5qOKvKWF575ofc06JFQSq83Jz99A,696
 model_compression_toolkit/core/common/mixed_precision/resource_utilization_tools/resource_utilization.py,sha256=PKkhc5q8pEPnNLXwo3U56EOCfYnPXIvPs0LlCGZOoKU,4426
 model_compression_toolkit/core/common/mixed_precision/resource_utilization_tools/resource_utilization_calculator.py,sha256=gs8xjQCbk16oOzmtOVc_vO7cl8SCHFe4xZdPW-ePzS4,39948
@@ -529,7 +529,7 @@ model_compression_toolkit/xquant/pytorch/model_analyzer.py,sha256=b93o800yVB3Z-i
 model_compression_toolkit/xquant/pytorch/pytorch_report_utils.py,sha256=Y0oBl8qPFsdNrK49XczwmVacInJcOPHslVnFBs-iTCc,3742
 model_compression_toolkit/xquant/pytorch/similarity_functions.py,sha256=CERxq5K8rqaiE-DlwhZBTUd9x69dtYJlkHOPLB54vm8,2354
 model_compression_toolkit/xquant/pytorch/tensorboard_utils.py,sha256=n0HvWBzkBkUJZlS3WeynhpsRTps2qQkjlq7luliBHNU,9627
-mct_nightly-2.4.0.20250623.629.dist-info/METADATA,sha256=nG19wrn1KD_d1DISqjoudfucrPMlwbF1Q8EUE9pr8os,25087
-mct_nightly-2.4.0.20250623.629.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mct_nightly-2.4.0.20250623.629.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-2.4.0.20250623.629.dist-info/RECORD,,
+mct_nightly-2.4.0.20250624.605.dist-info/METADATA,sha256=3keDUJm6-DNc56fOyYPJBDCxIO9Rfavef0GH3Dprn6A,25087
+mct_nightly-2.4.0.20250624.605.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mct_nightly-2.4.0.20250624.605.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-2.4.0.20250624.605.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.4.0.20250623.000629"
+__version__ = "2.4.0.20250624.000605"

model_compression_toolkit/core/common/mixed_precision/solution_refinement_procedure.py

@@ -50,9 +50,12 @@ def greedy_solution_refinement_procedure(mp_solution: Dict[BaseNode, int],
     if target_resource_utilization.bops_restricted():
         Logger.info(f'Target resource utilization constraint BOPs - Skipping MP greedy solution refinement')
         return mp_solution
-
     assert search_manager.using_virtual_graph is False
 
+    tru = target_resource_utilization
+    activation_restricted = tru.activation_restricted() or tru.total_mem_restricted() or tru.bops_restricted()
+    weights_restricted = tru.weight_restricted() or tru.total_mem_restricted() or tru.bops_restricted()
+
     new_solution = mp_solution.copy()
     changed = True
 
@@ -62,7 +65,7 @@ def greedy_solution_refinement_procedure(mp_solution: Dict[BaseNode, int],
         nodes_next_candidate = {}
 
         for node in search_manager.mp_topo_configurable_nodes:
-            if new_solution[node] == 0:
+            if new_solution[node] == node.find_max_candidate_index():
                 # layer has max config in the given solution, nothing to optimize
                 continue
 
@@ -71,9 +74,8 @@ def greedy_solution_refinement_procedure(mp_solution: Dict[BaseNode, int],
             # only weights kernel attribute is quantized with weights mixed precision
             valid_candidates = _get_valid_candidates_indices(node_candidates,
                                                              new_solution[node],
-                                                             target_resource_utilization.activation_restricted(),
-                                                             target_resource_utilization.weight_restricted()
-                                                             )
+                                                             activation_restricted,
+                                                             weights_restricted)
 
             # Create a list of ru for the valid candidates.
             updated_ru = []