tico 0.1.0.dev250820__py3-none-any.whl → 0.1.0.dev250821__py3-none-any.whl

tico/__init__.py

@@ -29,7 +29,7 @@ __all__ = [
 ]
 
 # THIS LINE IS AUTOMATICALLY GENERATED BY setup.py
-__version__ = "0.1.0.dev250820"
+__version__ = "0.1.0.dev250821"
 
 MINIMUM_SUPPORTED_VERSION = "2.5.0"
 SECURE_TORCH_VERSION = "2.6.0"

tico/experimental/quantization/ptq/wrappers/ptq_wrapper.py

@@ -0,0 +1,54 @@
+# Copyright (c) 2025 Samsung Electronics Co., Ltd. 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 typing import Optional
+
+import torch
+
+from tico.experimental.quantization.ptq.quant_config import QuantConfig
+from tico.experimental.quantization.ptq.wrappers.quant_module_base import (
+    QuantModuleBase,
+)
+from tico.experimental.quantization.ptq.wrappers.registry import lookup
+
+
+class PTQWrapper(QuantModuleBase):
+    """
+    Adapter that turns a fp module into its quantized counterpart.
+
+    It is itself a QuantModuleBase so composite wrappers can treat
+     it exactly like any other quant module.
+    """
+
+    def __init__(
+        self,
+        module: torch.nn.Module,
+        qcfg: Optional[QuantConfig] = None,
+        *,
+        fp_name: Optional[str] = None,
+    ):
+        super().__init__(qcfg)
+        wrapped_cls = lookup(type(module))
+        if wrapped_cls is None:
+            raise NotImplementedError(f"No quant wrapper for {type(module).__name__}")
+        self.wrapped: QuantModuleBase = wrapped_cls(module, qcfg=qcfg, fp_name=fp_name)  # type: ignore[arg-type, misc]
+
+    def forward(self, *args, **kwargs):
+        return self.wrapped(*args, **kwargs)
+
+    def _all_observers(self):
+        yield from self.wrapped._all_observers()
+
+    def extra_repr(self) -> str:
+        return self.wrapped.extra_repr()

tico/experimental/quantization/ptq/wrappers/quant_elementwise.py

@@ -0,0 +1,113 @@
+# Copyright (c) 2025 Samsung Electronics Co., Ltd. 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 typing import Callable, Optional
+
+import torch
+import torch.nn as nn
+
+from tico.experimental.quantization.ptq.quant_config import QuantConfig
+from tico.experimental.quantization.ptq.wrappers.quant_module_base import (
+    QuantModuleBase,
+)
+from tico.experimental.quantization.ptq.wrappers.registry import register
+
+
+class QuantElementwise(QuantModuleBase):
+    """
+    Generic wrapper for any 1-to-1 element-wise op `y = f(x)`.
+
+    Sub-classes only need to implement:
+        • `FUNC`: a Callable that maps tensor→tensor
+    """
+
+    # subclass must set this
+    FUNC: Callable[[torch.Tensor], torch.Tensor] | None = None
+
+    def __init_subclass__(cls, **kwargs):
+        super().__init_subclass__(**kwargs)
+        if cls is QuantElementwise:
+            return
+        if cls.FUNC is None:
+            raise NotImplementedError(
+                f"{cls.__name__} must define a staticmethod `FUNC(tensor) -> tensor`"
+            )
+
+    def __init__(
+        self,
+        fp_module: nn.Module,
+        *,
+        qcfg: Optional[QuantConfig] = None,
+        fp_name: Optional[str] = None,
+    ):
+        super().__init__(qcfg, fp_name=fp_name)
+        self.module = fp_module
+        self.act_in_obs = self._make_obs("act_in")
+        self.act_out_obs = self._make_obs("act_out")
+
+    # ------------------------------------------------------------
+    def forward(self, x):
+        x_q = self._fq(x, self.act_in_obs)
+        assert self.FUNC is not None
+        y = self.FUNC(x_q)  # element-wise op
+        y_q = self._fq(y, self.act_out_obs)
+        return y_q
+
+    # ------------------------------------------------------------
+    def _all_observers(self):
+        return (self.act_in_obs, self.act_out_obs)
+
+
+"""
+Why `FUNC` is a `staticmethod`
+
+- Prevents automatic binding: calling `self.FUNC(x)` will not inject `self`,
+  so the callable keeps the expected signature `Tensor -> Tensor`
+  (e.g., `torch.sigmoid(x)`), avoiding TypeErrors.
+
+- Expresses purity and statelessness: `FUNC` is a pure, element-wise transform
+  that must not read or mutate module state (params, buffers, config).
+
+- Tracing/export friendly (FX / TorchScript): the call is captured as
+  `call_function(torch.*)` instead of a bound `call_method`, which makes graph
+  rewrites/pattern-matching and backends' substitutions more reliable.
+
+- Avoids submodule pollution: we keep a functional op (`torch.relu`) rather
+  than an `nn.Module` instance that would appear in the module tree.
+
+- Small perf/alloc win: no bound-method objects are created on each call.
+"""
+
+# Sigmoid
+@register(nn.Sigmoid)
+class QuantSigmoid(QuantElementwise):
+    FUNC = staticmethod(torch.sigmoid)
+
+
+# Tanh
+@register(nn.Tanh)
+class QuantTanh(QuantElementwise):
+    FUNC = staticmethod(torch.tanh)
+
+
+# ReLU
+@register(nn.ReLU)
+class QuantReLU(QuantElementwise):
+    FUNC = staticmethod(torch.relu)
+
+
+# GELU (approximate)
+@register(nn.GELU)
+class QuantGELU(QuantElementwise):
+    FUNC = staticmethod(torch.nn.functional.gelu)

