tico 0.1.0.dev250518__py3-none-any.whl → 0.1.0.dev250520__py3-none-any.whl

tico/__init__.py

@@ -22,7 +22,7 @@ from tico.config import CompileConfigV1, get_default_config
 from tico.utils.convert import convert, convert_from_exported_program, convert_from_pt2
 
 # THIS LINE IS AUTOMATICALLY GENERATED BY setup.py
-__version__ = "0.1.0.dev250518"
+__version__ = "0.1.0.dev250520"
 
 
 if Version(torch.__version__) < Version("2.5.0"):

tico/experimental/quantization/passes/remove_weight_dequant_op.py

@@ -17,6 +17,12 @@ from typing import List, Optional, TYPE_CHECKING, Union
 if TYPE_CHECKING:
     import torch.fx
 import torch
+from torch._export.utils import (
+    get_buffer,
+    get_lifted_tensor_constant,
+    is_buffer,
+    is_lifted_tensor_constant,
+)
 from torch._subclasses.fake_tensor import FakeTensor
 from torch.export import ExportedProgram
 
@@ -34,10 +40,10 @@ def get_constant(exported_program: ExportedProgram, node: torch.fx.Node):
     assert isinstance(node, torch.fx.Node)
     if node.name in exported_program.constants:
         return exported_program.constants[node.name]
-    elif node.name in exported_program.graph_signature.inputs_to_buffers:
-        buffer_name = exported_program.graph_signature.inputs_to_buffers[node.name]
-        named_buffer = dict(exported_program.named_buffers())
-        return named_buffer[buffer_name]
+    elif is_buffer(exported_program, node):
+        return get_buffer(exported_program, node)
+    elif is_lifted_tensor_constant(exported_program, node):
+        return get_lifted_tensor_constant(exported_program, node)
     else:
         raise RuntimeError("NYI constant")
 

tico/passes/decompose_fake_quantize_tensor_qparams.py

@@ -18,6 +18,12 @@ if TYPE_CHECKING:
     import torch._ops
     import torch.fx
 import torch
+from torch._export.utils import (
+    get_buffer,
+    get_lifted_tensor_constant,
+    is_buffer,
+    is_lifted_tensor_constant,
+)
 
 # To import torch.ops.quantized_decomposed related operator
 from torch.ao.quantization.fx._decomposed import quantized_decomposed_lib
