tico 0.3.0.dev261007__tar.gz → 0.3.0.dev261008__tar.gz

{tico-0.3.0.dev261007 → tico-0.3.0.dev261008}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tico
-Version: 0.3.0.dev261007
+Version: 0.3.0.dev261008
 Summary: Convert Exported Torch Module To Circle
 License: This file provides full text of licenses used in this project
         

tico-0.3.0.dev261008/tico/_version.py

@@ -0,0 +1 @@
+__version__ = "0.3.0.dev261008"

{tico-0.3.0.dev261007 → tico-0.3.0.dev261008}/tico/quantization/recipes/override_policies.py

@@ -89,6 +89,7 @@ def apply_ptq_override_policies(
     *,
     family: str,
     model: Any,
+    applied_paths: set[tuple[str, ...]] | None = None,
 ) -> PTQConfig:
     """Apply recipe-level PTQ override policies to an existing PTQConfig.
 
@@ -97,6 +98,11 @@ def apply_ptq_override_policies(
         stage_cfg: PTQ stage YAML payload.
         family: Registered recipe adapter family.
         model: Loaded model instance used to infer component metadata.
+        applied_paths: Optional set that receives the exact override path of
+            every selector or raw override applied to ``qcfg``. Paths are added
+            as each override is applied; existing entries are never removed,
+            so pass a fresh empty set to observe one call in isolation. When
+            ``None``, no paths are collected.
 
     Returns:
         The same PTQConfig instance with selector and raw overrides applied.
@@ -105,15 +111,27 @@ def apply_ptq_override_policies(
         return qcfg
 
     context = build_quant_target_resolver_context(family=family, model=model)
-    return apply_ptq_override_policies_to_config(qcfg, stage_cfg, context)
+    return apply_ptq_override_policies_to_config(
+        qcfg,
+        stage_cfg,
+        context,
+        applied_paths=applied_paths,
+    )
 
 
 def apply_ptq_override_policies_to_config(
     qcfg: PTQConfig,
     stage_cfg: Mapping[str, Any],
     context: QuantTargetResolverContext,
+    *,
+    applied_paths: set[tuple[str, ...]] | None = None,
 ) -> PTQConfig:
-    """Apply PTQ override policies using an already-built resolver context."""
+    """Apply PTQ override policies using an already-built resolver context.
+
+    ``applied_paths`` follows the same contract as in
+    :func:`apply_ptq_override_policies`: applied override paths are added to
+    the given set in application order and existing entries are preserved.
+    """
     specs = parse_named_specs(stage_cfg.get("specs", {}))
     resolved = compile_override_policies(
         stage_cfg.get("override_policies", []),
@@ -122,12 +140,16 @@ def apply_ptq_override_policies_to_config(
     )
     for override in resolved:
         qcfg.set_override(override.path, override.value)
+        if applied_paths is not None:
+            applied_paths.add(override.path)
 
     for path, value in compile_raw_overrides(
         stage_cfg.get("raw_overrides", {}),
         specs=specs,
     ):
         qcfg.set_override(path, value)
+        if applied_paths is not None:
+            applied_paths.add(path)
 
     return qcfg
 

{tico-0.3.0.dev261007 → tico-0.3.0.dev261008}/tico/quantization/recipes/stages/ptq.py

@@ -17,13 +17,7 @@ from typing import Any, Mapping
 from tico.quantization import convert, prepare
 from tico.quantization.config.ptq import PTQConfig
 from tico.quantization.recipes.context import RecipeContext
-from tico.quantization.recipes.override_policies import (
-    apply_ptq_override_policies,
-    build_quant_target_resolver_context,
-    compile_override_policies,
-    compile_raw_overrides,
-    parse_named_specs,
-)
+from tico.quantization.recipes.override_policies import apply_ptq_override_policies
 from tico.quantization.recipes.qparams import (
     clear_gptq_quantizers,
     find_gptq_quantizers,
@@ -86,42 +80,6 @@ def _reuse_gptq_qparams(stage_cfg: Mapping[str, Any]) -> bool:
     return value
 
 
-def _resolve_configured_override_paths(
-    stage_cfg: Mapping[str, Any],
-    *,
-    family: str,
-    model: Any,
-) -> tuple[tuple[str, ...], ...]:
-    """Resolve exact observer paths targeted by selector and raw overrides."""
-    policies = stage_cfg.get("override_policies")
-    raw_overrides = stage_cfg.get("raw_overrides")
-    if not policies and not raw_overrides:
-        return ()
-
-    specs = parse_named_specs(stage_cfg.get("specs", {}))
-    paths: set[tuple[str, ...]] = set()
-
-    if policies:
-        context = build_quant_target_resolver_context(family=family, model=model)
-        paths.update(
-            override.path
-            for override in compile_override_policies(
-                policies,
-                specs=specs,
-                context=context,
-            )
-        )
-
-    paths.update(
-        path
-        for path, _ in compile_raw_overrides(
-            raw_overrides,
-            specs=specs,
-        )
-    )
-    return tuple(sorted(paths))
-
-
 def _get_effective_override_fields(
     overrides: Mapping[str, Any],
     path: tuple[str, ...],
@@ -188,20 +146,25 @@ class PTQStage(Stage):
         print("Wrapping model with PTQ wrappers …")
         reuse_gptq_qparams = _reuse_gptq_qparams(stage_cfg)
         ptq_config = ctx.adapter.build_ptq_config(ctx, stage_cfg)
+        # Collect the exact paths applied by recipe overrides only when they
+        # will be printed, so the diagnostic reuses the single policy
+        # resolution instead of re-interpreting the stage configuration.
+        applied_override_paths: set[tuple[str, ...]] | None = (
+            set() if bool(stage_cfg.get("print_overrides", False)) else None
+        )
         ptq_config = apply_ptq_override_policies(
             ptq_config,
             stage_cfg,
             family=ctx.adapter.family,
             model=ctx.require_model(),
+            applied_paths=applied_override_paths,
         )
 
-        if bool(stage_cfg.get("print_overrides", False)):
-            override_paths = _resolve_configured_override_paths(
-                stage_cfg,
-                family=ctx.adapter.family,
-                model=ctx.require_model(),
+        if applied_override_paths is not None:
+            _print_effective_overrides(
+                ptq_config,
+                tuple(sorted(applied_override_paths)),
             )
-            _print_effective_overrides(ptq_config, override_paths)
 
         q_model = prepare(ctx.require_model(), ptq_config)
 

{tico-0.3.0.dev261007 → tico-0.3.0.dev261008}/tico.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tico
-Version: 0.3.0.dev261007
+Version: 0.3.0.dev261008
 Summary: Convert Exported Torch Module To Circle
 License: This file provides full text of licenses used in this project
         

tico-0.3.0.dev261007/tico/_version.py

@@ -1 +0,0 @@
-__version__ = "0.3.0.dev261007"