tico 0.1.0.dev250721__py3-none-any.whl → 0.1.0.dev250723__py3-none-any.whl

tico/__init__.py

@@ -21,7 +21,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.dev250721"
+__version__ = "0.1.0.dev250723"
 
 MINIMUM_SUPPORTED_VERSION = "2.5.0"
 SECURE_TORCH_VERSION = "2.6.0"

tico/experimental/quantization/algorithm/pt2e/annotation/annotator.py

@@ -21,12 +21,7 @@ if TYPE_CHECKING:
     import torch.fx
     from torch.ao.quantization.qconfig import _ObserverOrFakeQuantizeConstructor
 import torch
-from torch.ao.quantization.observer import (
-    MinMaxObserver,
-    MovingAverageMinMaxObserver,
-    MovingAveragePerChannelMinMaxObserver,
-    PerChannelMinMaxObserver,
-)
+from torch.ao.quantization.observer import MinMaxObserver, PerChannelMinMaxObserver
 from torch.ao.quantization.quantizer import QuantizationSpec, Quantizer
 from torch.ao.quantization.quantizer.utils import _get_module_name_filter
 

tico/experimental/quantization/algorithm/pt2e/annotation/op/linear.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Callable, List, Optional, TYPE_CHECKING
+from typing import Callable, Optional, TYPE_CHECKING
 
 if TYPE_CHECKING:
     import torch.fx

tico/experimental/quantization/algorithm/pt2e/utils.py

@@ -19,7 +19,6 @@ if TYPE_CHECKING:
 import torch
 from torch.ao.quantization.quantizer import QuantizationSpec
 from torch.ao.quantization.quantizer.utils import _get_module_name_filter
-from torch.utils import _pytree as pytree
 
 from tico.experimental.quantization.algorithm.pt2e.annotation.config import (
     QuantizationConfig,

tico/experimental/quantization/algorithm/smoothquant/smooth_quant.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Dict, List, Optional
+from typing import Dict, List, Optional
 
 import torch
 

tico/experimental/quantization/evaluation/evaluate.py

@@ -166,7 +166,7 @@ def evaluate(
         )
     if not isinstance(backend, BACKEND):
         raise RuntimeError(
-            f"Invalid backend. Please use tico.quantization.evaluate.BACKEND enum class"
+            "Invalid backend. Please use tico.quantization.evaluate.BACKEND enum class"
         )
     # Make it a list for simpler logic.
     if input_data is not None:

tico/experimental/quantization/passes/fold_quant_ops.py

@@ -16,7 +16,6 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     import torch.fx
-import copy
 
 import torch
 from torch.export import ExportedProgram

tico/experimental/quantization/passes/insert_quantize_on_dtype_mismatch.py

@@ -19,7 +19,7 @@ if TYPE_CHECKING:
 import copy
 
 from collections import defaultdict
-from typing import Any, Callable
+from typing import Any
 
 import torch
 from torch.export import ExportedProgram

tico/experimental/quantization/passes/quantize_bias.py

@@ -16,7 +16,6 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     import torch.fx
-import copy
 
 import torch
 from torch.export import ExportedProgram

tico/passes/cast_aten_where_arg_type.py

@@ -176,7 +176,7 @@ class CastATenWhereArgType(PassBase):
             node_dtype = extract_torch_dtype(node)
             assert (
                 node_dtype == node_dtype_ori
-            ), f"Type casting doesn't change node's dtype."
+            ), "Type casting doesn't change node's dtype."
 
             logger.debug(
                 f"{to_cast.name}'s dtype was casted from {buf_data.dtype} to {dtype_to_cast}"

tico/passes/cast_mixed_type_args.py

@@ -178,7 +178,7 @@ class CastMixedTypeArgs(PassBase):
             node_dtype = extract_torch_dtype(node)
             assert (
                 node_dtype == node_dtype_ori
-            ), f"Type casting doesn't change node's dtype."
+            ), "Type casting doesn't change node's dtype."
 
         graph.eliminate_dead_code()
         graph.lint()

tico/passes/const_prop_pass.py

@@ -301,7 +301,7 @@ class ConstPropPass(PassBase):
         graph.eliminate_dead_code()
         graph_module.recompile()
 
-        logger.debug(f"Constant nodes are propagated")
+        logger.debug("Constant nodes are propagated")
         # Constant folding can be done with only one time run. Let's set `modified` to False.
         modified = False
         return PassResult(modified)

tico/passes/convert_conv1d_to_conv2d.py

@@ -19,7 +19,7 @@ if TYPE_CHECKING:
 import torch
 from torch.export import ExportedProgram
 
-from tico.serialize.circle_graph import extract_shape
+from tico.serialize.circle_mapping import extract_shape
 from tico.utils import logging
 from tico.utils.errors import NotYetSupportedError
 from tico.utils.graph import create_node

tico/passes/decompose_batch_norm.py

@@ -96,9 +96,9 @@ class DecomposeBatchNorm(PassBase):
             eps = args.eps
 
             if not running_mean:
-                raise NotYetSupportedError(f"running_mean=None is not supported yet")
+                raise NotYetSupportedError("running_mean=None is not supported yet")
             if not running_var:
