tico 0.1.0.dev250806__py3-none-any.whl → 0.1.0.dev250807__py3-none-any.whl

tico/__init__.py

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

tico/experimental/quantization/evaluation/metric.py

@@ -42,6 +42,22 @@ def compute_peir(base: torch.Tensor, target: torch.Tensor) -> float:
     return peak_error / interval
 
 
+def mse(base: torch.Tensor, target: torch.Tensor) -> float:
+    """
+    Mean Squared Error (MSE).
+    Penalizes **larger** deviations more heavily than MAE by squaring each
+    difference — helpful to expose occasional large spikes.
+    Formula
+    -------
+        MSE = mean((base - target)²)
+    Returns
+    -------
+    float
+        Mean squared error. *Lower is better*.
+    """
+    return torch.mean((base.detach() - target.detach()) ** 2).item()
+
+
 class MetricCalculator:
     """
     Lightweight registry-and-dispatcher for **pair-wise tensor comparison metrics**.
@@ -83,6 +99,7 @@ class MetricCalculator:
         "diff": compute_max_abs_diff,
         "max_abs_diff": compute_max_abs_diff,
         "peir": compute_peir,
+        "mse": mse,
     }
 
     def __init__(

tico/experimental/quantization/ptq/__init__.py

@@ -0,0 +1,13 @@
+"""
+Public PTQ API — re-export the most common symbols.
+"""
+
+from tico.experimental.quantization.ptq.dtypes import DType
+from tico.experimental.quantization.ptq.mode import Mode
+from tico.experimental.quantization.ptq.qscheme import QScheme
+
+__all__ = [
+    "DType",
+    "Mode",
+    "QScheme",
+]

tico/experimental/quantization/ptq/dtypes.py

@@ -0,0 +1,70 @@
+# 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 dataclasses import dataclass
+
+
+@dataclass(frozen=True)
+class DType:
+    """
+    Self-contained integer dtypes for quantization.
+
+    A DType is just an immutable value-object with two fields:
+      - bits
+      - signed
+
+    Common presets (INT8, UINT4, ..) are provided as constants for convenience.
+    """
+
+    bits: int  # pylint: disable=used-before-assignment
+    signed: bool = False  # False -> unsigned
+
+    @property
+    def qmin(self) -> int:
+        assert self.bits is not None
+        if self.signed:
+            return -(1 << (self.bits - 1))
+        return 0
+
+    @property
+    def qmax(self) -> int:
+        assert self.bits is not None
+        if self.signed:
+            return (1 << (self.bits - 1)) - 1
+        return (1 << self.bits) - 1
+
+    def __str__(self) -> str:
+        prefix = "int" if self.signed else "uint"
+        return f"{prefix}{self.bits}"
+
+    # ────────────────────────────────
+    #  Factory helpers
+    # ────────────────────────────────
+    @staticmethod
+    def int(bits: int):  # type: ignore[valid-type]
+        return DType(bits, signed=True)
+
+    @staticmethod
+    def uint(bits: int):  # type: ignore[valid-type]
+        return DType(bits, signed=False)
+
+
+# ---------------------------------------------------------------------
+#  Convenient canned versions
+# ---------------------------------------------------------------------
+UINT4 = DType.uint(4)
+INT4 = DType.int(4)
+INT8 = DType.int(8)
+UINT8 = DType.uint(8)
+INT16 = DType.int(16)

tico/experimental/quantization/ptq/mode.py

@@ -0,0 +1,32 @@
+# 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 enum import auto, Enum
+
+
+class Mode(Enum):
+    """
+    Mode — global FSM for PTQWrapper & Handlers.
+
+      • NO_QUANT : pure pass-through (no stats, no fake-quant)
+      • CALIB    : collect observer statistics only
+      • QUANT    : use cached (scale, zero-point) → fake-quant enabled
+    """
+
+    NO_QUANT = auto()
+    CALIB = auto()
+    QUANT = auto()
+
+    def __str__(self) -> str:
+        return self.name.lower()

tico/experimental/quantization/ptq/qscheme.py

@@ -0,0 +1,40 @@
+# 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 enum import auto, Enum
+
+
+class QScheme(Enum):
+    # ───── Per-tensor ────────────
+    PER_TENSOR_ASYMM = auto()
+    PER_TENSOR_SYMM = auto()
+    # ───── Per-channel ───────────
+    PER_CHANNEL_ASYMM = auto()
+    PER_CHANNEL_SYMM = auto()
+
+    # helper
+    def is_per_channel(self) -> bool:
+        return self in {
+            QScheme.PER_CHANNEL_ASYMM,
+            QScheme.PER_CHANNEL_SYMM,
+        }
+
+    def is_symmetric(self) -> bool:
+        return self in {
+            QScheme.PER_TENSOR_SYMM,
+            QScheme.PER_CHANNEL_SYMM,
+        }
+
+    def __str__(self) -> str:
+        return self.name.lower()

{tico-0.1.0.dev250806.dist-info → tico-0.1.0.dev250807.dist-info}/METADATA

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

{tico-0.1.0.dev250806.dist-info → tico-0.1.0.dev250807.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tico/__init__.py,sha256=ElClZCa-CW2dIDsOnUp1MCbDGvocTAgpyklSI4vEPiQ,1883
+tico/__init__.py,sha256=xXynWlns6HYdl6Zf9xwJtdTnFkscm8b49vqhn8FYJZg,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
@@ -43,7 +43,7 @@ tico/experimental/quantization/algorithm/smoothquant/smooth_quant.py,sha256=O1h7
 tico/experimental/quantization/evaluation/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/experimental/quantization/evaluation/backend.py,sha256=CZL9rZOA0t8cH7PHp6u9l7dGqWNvTj9bKOvwo0PVul0,692
 tico/experimental/quantization/evaluation/evaluate.py,sha256=kfa_GvFaX6DoSTAmuCImMJqF2jgqtnor5UpC7wVmGPI,7877
-tico/experimental/quantization/evaluation/metric.py,sha256=ma8nrvt3jjKlE8d7iK3uABTkxOSrEDb8hHevdoc2pIw,5011
+tico/experimental/quantization/evaluation/metric.py,sha256=t9M058dOQ8iy_2PcrbNMAebBNJs8TU8USZw_nbi2iWI,5488
 tico/experimental/quantization/evaluation/utils.py,sha256=82RG_e5LuKfWo786wEZUVwXY93nNl901n04fB7D0Z6k,5909
 tico/experimental/quantization/evaluation/executor/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/experimental/quantization/evaluation/executor/backend_executor.py,sha256=3kLu3_rcsreA_NK42yRgRgubPtZmVp7QCRvaqLNw10E,1522
@@ -56,6 +56,10 @@ tico/experimental/quantization/passes/propagate_qparam_backward.py,sha256=TGtyW0
 tico/experimental/quantization/passes/propagate_qparam_forward.py,sha256=RhUHGCR2RpBO5KYkQ7Z8U5u7HEwDq2wdKHLKAJCi-5c,5138
 tico/experimental/quantization/passes/quantize_bias.py,sha256=T7YxJ70N0tSK0FF9VJZA5iP0sHdnnsX9GX4AT4JDFSk,4325
 tico/experimental/quantization/passes/remove_weight_dequant_op.py,sha256=gI1MtrHazWpdNfys7f1ngTTWplzluF7SA-uX0HMR5Mc,6592
+tico/experimental/quantization/ptq/__init__.py,sha256=ZoPdEwZ1i1n5pBFChx8GuUrkfRP2vsSoLPNILQjNBaA,298
+tico/experimental/quantization/ptq/dtypes.py,sha256=xfCBtq6mQmUYRwsoFgII6gvRl1raQi0Inj9pznDuKwQ,2236
+tico/experimental/quantization/ptq/mode.py,sha256=lT-T8vIv8YWcwrjT7xXVhOw1g7aoAdh_3PWB-ptPKaI,1052
+tico/experimental/quantization/ptq/qscheme.py,sha256=uwhv7bCxOOXB3I-IKlRyr_u4eXOq48uIqGy4TLDqGxY,1301
 tico/interpreter/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/interpreter/infer.py,sha256=1ZFe3DVMR2mlwBosoedqoL0-CGN_01CKLgMgxuw62KA,4861
 tico/interpreter/interpreter.py,sha256=tGbluCbrehTCqBu8mtGDNzby_ieJ2ry8_RH_eC0CQxk,3828
@@ -206,9 +210,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.dev250806.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev250806.dist-info/METADATA,sha256=_lGucGVdG9FrBy8SvOy1BxwlH3mxLLk5HpZWIUPdA48,8450
-tico-0.1.0.dev250806.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev250806.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev250806.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev250806.dist-info/RECORD,,
+tico-0.1.0.dev250807.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.1.0.dev250807.dist-info/METADATA,sha256=ATnzGR44gSRsXtSTjRBwY0DOQK512GtszXrNo3r3rD4,8450
+tico-0.1.0.dev250807.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tico-0.1.0.dev250807.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
+tico-0.1.0.dev250807.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.1.0.dev250807.dist-info/RECORD,,