{tico-0.1.0.dev250820.dist-info → tico-0.1.0.dev250821.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tico
-Version: 0.1.0.dev250820
+Version: 0.1.0.dev250821
 Summary: Convert exported Torch module to circle
 Home-page: UNKNOWN
 License: UNKNOWN

{tico-0.1.0.dev250820.dist-info → tico-0.1.0.dev250821.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tico/__init__.py,sha256=I2l2YkiyVNNCBv8PmWOlk2OxIDiMcpubxCENlSiOh8E,1883
+tico/__init__.py,sha256=yEGf_7r1bSZi_LwntDyrVf0WUNvouR4X_fcua51LRUI,1883
 tico/pt2_to_circle.py,sha256=gu3MD4Iqc0zMZcCZ2IT8oGbyj21CTSbT3Rgd9s2B_9A,2767
 tico/config/__init__.py,sha256=xZzCXjZ84qE-CsBi-dfaL05bqpQ3stKKfTXhnrJRyVs,142
 tico/config/base.py,sha256=q5xMqGxTUZs4mFqt5c7i_y9U00fYgdMGl9nUqIVMlCo,1248
@@ -71,6 +71,8 @@ tico/experimental/quantization/ptq/observers/mx.py,sha256=aP4qmBgeiRIYZJksShN5gs
 tico/experimental/quantization/ptq/utils/__init__.py,sha256=PL9IZgiWoMtsXVljeOy7KymmLVP238SXEFRLXYK72WQ,126
 tico/experimental/quantization/ptq/utils/reduce_utils.py,sha256=3kWawLB91EcvvHlCrNqqfZF7tpgr22htBSA049mKw_4,973
 tico/experimental/quantization/ptq/wrappers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tico/experimental/quantization/ptq/wrappers/ptq_wrapper.py,sha256=KRw_VvFJYvd2OBj4K1sYEXxUwZk9QghMw3NsgjKIAGk,1857
+tico/experimental/quantization/ptq/wrappers/quant_elementwise.py,sha256=LhEoobfvto6zKrBOKL4gmxfFFc31jHzyQV_zfps-iQM,3604
 tico/experimental/quantization/ptq/wrappers/quant_module_base.py,sha256=6RK4bn9G1pzFmkIdBdFf7liBOpb-b7rpthgD83AgkbQ,5256
 tico/experimental/quantization/ptq/wrappers/registry.py,sha256=exXl2wNNzVgC2P9gMjpF_-PqIBgYERGruzh0u1Pril0,4367
 tico/experimental/quantization/ptq/wrappers/nn/__init__.py,sha256=q4A9BiGlsa8ZdGV3y0SDiSkzkdVugsK2iz2daiJqBCY,118
@@ -229,9 +231,9 @@ tico/utils/mx/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/utils/mx/elemwise_ops.py,sha256=V6glyAHsVR1joqpsgnNytatCD_ew92xNWZ19UFDoMTA,10281
 tico/utils/mx/formats.py,sha256=uzNWyu-1onUlwQfX5cZ6fZSUfHMRqorper7_T1k3jfk,3404
 tico/utils/mx/mx_ops.py,sha256=RcfUTYVi-wilGB2sC35OeARdwDqnixv7dG5iyZ-fQT8,8555
-tico-0.1.0.dev250820.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev250820.dist-info/METADATA,sha256=SF_FvVX_JdbjpMAN2-VJGo7zsYU2_nEPXCHap296J3Y,8450
-tico-0.1.0.dev250820.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev250820.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev250820.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev250820.dist-info/RECORD,,
+tico-0.1.0.dev250821.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.1.0.dev250821.dist-info/METADATA,sha256=h3a04PlCzUhqD8MvbgSQHopBPU9IcpNEFx7GthfXVM0,8450
+tico-0.1.0.dev250821.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tico-0.1.0.dev250821.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
+tico-0.1.0.dev250821.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.1.0.dev250821.dist-info/RECORD,,