-                raise NotYetSupportedError(f"running_var=None is not supported yet")
+                raise NotYetSupportedError("running_var=None is not supported yet")
 
             """
             Only support the cases generated from torch.nn.BatchNorm2d module,

tico/passes/decompose_fake_quantize.py

@@ -19,7 +19,6 @@ if TYPE_CHECKING:
 import torch
 
 # To import torch.ops.quantized_decomposed related operator
-from torch.ao.quantization.fx._decomposed import quantized_decomposed_lib
 from torch.export import ExportedProgram
 
 from tico.utils import logging

tico/passes/decompose_fake_quantize_tensor_qparams.py

@@ -26,10 +26,8 @@ from torch._export.utils import (
 )
 
 # To import torch.ops.quantized_decomposed related operator
-from torch.ao.quantization.fx._decomposed import quantized_decomposed_lib
 from torch.export import ExportedProgram
 
-from tico.utils import logging
 from tico.utils.graph import create_node
 from tico.utils.passes import PassBase, PassResult
 from tico.utils.trace_decorators import (

tico/passes/legalize_predefined_layout_operators.py

@@ -20,7 +20,7 @@ if TYPE_CHECKING:
 import torch
 from torch.export import ExportedProgram
 
-from tico.serialize.circle_graph import extract_shape
+from tico.serialize.circle_mapping import extract_shape
 from tico.utils import logging
 from tico.utils.errors import NotYetSupportedError
 from tico.utils.graph import create_node

tico/passes/lower_to_slice.py

@@ -28,7 +28,7 @@ from torch._export.utils import (
 from torch.export import ExportedProgram
 
 from tico.passes import ops
-from tico.serialize.circle_graph import extract_shape
+from tico.serialize.circle_mapping import extract_shape
 from tico.utils import logging
 from tico.utils.graph import create_node, is_single_value_tensor
 from tico.utils.passes import PassBase, PassResult

tico/passes/merge_consecutive_cat.py

@@ -51,7 +51,7 @@ class MergeConsecutiveCat(PassBase):
                 if not prev_cat.op == "call_function":
                     continue
 
-                if not prev_cat.target in ops.aten.cat:
+                if prev_cat.target not in ops.aten.cat:
                     continue
 
                 prev_args = CatArgs(*prev_cat.args, **prev_cat.kwargs)  # type: ignore[arg-type]

tico/passes/remove_redundant_expand.py

@@ -12,11 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    import torch.fx
-import torch
 from torch.export import ExportedProgram
 
 from tico.passes import ops

tico/passes/remove_redundant_reshape.py

@@ -262,7 +262,7 @@ class RemoveRedundantReshapePattern3(PassBase):
                 continue
 
             # add
-            if not add.target in ops.aten.add:
+            if add.target not in ops.aten.add:
                 continue
             add_args = AddTensorArgs(*add.args, **add.kwargs)  # type: ignore[arg-type]
             reshape_2, reshape_3 = add_args.input, add_args.other
@@ -272,7 +272,7 @@ class RemoveRedundantReshapePattern3(PassBase):
             # reshape_2
             if not reshape_2.op == "call_function":
                 continue
-            if not reshape_2.target in ops.aten.reshape:
+            if reshape_2.target not in ops.aten.reshape:
                 continue
             reshape_2_args = ReshapeArgs(*reshape_2.args, **reshape_2.kwargs)  # type: ignore[arg-type]
             reshape_2_input = reshape_2_args.input
@@ -280,7 +280,7 @@ class RemoveRedundantReshapePattern3(PassBase):
             # reshape_3
             if not reshape_3.op == "call_function":
                 continue
-            if not reshape_3.target in ops.aten.reshape:
+            if reshape_3.target not in ops.aten.reshape:
                 continue
             reshape_3_args = ReshapeArgs(*reshape_3.args, **reshape_3.kwargs)  # type: ignore[arg-type]
             reshape_3_input = reshape_3_args.input

tico/passes/segment_index_select.py

@@ -29,7 +29,7 @@ from torch._export.utils import (
 from torch.export import ExportedProgram
 
 from tico.passes import ops
-from tico.serialize.circle_graph import extract_shape
+from tico.serialize.circle_mapping import extract_shape
 from tico.utils import logging
 from tico.utils.graph import add_placeholder, create_node, is_single_value_tensor
 from tico.utils.passes import PassBase, PassResult

tico/serialize/circle_serializer.py

@@ -18,12 +18,7 @@ from typing import Dict
 import flatbuffers
 import torch
 from circle_schema import circle
-from torch.export.exported_program import (
-    ConstantArgument,
-    ExportedProgram,
-    InputKind,
-    TensorArgument,
-)
+from torch.export.exported_program import ConstantArgument, ExportedProgram, InputKind
 
 from tico.serialize.circle_mapping import to_circle_dtype
 from tico.serialize.operators import *
@@ -39,147 +34,38 @@ multiple_output_ops = [
     torch.ops.aten.max.dim,
 ]
 
-# Build circle model from ExportedProgram
-# Return raw bytes of circle model
-def build_circle(edge_program: ExportedProgram) -> bytes:
-    logger = logging.getLogger(__name__)
 
-    builder = flatbuffers.Builder()
+def _initialize_model() -> tuple[CircleModel, CircleSubgraph]:
+    """Initialize a new Circle model and subgraph.
 
-    # Init Model
+    Returns:
+        Tuple containing the model and subgraph
+    """
     model = CircleModel()
-
-    # Add empty buffer at the front (convention)
-    model.add_buffer(circle.Buffer.BufferT())
-
-    # Create an empty subgraph (assume a single subgraph)
+    model.add_buffer(circle.Buffer.BufferT())  # Add empty buffer at the front
     graph = CircleSubgraph(model)
+    return model, graph
 
-    # Export tensors
-    logger.debug("---------------Export tensors--------------")
-    buf_name_to_data = {name: buf for name, buf in edge_program.named_buffers()}
-    for node in edge_program.graph.nodes:
-        if node.op == "call_function":
-            if node.target in multiple_output_ops:
-                continue
-            node_val = node.meta["val"]
-            if node_val.layout != torch.strided:
-                raise RuntimeError(
-                    f"Only support dense tensors (node layout: {node_val.layout})"
-                )
-            graph.add_tensor_from_node(node)
-            logger.debug(f"call_function: {node.name} tensor exported.")
-
-        # placeholder: function input (including parameters, buffers, constant tensors)
-        elif node.op == "placeholder":
-            # placeholder invariants
-            assert node.args is None or len(node.args) == 0  # Not support default param
-
-            # parameters
-            if node.name in edge_program.graph_signature.inputs_to_parameters:
-                param_name = edge_program.graph_signature.inputs_to_parameters[
-                    node.name
-                ]
-                param_data = edge_program.state_dict[param_name]
-
-                assert isinstance(
-                    param_data, torch.Tensor
-                ), "Expect parameters to be a tensor"
-                param_value = param_data.cpu().detach().numpy()
-
-                graph.add_tensor_from_node(node, param_value)
-                logger.debug(f"placeholder(param): {node.name} tensor exported.")
-            elif node.name in edge_program.graph_signature.inputs_to_buffers:
-                buffer_name = edge_program.graph_signature.inputs_to_buffers[node.name]
-                assert buffer_name in buf_name_to_data
-                buffer_data = buf_name_to_data[buffer_name]
-                assert isinstance(
-                    buffer_data, torch.Tensor
-                ), "Expect buffers to be a tensor"
-                buffer_value = buffer_data.cpu().detach().numpy()
-
-                graph.add_tensor_from_node(node, buffer_value)
-                logger.debug(f"placeholder(buffer): {node.name} tensor exported.")
-            elif (
-                node.name
-                in edge_program.graph_signature.inputs_to_lifted_tensor_constants
-            ):
-                ctensor_name = (
-                    edge_program.graph_signature.inputs_to_lifted_tensor_constants[
-                        node.name
-                    ]
-                )
-                ctensor_data = edge_program.constants[ctensor_name]
-
-                assert isinstance(
-                    ctensor_data, torch.Tensor
-                ), "Expect constant tensor to be a tensor"
-                ctensor_value = ctensor_data.cpu().detach().numpy()
-
-                graph.add_tensor_from_node(node, ctensor_value)
-                logger.debug(
-                    f"placeholder(constant tensor): {node.name} tensor exported."
-                )
-            else:
-                user_inputs = [
-                    specs
-                    for specs in edge_program.graph_signature.input_specs
-                    if specs.kind == InputKind.USER_INPUT
-                ]
-                constant_inputs = [
-                    specs
-                    for specs in user_inputs
-                    if isinstance(specs.arg, ConstantArgument)
-                ]
-                name_to_value = {
-                    specs.arg.name: specs.arg.value for specs in constant_inputs
-                }
-                # NoneType ConstantArgument is ignored.
-                if node.name in name_to_value and name_to_value[node.name] == None:
-                    continue
-                graph.add_tensor_from_node(node)
-                logger.debug(f"placeholder: {node.name} tensor exported.")
-
-        # get_attr: retrieve parameter
-        elif node.op == "get_attr":
-            # node.name: Place where fetched attribute is saved
-            # node.target: Attribute in the module
-            attr_tensor = getattr(node.graph.owning_module, node.target)
-            assert isinstance(attr_tensor, torch.Tensor)
 
-            graph.add_tensor_from_scratch(
-                prefix=node.name,
-                shape=list(attr_tensor.shape),
-                dtype=to_circle_dtype(attr_tensor.dtype),
-                source_node=node,
-            )
+def build_circle(ep: ExportedProgram) -> bytes:
+    """Convert ExportedProgram to Circle format.
 
