[WIP][high-risk][Step 1]Refactor compressor and quantizer

auto_round/algorithms/base.py

@@ -13,36 +13,17 @@
 # limitations under the License.
 
 
-from contextlib import contextmanager
 from typing import Any
 
 from auto_round.algorithms.registry import resolve_pipeline_member
-from auto_round.schemes import QuantizationScheme
-
-
-class BaseAlgorithm:
-    pass
 
 
+# TODO later wenhuach may be deleted
 class BasePipelineMember:
     """Shared interface for all members of a quantization pipeline."""
 
     model_context = None
     compress_context = None
-    _scheme_context_fields = set(QuantizationScheme.get_attributes())
-    bits: int | None
-    group_size: int | tuple | None
-    sym: bool | None
-    data_type: str | None
-    act_bits: int | None
-    act_group_size: int | None
-    act_sym: bool | None
-    act_data_type: str | None
-    act_dynamic: bool | None
-    super_bits: int | None
-    super_group_size: int | None
-    super_sym: bool | None
-    scale_dtype: str | None
 
     def __init__(self, config: Any = None) -> None:
         self.config = config
@@ -58,6 +39,7 @@ def from_config(cls, config: Any) -> "BasePipelineMember":
 
     def bind(self, compressor: Any) -> None:
         """Wire shared context from the owning compressor."""
+        # self.compressor = compressor
         self.model_context = compressor.model_context
         self.compress_context = compressor.compress_context
         self.scheme = getattr(compressor, "scheme_context", None)
@@ -66,35 +48,6 @@ def prepare_run(self, compressor: Any) -> None:
         """Model-level preparation called once before block iteration starts."""
         return
 
-    def get_act_calib_policy(self, ctx: Any) -> Any:
-        """Return the activation calibration policy for this block."""
-        from auto_round.algorithms.pipeline import ActCalibPolicy, CalibTiming, InputSource
-
-        return ActCalibPolicy(when=CalibTiming.SKIP, source=InputSource.FP_CACHE)
-
-    @contextmanager
-    def block_forward_hooks(self, ctx: Any) -> Any:
-        """Register algorithm-specific forward hooks for the reference forward."""
-        yield []
-
     def finalize_run(self, compressor: Any) -> None:
         """Model-level teardown called once after all blocks are processed."""
         return
-
-
-def _make_scheme_property(name):
-    def getter(self):
-        scheme = getattr(self, "scheme", None)
-        return getattr(scheme, name, None) if scheme is not None else None
-
-    def setter(self, value):
-        scheme = getattr(self, "scheme", None)
-        if scheme is None:
-            raise AttributeError(f"{type(self).__name__} has no bound scheme")
-        setattr(scheme, name, value)
-
-    return property(getter, setter)
-
-
-for _scheme_field in QuantizationScheme.get_attributes():
-    setattr(BasePipelineMember, _scheme_field, _make_scheme_property(_scheme_field))

auto_round/algorithms/config_resolver.py

@@ -150,13 +150,13 @@ def resolve_shared_config_values(configs: list[Any]) -> list[Any]:
     return configs
 
 
-def sync_shared_config_from(source_config: Any, target_configs: list[Any]) -> None:
-    """Propagate resolved source values to targets that already define matching attrs."""
-    source_attrs = _public_config_attrs(source_config)
-    for target in _quantization_configs(target_configs):
-        if target is source_config:
-            continue
-        target_attrs = _public_config_attrs(target)
-        for attr_name, source_value in source_attrs.items():
-            if attr_name in target_attrs and source_value is not None:
-                setattr(target, attr_name, source_value)
+# def sync_shared_config_from(source_config: Any, target_configs: list[Any]) -> None:
+#     """Propagate resolved source values to targets that already define matching attrs."""
+#     source_attrs = _public_config_attrs(source_config)
+#     for target in _quantization_configs(target_configs):
+#         if target is source_config:
+#             continue
+#         target_attrs = _public_config_attrs(target)
+#         for attr_name, source_value in source_attrs.items():
+#             if attr_name in target_attrs and source_value is not None:
+#                 setattr(target, attr_name, source_value)