@@ -55,6 +61,14 @@ def get_constant_from_tensor(
     """
     if isinstance(node, float):
         return node
+    if is_buffer(ep, node):
+        buf = get_buffer(ep, node)
+        assert isinstance(buf, torch.Tensor)
+        return buf.item()
+    elif is_lifted_tensor_constant(ep, node):
+        lifted = get_lifted_tensor_constant(ep, node)
+        assert isinstance(lifted, torch.Tensor)
+        return lifted.item()
     assert isinstance(node.target, torch._ops.OpOverload)
     if node.target.__name__ == "mul.Tensor":
         assert len(node.args) == 2

{tico-0.1.0.dev250518.dist-info → tico-0.1.0.dev250520.dist-info}/METADATA

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

{tico-0.1.0.dev250518.dist-info → tico-0.1.0.dev250520.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tico/__init__.py,sha256=8KpDCl78F5Bkvpxm_fHYEUYWIlJkDsu9cJTyyemrkbs,1181
+tico/__init__.py,sha256=80nv6fiJM89LQkz5EODIec-qHWe6hHtEOZ5u7L1wYnY,1181
 tico/pt2_to_circle.py,sha256=PPmFNw20jw2Z2VyM3ln9pX__jTzBOAZiv0gT5a-p-Y8,2666
 tico/config/__init__.py,sha256=xZzCXjZ84qE-CsBi-dfaL05bqpQ3stKKfTXhnrJRyVs,142
 tico/config/base.py,sha256=anwOiJFkUxUi7Cef573JgQcjk6S-FSi6O_TLjYASW-g,1244
@@ -54,7 +54,7 @@ tico/experimental/quantization/passes/fold_quant_ops.py,sha256=Jq5wmQDhdjsXxae2p
 tico/experimental/quantization/passes/insert_quantize_on_dtype_mismatch.py,sha256=i4rkM1vlN85fXA9oOrU25o8KWAaqA65NKngTX6MgctQ,12960
 tico/experimental/quantization/passes/propagate_qparam_backward.py,sha256=TGtyW0Z2qOTgVIasBdGRgbwH31YYd6ek7OvLTmCV614,3118
 tico/experimental/quantization/passes/propagate_qparam_forward.py,sha256=RhUHGCR2RpBO5KYkQ7Z8U5u7HEwDq2wdKHLKAJCi-5c,5138
-tico/experimental/quantization/passes/remove_weight_dequant_op.py,sha256=V-0tZqLxHbB0Dvw2D69aesfGdr3vmhvFlfNAIYnGTuo,6361
+tico/experimental/quantization/passes/remove_weight_dequant_op.py,sha256=lNemHkr_IMg6kTIQjk4xLgW4DkDNBr0wTW3miNqmvkc,6450
 tico/interpreter/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/interpreter/infer.py,sha256=vJ3b69ce9HrxNT0gFwbEhHpAyvVyuiunTgAeiqn5t64,4350
 tico/interpreter/interpreter.py,sha256=tGbluCbrehTCqBu8mtGDNzby_ieJ2ry8_RH_eC0CQxk,3828
@@ -69,7 +69,7 @@ tico/passes/convert_to_relu6.py,sha256=3sfKfggvjbl9N73pLOwgUTNyoecODsy367nwoX2S-
 tico/passes/decompose_addmm.py,sha256=_yNX7wx1Y9HJI5ksUJI-UQLHpoNawbUbF8kcm2zGHw0,4221
 tico/passes/decompose_batch_norm.py,sha256=d1V9UOkm_5BV0NGLyuQfz4I9NpO7I3ZrRugt7EXM-XM,7016
 tico/passes/decompose_fake_quantize.py,sha256=7ZJyTIDj2iKgWa5q8mBSq6k0GX0vs_XyQdsIiWFJoTU,5175
-tico/passes/decompose_fake_quantize_tensor_qparams.py,sha256=IXJK8pu9V8FRFCdhV1Hk_xfspertqPGmLVr42i3Lk8k,13382
+tico/passes/decompose_fake_quantize_tensor_qparams.py,sha256=kOQaODKl_GCE19h-UZGmxnTcHtlvphI63dVAmMQL_Bk,13823
 tico/passes/decompose_group_norm.py,sha256=UVEzOBi4aUr3IlP6iWlYlo4u-8cmJ4JvKl8r2-qGBj4,9670
 tico/passes/decompose_grouped_conv2d.py,sha256=KJhH6PX7l9k9T8KBV8JDAvaSfJuUnRo_jtvGF2aM-LA,8277
 tico/passes/decompose_slice_scatter.py,sha256=ko9p8v-zY5rOx4aSpWomwSdSWb1lIF32gnU7ik5xgII,5604
@@ -192,9 +192,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.dev250518.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev250518.dist-info/METADATA,sha256=A9UQFzzgACca25PrIho3VrAp6jKE2-LV5T66yayLxOE,8633
-tico-0.1.0.dev250518.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev250518.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev250518.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev250518.dist-info/RECORD,,
+tico-0.1.0.dev250520.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.1.0.dev250520.dist-info/METADATA,sha256=E3Tizp7XTBYhstti5Qh5Q5HYd-7XiVA6zfwUD8T0DII,8633
+tico-0.1.0.dev250520.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tico-0.1.0.dev250520.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
+tico-0.1.0.dev250520.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.1.0.dev250520.dist-info/RECORD,,