-            logger.debug(f"get_attr: {node.name} tensor exported.")
+    Args:
+        ep: The exported PyTorch program to convert
 
-        # output: function output
-        elif node.op == "output":
-            # output node itself does not need a buffer
-            # argument of output node is assumed to be exported beforehand
-            for output in node.args[0]:
-                if isinstance(output, torch.fx.Node):
-                    assert graph.has_tensor(output.name)
-            continue
-
-        # call_method: call method
-        elif node.op == "call_method":
-            raise AssertionError("Not yet implemented")
-
-        # call_module: call 'forward' of module
-        elif node.op == "call_module":
-            raise AssertionError("Not yet implemented")
+    Returns:
+        bytes: Raw bytes of the Circle model
+    """
+    logger = logging.getLogger(__name__)
+    builder = flatbuffers.Builder()
+    model, graph = _initialize_model()
 
-        else:
-            # Add more if fx.Node is extended
-            raise AssertionError(f"Unknown fx.Node op {node.op}")
+    # Export tensors
+    _export_tensors(graph, ep)
 
     # Register inputs
     logger.debug("---------------Register inputs--------------")
-    for in_spec in edge_program.graph_signature.input_specs:
+    for in_spec in ep.graph_signature.input_specs:
         if in_spec.kind != InputKind.USER_INPUT:
             continue
         # NoneType ConstantArgument is ignored.
@@ -191,9 +77,9 @@ def build_circle(edge_program: ExportedProgram) -> bytes:
 
     # Register outputs
     logger.debug("---------------Register outputs--------------")
-    for user_output in edge_program.graph_signature.user_outputs:
+    for user_output in ep.graph_signature.user_outputs:
         if user_output == None:
-            logger.debug(f"Ignore 'None' output")
+            logger.debug("Ignore 'None' output")
             continue
 
         graph.add_output(user_output)
@@ -203,7 +89,7 @@ def build_circle(edge_program: ExportedProgram) -> bytes:
     logger.debug("---------------Export operators--------------")
     op_codes: Dict[OpCode, int] = {}
     visitors = get_node_visitors(op_codes, graph)
-    for node in edge_program.graph.nodes:
+    for node in ep.graph.nodes:
         if node.op != "call_function":
             continue
 
@@ -227,10 +113,8 @@ def build_circle(edge_program: ExportedProgram) -> bytes:
         code for code, _ in sorted(op_codes.items(), key=lambda x: x[1])
     ]
 
-    # Description
+    # Final model settings
     model.description = "circle"
-
-    # Set version
     model.version = 0
 
     # Finish model
@@ -238,3 +122,212 @@ def build_circle(edge_program: ExportedProgram) -> bytes:
     buf = builder.Output()
 
     return bytes(buf)
+
+
+def _export_tensors(graph: CircleSubgraph, ep: ExportedProgram) -> None:
+    """Export all tensors from the exported program to the circle graph.
+
+    Args:
+        graph: The CircleSubgraph to add tensors to
+        ep: The exported PyTorch program
+    """
+    logger = logging.getLogger(__name__)
+    logger.debug("---------------Export tensors--------------")
+    buf_name_to_data = {name: buf for name, buf in ep.named_buffers()}
+
+    for node in ep.graph.nodes:
+        if node.op == "call_function":
+            if node.target in multiple_output_ops:
+                continue
+            node_val = node.meta["val"]
+            if node_val.layout != torch.strided:
+                raise RuntimeError(
+                    f"Only support dense tensors (node layout: {node_val.layout})"
+                )
+            graph.add_tensor_from_node(node)
+            logger.debug(f"call_function: {node.name} tensor exported.")
+
+        elif node.op == "placeholder":
+            _handle_placeholder_node(graph, node, ep, buf_name_to_data)
+
+        elif node.op == "get_attr":
+            _handle_get_attr_node(graph, node)
+
+        elif node.op == "output":
+            for output in node.args[0]:
+                if isinstance(output, torch.fx.Node):
+                    assert graph.has_tensor(output.name)
+            continue
+
+        elif node.op == "call_method":
+            raise AssertionError("Not yet implemented")
+
+        elif node.op == "call_module":
+            raise AssertionError("Not yet implemented")
+
+        else:
+            raise AssertionError(f"Unknown fx.Node op {node.op}")
+
+
+def _handle_placeholder_node(
+    graph: CircleSubgraph,
+    node: torch.fx.Node,
+    ep: ExportedProgram,
+    buf_name_to_data: dict,
+) -> None:
+    """Handle a placeholder node during tensor export."""
+    # placeholder invariants
+    assert node.args is None or len(node.args) == 0  # Not support default param
+
+    if node.name in ep.graph_signature.inputs_to_parameters:
+        _handle_parameter_node(graph, node, ep)
+    elif node.name in ep.graph_signature.inputs_to_buffers:
+        _handle_buffer_node(graph, node, ep, buf_name_to_data)
+    elif node.name in ep.graph_signature.inputs_to_lifted_tensor_constants:
+        _handle_constant_tensor_node(graph, node, ep)
+    else:
+        _handle_user_input_node(graph, node, ep)
+
+
+def _handle_parameter_node(
+    graph: CircleSubgraph,
+    node: torch.fx.Node,
+    ep: ExportedProgram,
+) -> None:
+    """Handle a parameter placeholder node by exporting its tensor data.
+
+    Args:
+        graph: CircleSubgraph to add tensor to
+        node: The parameter node to process
+        ep: ExportedProgram containing parameter data
+    """
+    param_name = ep.graph_signature.inputs_to_parameters[node.name]
+    param_data = ep.state_dict[param_name]
+
+    if not isinstance(param_data, torch.Tensor):
+        raise ValueError(f"Parameter {param_name} is not a tensor")
+
+    tensor_value = param_data.cpu().detach().numpy()
+    graph.add_tensor_from_node(node, tensor_value)
+
+    logger = logging.getLogger(__name__)
+    logger.debug(f"Exported parameter tensor: {node.name}")
+
+
+def _handle_buffer_node(
+    graph: CircleSubgraph,
+    node: torch.fx.Node,
+    ep: ExportedProgram,
+    buf_name_to_data: dict,
+) -> None:
+    """Handle a buffer placeholder node by exporting its tensor data.
+
+    Args:
+        graph: CircleSubgraph to add tensor to
+        node: The buffer node to process
+        ep: ExportedProgram containing buffer info
+        buf_name_to_data: Mapping of buffer names to data
+    """
+    buffer_name = ep.graph_signature.inputs_to_buffers[node.name]
+
+    if buffer_name not in buf_name_to_data:
+        raise ValueError(f"Buffer {buffer_name} not found in buffer data")
+
+    buffer_data = buf_name_to_data[buffer_name]
+
+    if not isinstance(buffer_data, torch.Tensor):
+        raise ValueError(f"Buffer {buffer_name} is not a tensor")
+
+    tensor_value = buffer_data.cpu().detach().numpy()
+    graph.add_tensor_from_node(node, tensor_value)
+
+    logger = logging.getLogger(__name__)
+    logger.debug(f"Exported buffer tensor: {node.name}")
+
+
+def _handle_constant_tensor_node(
+    graph: CircleSubgraph,
+    node: torch.fx.Node,
+    ep: ExportedProgram,
+) -> None:
+    """Handle a constant tensor placeholder node by exporting its tensor data.
+
+    Args:
+        graph: CircleSubgraph to add tensor to
+        node: The constant tensor node to process
+        ep: ExportedProgram containing constant data
+    """
+    ctensor_name = ep.graph_signature.inputs_to_lifted_tensor_constants[node.name]
+
+    if ctensor_name not in ep.constants:
+        raise ValueError(f"Constant tensor {ctensor_name} not found")
+
+    ctensor_data = ep.constants[ctensor_name]
+
+    if not isinstance(ctensor_data, torch.Tensor):
+        raise ValueError(f"Constant tensor {ctensor_name} is not a tensor")
+
+    tensor_value = ctensor_data.cpu().detach().numpy()
+    graph.add_tensor_from_node(node, tensor_value)
+
+    logger = logging.getLogger(__name__)
+    logger.debug(f"Exported constant tensor: {node.name}")
+
+
+def _handle_user_input_node(
+    graph: CircleSubgraph,
+    node: torch.fx.Node,
+    ep: ExportedProgram,
+) -> None:
+    """Handle a user input placeholder node by exporting its tensor data.
+
+    Args:
+        graph: CircleSubgraph to add tensor to
+        node: The user input node to process
+        ep: ExportedProgram containing input specs
+    """
+    user_inputs = [
+        specs
+        for specs in ep.graph_signature.input_specs
+        if specs.kind == InputKind.USER_INPUT
+    ]
+    constant_inputs = [
+        specs for specs in user_inputs if isinstance(specs.arg, ConstantArgument)
+    ]
+    name_to_value = {specs.arg.name: specs.arg.value for specs in constant_inputs}
+
+    # Skip NoneType ConstantArgument
+    if node.name in name_to_value and name_to_value[node.name] is None:
+        return
+
+    graph.add_tensor_from_node(node)
+
+    logger = logging.getLogger(__name__)
+    logger.debug(f"Exported user input tensor: {node.name}")
+
+
+def _handle_get_attr_node(
+    graph: CircleSubgraph,
+    node: torch.fx.Node,
+) -> None:
+    """Handle a get_attr node by exporting its tensor data.
+
+    Args:
+        graph: CircleSubgraph to add tensor to
+        node: The get_attr node to process
+    """
+    assert isinstance(node.target, str)
+    attr_tensor = getattr(node.graph.owning_module, node.target)
+
+    if not isinstance(attr_tensor, torch.Tensor):
+        raise ValueError(f"Attribute {node.target} is not a tensor")
+
+    graph.add_tensor_from_scratch(
+        prefix=node.name,
+        shape=list(attr_tensor.shape),
+        dtype=to_circle_dtype(attr_tensor.dtype),
+        source_node=node,
+    )
+
+    logger = logging.getLogger(__name__)
+    logger.debug(f"Exported attribute tensor: {node.name}")

tico/serialize/operators/op_any.py

@@ -22,7 +22,6 @@ from circle_schema import circle
 from tico.serialize.circle_graph import CircleSubgraph
 from tico.serialize.circle_mapping import (
     circle_legalize_dtype_to,
-    extract_circle_dtype,
     extract_shape,
     extract_torch_dtype,
 )

tico/serialize/operators/op_clamp.py

@@ -21,12 +21,9 @@ import torch
 from circle_schema import circle
 
 from tico.passes import ops
+from tico.serialize.circle_graph import CircleSubgraph
 
-from tico.serialize.circle_graph import (
-    CircleSubgraph,
-    extract_circle_dtype,
-    extract_shape,
-)
+from tico.serialize.circle_mapping import extract_circle_dtype, extract_shape
 from tico.serialize.operators.hashable_opcode import OpCode
 from tico.serialize.operators.node_visitor import NodeVisitor, register_node_visitor
 from tico.serialize.operators.utils import create_builtin_operator, get_op_index

tico/serialize/operators/op_full_like.py

@@ -21,10 +21,8 @@ import torch
 from circle_schema import circle
 
 from tico.serialize.circle_graph import CircleSubgraph
-from tico.serialize.circle_mapping import to_circle_dtype
 from tico.serialize.operators.hashable_opcode import OpCode
 from tico.serialize.operators.node_visitor import NodeVisitor, register_node_visitor
-from tico.serialize.operators.utils import create_builtin_operator, get_op_index
 from tico.utils.validate_args_kwargs import FullLikeArgs
 
 

tico/utils/convert.py

@@ -157,7 +157,7 @@ def check_unsupported_target(exported_program: ExportedProgram):
     for n in exported_program.graph.nodes:
         if n.op != "call_function":
             continue
-        if not n.target in supported_target:
+        if n.target not in supported_target:
             unsupported.append(n)
 
     if unsupported:

tico/utils/graph.py

@@ -24,7 +24,7 @@ import torch
 from torch.export import ExportedProgram
 from torch.export.exported_program import InputKind, InputSpec, TensorArgument
 
-from tico.utils.utils import get_fake_mode, set_new_meta_val
+from tico.utils.utils import get_fake_mode
 
 
 def is_torch_param(node: torch.fx.Node, ep: ExportedProgram):

tico/utils/padding.py

@@ -15,8 +15,6 @@
 from enum import IntEnum
 from typing import NamedTuple, Optional, Sequence, Tuple, Union
 
-import torch
-
 from tico.utils.errors import InvalidArgumentError
 
 

tico/utils/serialize.py

@@ -12,9 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Optional
-
-import torch
 
 from tico.serialize.circle_graph import CircleSubgraph
 from tico.utils.graph import get_module_name_chain

tico/utils/utils.py

@@ -21,7 +21,6 @@ from typing import List
 
 import torch
 from circle_schema import circle
-from packaging.version import Version
 from torch._guards import detect_fake_mode
 from torch.export import ExportedProgram
 from torch.utils import _pytree as pytree

tico/utils/validate_args_kwargs.py

@@ -16,10 +16,8 @@ from dataclasses import dataclass, field
 from typing import List, Optional, TYPE_CHECKING, Union
 
 if TYPE_CHECKING:
-    import torch._ops
     import torch.fx
 import torch
-import torch.fx.node
 
 from tico.utils.utils import enforce_type
 
@@ -137,7 +135,7 @@ class AvgPool2dArgs:
             assert len(self.padding) == 2, len(self.padding)
         if self.divisor_override is not None:
             assert isinstance(self.divisor_override, int), type(self.divisor_override)
-            assert self.divisor_override != 0, f"Divisor must be not zero."
+            assert self.divisor_override != 0, "Divisor must be not zero."
 
 
 @enforce_type

{tico-0.1.0.dev250721.dist-info → tico-0.1.0.dev250723.dist-info}/METADATA

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

{tico-0.1.0.dev250721.dist-info → tico-0.1.0.dev250723.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tico/__init__.py,sha256=QY54qph93oHLtUUYMg8T_e_Bpn4Zmxfif9POfAGssHI,1743
+tico/__init__.py,sha256=e1nx3N3UdH9Ok1YI1Q0qGVr_5_AqoG4ei8NnoEsxeuk,1743
 tico/pt2_to_circle.py,sha256=gu3MD4Iqc0zMZcCZ2IT8oGbyj21CTSbT3Rgd9s2B_9A,2767
 tico/config/__init__.py,sha256=xZzCXjZ84qE-CsBi-dfaL05bqpQ3stKKfTXhnrJRyVs,142
 tico/config/base.py,sha256=anwOiJFkUxUi7Cef573JgQcjk6S-FSi6O_TLjYASW-g,1244
@@ -17,9 +17,9 @@ tico/experimental/quantization/algorithm/gptq/quantizer.py,sha256=icaFDXA1UibgRI
 tico/experimental/quantization/algorithm/gptq/utils.py,sha256=vDIW5ow5c1VSFpub7QumMWorHrV86c0kOtlBxMw2Y2Y,1808
 tico/experimental/quantization/algorithm/pt2e/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/experimental/quantization/algorithm/pt2e/quantizer.py,sha256=mdTvsG87bo8fu0GaWqSM8iBCs-4f4EfUlVtk-Ko6M34,2546
-tico/experimental/quantization/algorithm/pt2e/utils.py,sha256=aQeJSp9Aul8smkHHRgYGiYu58xqEbct2UpEPMF8vq8Y,4848
+tico/experimental/quantization/algorithm/pt2e/utils.py,sha256=URjTGgsnDdhUC2Nr0-YJ9GWbVOKmjElfLr83Y8eCz-M,4806
 tico/experimental/quantization/algorithm/pt2e/annotation/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/experimental/quantization/algorithm/pt2e/annotation/annotator.py,sha256=szUSODiOUtUfCEY2fdCav5dkFDrTGHlECIcUv2o_P7c,7640
+tico/experimental/quantization/algorithm/pt2e/annotation/annotator.py,sha256=lFfblxglPxcN2IcrjAVYq7GOECIAQ4rr7M4euPp3yWc,7551
 tico/experimental/quantization/algorithm/pt2e/annotation/config.py,sha256=8x6YvixIogXh_PwqhKNb3L6lJyBDFQ_OHotonG_hWD0,977
 tico/experimental/quantization/algorithm/pt2e/annotation/spec.py,sha256=6CXHmeTbl8pzpHmrj9-bqIHMhHVGqPy7sYxyt-kWkKA,1437
 tico/experimental/quantization/algorithm/pt2e/annotation/utils.py,sha256=oBf4vjFzVQ3jfk1N-6S7A_3woO27QIc8zD4fM8OAZss,3161
@@ -28,7 +28,7 @@ tico/experimental/quantization/algorithm/pt2e/annotation/op/adaptive_avg_pool2d.
 tico/experimental/quantization/algorithm/pt2e/annotation/op/add.py,sha256=cYsuGBiEmogkx5v_LtpzDKVf_YPHukpHPEjV4D1Hvbk,2214
 tico/experimental/quantization/algorithm/pt2e/annotation/op/conv2d.py,sha256=jaQ72qzM82AN1JKgdYslatmQ2ZGzLwUrbyVIMC7HYwY,3692
 tico/experimental/quantization/algorithm/pt2e/annotation/op/div.py,sha256=uPKorN80d3yFOmIDI6fK7uQ4eyqehgkoGnyCCZ_pfsg,2196
-tico/experimental/quantization/algorithm/pt2e/annotation/op/linear.py,sha256=hkqOXqkygXRoymU_tTt3GKGiSJP-jKx9ygXAC-XlYGg,3497
+tico/experimental/quantization/algorithm/pt2e/annotation/op/linear.py,sha256=u84jvgf2i-lia3gqdmHrbqXS_hTWDxU1gio91E0JjG8,3491
 tico/experimental/quantization/algorithm/pt2e/annotation/op/mean.py,sha256=gXZ7LQWujO6iN5ZjsV3emBSJJCt1VuhUnlnr7rJmpbU,2020
 tico/experimental/quantization/algorithm/pt2e/annotation/op/mul.py,sha256=MPfk2aPBqX6W4TuOI_ZElrU_hAc-9OtClOxkXvH6tsA,2211
 tico/experimental/quantization/algorithm/pt2e/annotation/op/relu6.py,sha256=UDWVX7xMlIfuyHlGVAJMAxRCC2C6ldfYqRwluChhhew,2017
@@ -39,10 +39,10 @@ tico/experimental/quantization/algorithm/pt2e/transformation/convert_scalars_to_
 tico/experimental/quantization/algorithm/smoothquant/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/experimental/quantization/algorithm/smoothquant/observer.py,sha256=yi9nR_BEuxKVjgFcYPeldhXlEbE9V-0r4kRRPcI3C70,2639
 tico/experimental/quantization/algorithm/smoothquant/quantizer.py,sha256=rQMtnqM1dBzjhY-KJRp3TWZSW-dzXrs5N5FagXzh0IQ,2564
-tico/experimental/quantization/algorithm/smoothquant/smooth_quant.py,sha256=uaMvdFzWUxUllhXfIEgCxYi1glXspICFCar4d4KyDvc,6180
+tico/experimental/quantization/algorithm/smoothquant/smooth_quant.py,sha256=O1h7IojcsJaprFvRM9tsAuR3q7vTjKKRi88jD-nH79Y,6175
 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=DvAbvECeTUcEsiHUhscqEMC2msPvzbz3AD6LfjIt1n8,7989
+tico/experimental/quantization/evaluation/evaluate.py,sha256=cI9_4cbJKeYMmDt_PD85MHEseHGhwYRr3RHsKdix1FI,7988
 tico/experimental/quantization/evaluation/metric.py,sha256=Ae-vAkA8n6T9_aGv47rkz1MLCw9ji1oDm1Rdd_MIxNQ,3744
 tico/experimental/quantization/evaluation/utils.py,sha256=82RG_e5LuKfWo786wEZUVwXY93nNl901n04fB7D0Z6k,5909
 tico/experimental/quantization/evaluation/executor/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
@@ -50,28 +50,28 @@ tico/experimental/quantization/evaluation/executor/backend_executor.py,sha256=3k
 tico/experimental/quantization/evaluation/executor/circle_executor.py,sha256=eCCJ9wTwR0vUJ0oN7jxtQxZ9598GRw6P6KUxiuGsIIM,2685
 tico/experimental/quantization/evaluation/executor/triv24_executor.py,sha256=sUoXl6oOO2arAKaNjOBg7HiQja145_Jv6qgY7XtR7A8,5159
 tico/experimental/quantization/passes/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/experimental/quantization/passes/fold_quant_ops.py,sha256=iaBMyO49CwVkhebMz3rjkHWfWE2LhwH6fORe7n4S6XQ,7040
-tico/experimental/quantization/passes/insert_quantize_on_dtype_mismatch.py,sha256=t3bnNY9Abm8CZfSWzsbvx59luylXVxqmUvCKPBVPAIE,14731
+tico/experimental/quantization/passes/fold_quant_ops.py,sha256=bRHYSeHdSTaz3261skIkK5Aso2Lbv7ql0zFI9ICmbDY,7028
+tico/experimental/quantization/passes/insert_quantize_on_dtype_mismatch.py,sha256=4xwOzUrnMH4MQfdOHBHHXQo1EyWSkqNNVfBg47J5UCg,14721
 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/quantize_bias.py,sha256=ZQ3rETYStpW28JUbODRixbq5sDEOiIOB_qWA-Jzuu-Y,4337
+tico/experimental/quantization/passes/quantize_bias.py,sha256=T7YxJ70N0tSK0FF9VJZA5iP0sHdnnsX9GX4AT4JDFSk,4325
 tico/experimental/quantization/passes/remove_weight_dequant_op.py,sha256=Klc_9-94tl0_AuAToKOjsWED_YPk5RB67eum0ddPX7o,6588
 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
 tico/passes/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/passes/cast_aten_where_arg_type.py,sha256=ybtGj1L7_2zGyfb_G-_y1N1mRgKHVq6fBZc-9-fH9sA,7229
+tico/passes/cast_aten_where_arg_type.py,sha256=QOaet85z23ad3s9c8md5r11q9dEw4-lfJqlpM7aiBic,7228
 tico/passes/cast_clamp_mixed_type_args.py,sha256=m3_HpXLywWmWERfE5lM5PgvjBod7C4BWu_Q-TkRyO8k,5387
-tico/passes/cast_mixed_type_args.py,sha256=ArJpIPnQP1LPNaWIwee13Nbj749_awFKDO-pAYvdYvI,7618
-tico/passes/const_prop_pass.py,sha256=QOeR2u3fo9ZhWXRhfAUW1dTtuWgqgoqdDJoQ516UDbQ,11532
-tico/passes/convert_conv1d_to_conv2d.py,sha256=7YljWJQBX5vBUMgGgRv8TvbJ9UpEL9hf4ZU3dNUhEZ8,5301
+tico/passes/cast_mixed_type_args.py,sha256=gsDJfqxVE2dqQnLjSEsPWWs9FGXT-tIyMzjLhxfktKA,7617
+tico/passes/const_prop_pass.py,sha256=hDxGgJNiRjsgOArdaoeAOcOOA-nKBvA1W1zcMZQA5yg,11531
+tico/passes/convert_conv1d_to_conv2d.py,sha256=ktS3h158y9rg1sQiW8BZZbflV_dk_UdjBPQnuiOKyzg,5303
 tico/passes/convert_layout_op_to_reshape.py,sha256=sCAFjkmVtiKjvDQSAgnjNBHl3_hWXJZElGDXQiTH-7s,2963
 tico/passes/convert_repeat_to_expand_copy.py,sha256=JbtFTmWyfJS2SSd_higP1IEhQeh7wHdN5dmTbbiFVCs,3237
 tico/passes/convert_to_relu6.py,sha256=1BJpUwUb6Zli_1y3eyJQo7dg9B1xvZ7sYjMbvEQsFJM,6442
 tico/passes/decompose_addmm.py,sha256=0IUZjRS8G-h7LPeYu27gxWyTKi3pKAF1uNU1y-mHLL0,4056
-tico/passes/decompose_batch_norm.py,sha256=hWR6PMqgzMuNME7oLiNaxDJl5ZwXZeqxSkcNEMW4vAE,6514
-tico/passes/decompose_fake_quantize.py,sha256=Y29HncMwgRffwmzl5-iua8MhWLd9zEAUWqlNhGhmoHg,5368
-tico/passes/decompose_fake_quantize_tensor_qparams.py,sha256=y519moei8WeEiSv24TnPoqy0i7xXc6j34rT-EykajCM,14020
+tico/passes/decompose_batch_norm.py,sha256=06LAxhSmpTxFZJmUelwB3I_GipNWrLoM7PfM6ZkxOZY,6512
+tico/passes/decompose_fake_quantize.py,sha256=dvXZ5LyjrbpgvasmcBgjk2-xJyijefPRoNXBjFmAkRI,5294
+tico/passes/decompose_fake_quantize_tensor_qparams.py,sha256=k9MJhMVABFNF6lXgEum1fJyGpdQwVRKxWOYhkMR2M7c,13915
 tico/passes/decompose_group_norm.py,sha256=rtjv3PrcFTtZ68uCsmE2LibnWaQU-e8NdKkblL4LqxE,10197
 tico/passes/decompose_grouped_conv2d.py,sha256=n2qv320akL1ju33ucZ6lU1cKEAaj0NI8YZ5CrUnkRLM,8512
 tico/passes/decompose_slice_scatter.py,sha256=xqMHKhW2595YoAeubKZ4jRhYW4TQ09EXPgLNgODqXG8,5653
@@ -80,25 +80,25 @@ tico/passes/fill_meta_val.py,sha256=Xbam6Aq90ZfWItZw1dgLIwH_q8RCiU5JodKNqkj-ink,
 tico/passes/fuse_leading_unsqueeze_reshape.py,sha256=88jwTP35yRyXOk9xdO6YW2OEfdKAws3KFRT16WQz0RI,4291
 tico/passes/fuse_redundant_reshape_to_mean.py,sha256=GhJS1ZKB6Ns4AhwcW3uUQ6q-0N-AzlD32B2EwusUJHg,3761
 tico/passes/legalize_causal_mask_value.py,sha256=0nfUKGd7XSe9Hg5TAi4dUi6Nn6-JRTWCwhULR5AEgqs,4079
-tico/passes/legalize_predefined_layout_operators.py,sha256=MNx7L2dAlsxSazb-F7c0onPqHleI17zAc7AzQAa9aJ4,18934
+tico/passes/legalize_predefined_layout_operators.py,sha256=MdpI47_AI6PDqxWHGlJCsjbg_qxJhDzMhTl1fWn8I04,18936
 tico/passes/lower_pow2_to_mul.py,sha256=nfJXa9ZTZMiLg6ownSyvkM4KF2z9tZW34Q3CCWI_vmQ,2402
 tico/passes/lower_to_resize_nearest_neighbor.py,sha256=N6F56Of8Aiv-KIiYLHnh33WX72W60ZVQSBEYWHdYqNQ,9005
-tico/passes/lower_to_slice.py,sha256=0qAX3WzZdyMFDW4DiO9b5JFXd4rL1-0doBT6lJvaw_I,7260
-tico/passes/merge_consecutive_cat.py,sha256=BYmiU170DsrHQMj7gMe7U6ZpndrX-S4OpvJweDdspec,2701
+tico/passes/lower_to_slice.py,sha256=OzlFzK3lBYyYwC3WThsWd94Ob4JINIJF8UaLAtnumzU,7262
+tico/passes/merge_consecutive_cat.py,sha256=ayZNLDA1DFM7Fxxi2Dmk1CujkgUuaVCH1rhQgLrvvOQ,2701
 tico/passes/ops.py,sha256=cSj3Sk2x2cOE9b8oU5pmSa_rHr-iX2lORzu3N_UHMSQ,2967
 tico/passes/remove_nop.py,sha256=Hf91p_EJAOC6DyWNthash0_UWtEcNc_M7znamQfYQ5Y,2686
 tico/passes/remove_redundant_assert_nodes.py,sha256=IONd3xBy6I8tH6_Y1eN3_eCHH7WTC8soBgjXzOju9cQ,1612
-tico/passes/remove_redundant_expand.py,sha256=5SIqN7eIIcqF68tlrB31n1482jSBSBOgKb1wddLX6lw,2197
+tico/passes/remove_redundant_expand.py,sha256=auyqIoQT4HJhiJfuUe6BrEtUhvz221ohnIK5EuszWeg,2112
 tico/passes/remove_redundant_permute.py,sha256=98UsaZzFZdQzEEAR1pIzRisAf6hgfXLa88aayjalt3E,4292
-tico/passes/remove_redundant_reshape.py,sha256=ScLYTShXMXJBTzOByAEhX-qJe5pmu92pLsXv5mh7u5c,16454
+tico/passes/remove_redundant_reshape.py,sha256=OnhU7FKpudGEaF6ufsMvkzf9hn84FNmNKk37M7FN3Pc,16454
 tico/passes/remove_redundant_slice.py,sha256=Iv7TbB39fktNb4eq0VdyZnwxL_VsKLJ90diMmaf3kZk,2087
 tico/passes/remove_redundant_to_copy.py,sha256=tKy4XKkO2l33fMxVPQ_iFkUeFvP15kbPvzPPhT_g0c8,3292
 tico/passes/restore_linear.py,sha256=xGJdNb-1CrkOKS9BnLbcblkZc6P2vVjKIi-7lRcs7Bk,4111
-tico/passes/segment_index_select.py,sha256=jn0M2sdUcDyjrvxfvM40wt5644iPQMY_ud0uvptXN84,5187
+tico/passes/segment_index_select.py,sha256=VVCKNLtYRkr9n5lGnlzEuQsQ0WVxEYXGchFrDnB1C40,5189
 tico/serialize/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/serialize/circle_graph.py,sha256=3t78g5eKzhHKvIBJqQ-CcwbqoV-2QwAdd_8wm4W1yXw,12317
 tico/serialize/circle_mapping.py,sha256=C9C3ORACQOdvBdnt5KRzlT8zao_TvzQklIxH794OhP0,5719
-tico/serialize/circle_serializer.py,sha256=KRx_Azx2Je9XNYe-pZuuiSMvbXEddd8M8qDATIt7XXk,8981
+tico/serialize/circle_serializer.py,sha256=eb90eTgU6WAZEWPclZIkInJsqAP1KjRSF2QeY5F3c3E,10766
 tico/serialize/pack.py,sha256=5HZ9kX3x6C6CyT_FWS6FRmvx_P7Dx21orjUNQxJ2xlo,1297
 tico/serialize/quant_param.py,sha256=s97GJyDOZULnqFUWPakHais31G_qqPuO0awPHCkZDvI,1342
 tico/serialize/operators/__init__.py,sha256=LIvXsNnN4yUCS2CGNQ5XW8p8oXDTV_WHWuOEAw1t6WY,990
@@ -107,13 +107,13 @@ tico/serialize/operators/node_visitor.py,sha256=UYyCwXqSCeRyimThMShstHnt7vKM9tsu
 tico/serialize/operators/op_abs.py,sha256=Y-vy7rcqPT-qD3QS5R8zbApWWTPpjY6xuMMVnbIhYmQ,1827
 tico/serialize/operators/op_add.py,sha256=otm062DMHVAThWmOtSTZdPyP3P5-2cv5VL_UWBJeLms,2346
 tico/serialize/operators/op_alias_copy.py,sha256=Xu9OiILbGf8oddh8yTqovvLfgVs8XYV7Cg4n6CesWcg,2175
-tico/serialize/operators/op_any.py,sha256=9oxP-8vS5R4oKX6KaePygzC4-jh8MVgbiS8Z5AWYOAw,5237
+tico/serialize/operators/op_any.py,sha256=vZ37qzSEAOVPjfakSkgPDym91S9kOMUpLisAqkP-3Po,5211
 tico/serialize/operators/op_arange_start_step.py,sha256=0T5lWwh3TfsFStmVv0v5qG03KENRDBmMix08RXQ4D-U,2132
 tico/serialize/operators/op_argmax.py,sha256=ARyGHlmWVmzwCct93V5x1-VyKqhxMOvV8GuM8yQWXdo,2290
 tico/serialize/operators/op_avg_pool2d.py,sha256=vc7WCakGXtGFPV1ix5EJmboH23tQ-cSI36ePY3PHKI4,7544
 tico/serialize/operators/op_bmm.py,sha256=AELjHC9ISFPIzEEl5Kr1s4GSNLZElwZmVZJWkEyCEoA,2189
 tico/serialize/operators/op_cat.py,sha256=XDYOh0XAyrM0TlxVm6Sa0OFFGrKk7aSDcGXC-hYX4gs,2204
-tico/serialize/operators/op_clamp.py,sha256=ZRAsXLGsZqJEh4wXxESEpRJkRtUuJWTDgAem6lr9_5I,4298
+tico/serialize/operators/op_clamp.py,sha256=6iDFZZTMkEqJ8lpNcEhKUpAIc4h2oPkBCBWnnt4lTDw,4322
 tico/serialize/operators/op_clone.py,sha256=vzDYJ8TS3tc2BAyd_z8nt5VqT1inpymSseMEhd9dva0,2394
 tico/serialize/operators/op_constant_pad_nd.py,sha256=OpP4AP-d1IFcWZolNa-o9ZxzXJQkMdG9WQ66soX3s-E,2675
 tico/serialize/operators/op_conv2d.py,sha256=UfYk5xnA9PqVYyjU9dUCSW0CiCmcEK3LnlnFh0WY4Gg,6599
@@ -129,7 +129,7 @@ tico/serialize/operators/op_eq.py,sha256=g17_K6IkWvnop_LaiUJzcGPUSFElz6UUrf7T0bo
 tico/serialize/operators/op_exp.py,sha256=WWCXRqpo8E5uIq2J8saJO5e_Sny5NhVThUFxgTZQmEs,2047
 tico/serialize/operators/op_expand.py,sha256=Wc-FK-Je2nDW1x3g9kjSES3jxAhM5m7W2Irw1pt3BgE,3547
 tico/serialize/operators/op_full.py,sha256=1Hv-H829F4gj5ZklrEb2wwKB9wSHBkc6aFS08jYFxuw,1679
-tico/serialize/operators/op_full_like.py,sha256=XNxojtsGftURQs_RWKB17q24X_L1QOLYV6DGOI5S7n8,2010
+tico/serialize/operators/op_full_like.py,sha256=T4DomZTy4vyzYysGw7A0UrWbkV8EJ53whbtuUUuHy_I,1871
 tico/serialize/operators/op_ge.py,sha256=TrgZ6wbIEYkDGfVFNtDlfM7ZkMMWjvcks5U5DankSbo,1892
 tico/serialize/operators/op_gelu.py,sha256=bS8-0rg5_bT__OI3mBDywxGx4xTO2Iqea3h-uC17MpU,2145
 tico/serialize/operators/op_gt.py,sha256=JAVbtuAUNLYhtJycJJCEkYo9QAvmiK4lTMdw5yHUd10,1886
@@ -183,31 +183,31 @@ tico/serialize/operators/op_view.py,sha256=5EMww-ve17Vm9XPuV03Tn7vJsjpU2J8U4d_FO
 tico/serialize/operators/op_where.py,sha256=doE81GSwygrPBm3JIfN9w7kKXxeIYKxgk0eoY22QIcg,2845
 tico/serialize/operators/utils.py,sha256=lXGpEJW1h8U_-gfc6EWjvvSiq3yJ9P-v1v3EMRT_pSk,2954
 tico/utils/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
-tico/utils/convert.py,sha256=lNbbNswbKyCTK5E5i5CXkJBykWWfF5HDChhM3DLscWo,13222
+tico/utils/convert.py,sha256=GgZwZtiqFzTdszfUQO0vcX39lKjs97gYwZ-Tiw_4Bbo,13222
 tico/utils/define.py,sha256=Ypgp7YffM4pgPl4Zh6TmogSn1OxGBMRw_e09qYGflZk,1467
 tico/utils/diff_graph.py,sha256=_eDGGPDPYQD4b--MXX0DLoVgSt_wLfNPt47UlolLLR4,5272
 tico/utils/dtype.py,sha256=4-k1iUaHivFFXAQuDs7up6fXt5y4FqldGNokAPa3kic,603
 tico/utils/errors.py,sha256=f3csJjgbXG9W1aHhqEcou008Aor19W57X8oT5Hx8w1M,954
-tico/utils/graph.py,sha256=Y6aODsnc_-9l61oanknb7K1jqJ8B35iPypOKkM0Qkk0,9149
+tico/utils/graph.py,sha256=jD6m58m5JmN9mPfaROA9CW3406iJxmnukke00AuwRqI,9131
 tico/utils/installed_packages.py,sha256=J0FTwnkCGs0MxRWoCMYAqiwH7Z0GWFDLV--x-IndSp4,1017
 tico/utils/logging.py,sha256=IlbBWscsaHidI0dNqro1HEXAbIcbkR3BD5ukLy2m95k,1286
 tico/utils/model.py,sha256=pPOIjD0qjQirLibiRxxfjOR6efimOcDAd9R-74eus-k,1282
-tico/utils/padding.py,sha256=jyNhGmlLZfruWZ6n5hll8RZOFg85iCZP8OJqnHGS97g,3293
+tico/utils/padding.py,sha256=0iEcS5G3gSFySPzBz1m5nxHYZ6MlLzG_KdAFJRauwIg,3279
 tico/utils/passes.py,sha256=kGmDe__5cPaO6i5EDAoXSVe6yXEoX9hAny4ROb3ZEmQ,2409
 tico/utils/pytree_utils.py,sha256=jrk3N6X6LiUnBCX_gM1K9nywbVAJBVnszlTAgeIeDUc,5219
 tico/utils/register_custom_op.py,sha256=3-Yl6iYmx1qQA2igNHt4hYhQhQMkdPb7gF50LIY8yvc,27350
-tico/utils/serialize.py,sha256=AQXMBOLu-Kg2Rn-qbqsAtHndjZAZIavlKA0QFgJREHM,1420
+tico/utils/serialize.py,sha256=cBtEUfi_SU_9_v0cq2CNikzn8GnzEz2RwRvUH2NkWu4,1378
 tico/utils/torch_compat.py,sha256=oc6PztVsXdHcQ3iaVR90wLLxrGaj6zFHWZ8K9rRS6q8,1795
 tico/utils/trace_decorators.py,sha256=ddLIiKQfSaQrxgF1kNpwjFTQnXENzeSfcr1kuAW4jGI,3221
-tico/utils/utils.py,sha256=kg4in3E0eH2nURpnt8XRPUS2Iam3HjJRadDZdyhUy0w,13014
-tico/utils/validate_args_kwargs.py,sha256=3dXkNll9E9eZq-p0HjYaV4YltQESqdEHBU34k-tIg1k,26733
+tico/utils/utils.py,sha256=Ln6FgrnWuwvKEL1ntzZXgH1tClcaOfQHt117dQ_mSb8,12976
+tico/utils/validate_args_kwargs.py,sha256=CRj_SXMUUn6onsl8XLAt-zPZCFxR4C0XOCoaad_ZD4I,26689
 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.dev250721.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev250721.dist-info/METADATA,sha256=OwOUs1qxfuulEmUn99ocFFSGB9L0XmUmCqNhZugmsCU,8430
-tico-0.1.0.dev250721.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev250721.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev250721.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev250721.dist-info/RECORD,,
+tico-0.1.0.dev250723.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.1.0.dev250723.dist-info/METADATA,sha256=MGuH7gedgNwadikvqXLQL_zTHDmzk20cwDohwgRkreg,8430
+tico-0.1.0.dev250723.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tico-0.1.0.dev250723.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
+tico-0.1.0.dev250723.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.1.0.dev250723.dist-info/RECORD,,