[WIP]Refactor of compressor and quantizer

auto_round/algorithms/base.py

@@ -13,39 +13,20 @@
 # 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
-
-
 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
+        self.config = config  # TODO wenhuach may be deleted
         self.scheme = getattr(config, "scheme", None)
 
     @classmethod
@@ -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,34 @@ 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
+    def register_fp_input_forward_hooks(self, block: Any) -> list:
+        """Register hooks for the FP-input reference forward pass.
 
-        return ActCalibPolicy(when=CalibTiming.SKIP, source=InputSource.FP_CACHE)
+        Subclasses override to collect statistics during the reference forward.
+        Returns a list of hook handles; caller must call h.remove() on each.
 
-    @contextmanager
-    def block_forward_hooks(self, ctx: Any) -> Any:
-        """Register algorithm-specific forward hooks for the reference forward."""
-        yield []
+        Default: no-op (empty list).
+        """
+        return []
 
     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))
+# 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))