tico 0.1.0.dev250611__py3-none-any.whl → 0.1.0.dev250616__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.dev250611"
+__version__ = "0.1.0.dev250616"
 
 MINIMUM_SUPPORTED_VERSION = "2.5.0"
 SECURE_TORCH_VERSION = "2.6.0"

tico/passes/fuse_leading_unsqueeze_reshape.py

@@ -58,7 +58,7 @@ class FuseLeadingUnsqueezeReshape(PassBase):
             if not is_target_node(reshape_back, ops.aten.reshape):
                 continue
             reshape_back_args = ReshapeArgs(*reshape_back.args, **reshape_back.kwargs)  # type: ignore[arg-type]
-            permute, reshape_back_size = reshape_back_args.input, reshape_back_args.size
+            permute = reshape_back_args.input
 
             if not is_target_node(permute, ops.aten.permute):
                 continue
@@ -70,7 +70,7 @@ class FuseLeadingUnsqueezeReshape(PassBase):
             reshape_front_args = ReshapeArgs(*reshape_front.args, **reshape_front.kwargs)  # type: ignore[arg-type]
             reshape_front_input, reshape_front_size = (
                 reshape_front_args.input,
-                reshape_front_args.size,
+                reshape_front_args.shape,
             )
 
             # ---- condition: only leading unsqueeze ------------------------

tico/passes/remove_redundant_reshape.py

@@ -75,7 +75,7 @@ class RemoveRedundantReshapePattern1(PassBase):
             if len(reshape1.users) != 1:
                 continue
             reshape1_args = ReshapeArgs(*reshape1.args, **reshape1.kwargs)  # type: ignore[arg-type]
-            reshape1_input, reshape1_size = reshape1_args.input, reshape1_args.size
+            reshape1_input = reshape1_args.input
             # `(AxBxC) - aten.reshape` - (1xAxBxC)
             if [1] + list(extract_shape(reshape1_input)) != list(
                 extract_shape(reshape1)
@@ -108,7 +108,7 @@ class RemoveRedundantReshapePattern1(PassBase):
             if len(reshape2.users) != 1:
                 continue
             reshape2_args = ReshapeArgs(*reshape2.args, **reshape2.kwargs)  # type: ignore[arg-type]
-            reshape2_input, reshape2_size = reshape2_args.input, reshape2_args.size
+            reshape2_input = reshape2_args.input
             # (1xAxCxB) - `aten.reshape - (AxCxB)
             if list(extract_shape(reshape2_input)) != [1] + list(
                 extract_shape(reshape2)
@@ -157,7 +157,7 @@ class RemoveRedundantReshapePattern2(PassBase):
             if len(reshape1.users) != 1:
                 continue
             reshape1_args = ReshapeArgs(*reshape1.args, **reshape1.kwargs)  # type: ignore[arg-type]
-            reshape1_input, reshape1_size = reshape1_args.input, reshape1_args.size
+            reshape1_input = reshape1_args.input
             # `(AxBxC) - aten.reshape` - (1xAxBxC)
             if [1] + list(extract_shape(reshape1_input)) != list(
                 extract_shape(reshape1)
@@ -183,7 +183,7 @@ class RemoveRedundantReshapePattern2(PassBase):
             if len(reshape2.users) != 1:
                 continue
             reshape2_args = ReshapeArgs(*reshape2.args, **reshape2.kwargs)  # type: ignore[arg-type]
-            reshape2_input, reshape2_size = reshape2_args.input, reshape2_args.size
+            reshape2_input, reshape2_size = reshape2_args.input, reshape2_args.shape
             # (Bx1xAxC) - `aten.reshape - (Bx(A*C))
             reshape2_input_shape = list(extract_shape(reshape2_input))
             assert len(reshape2_input_shape) == 4
@@ -270,7 +270,7 @@ class RemoveRedundantReshapePattern3(PassBase):
             if not reshape_2.target in ops.aten.reshape:
                 continue
             reshape_2_args = ReshapeArgs(*reshape_2.args, **reshape_2.kwargs)  # type: ignore[arg-type]
-            reshape_2_input, reshape_2_size = reshape_2_args.input, reshape_2_args.size
+            reshape_2_input = reshape_2_args.input
             assert isinstance(reshape_2_input, torch.fx.Node), type(reshape_2_input)
             # reshape_3
             if not reshape_3.op == "call_function":
@@ -278,7 +278,7 @@ class RemoveRedundantReshapePattern3(PassBase):
             if not reshape_3.target in ops.aten.reshape:
                 continue
             reshape_3_args = ReshapeArgs(*reshape_3.args, **reshape_3.kwargs)  # type: ignore[arg-type]
-            reshape_3_input, reshape_3_size = reshape_3_args.input, reshape_3_args.size
+            reshape_3_input = reshape_3_args.input
             assert isinstance(reshape_3_input, torch.fx.Node), type(reshape_3_input)
 
             # Check condition
@@ -347,7 +347,7 @@ class RemoveRedundantReshapePattern4(PassBase):
                 continue
 
             reshape1_args = ReshapeArgs(*reshape1.args, **reshape1.kwargs)  # type: ignore[arg-type]
-            reshape1_input, size = reshape1_args.input, reshape1_args.size
+            reshape1_input, size = reshape1_args.input, reshape1_args.shape
             assert isinstance(reshape1_input, torch.fx.Node), type(reshape1_input)
             assert isinstance(size, list), type(size)
             for s in size:
@@ -362,7 +362,7 @@ class RemoveRedundantReshapePattern4(PassBase):
                 continue
 
             reshape2_args = ReshapeArgs(*reshape2.args, **reshape2.kwargs)  # type: ignore[arg-type]
-            reshape2_input, reshape2_size = reshape2_args.input, reshape2_args.size
+            reshape2_input, reshape2_size = reshape2_args.input, reshape2_args.shape
             assert isinstance(reshape2_input, torch.fx.Node), type(reshape2_input)
             assert isinstance(reshape2_size, list), type(reshape2_size)
             for s in reshape2_size:
@@ -410,7 +410,7 @@ class RemoveRedundantReshapePattern5(PassBase):
                 continue
 
             args = ReshapeArgs(*node.args, **node.kwargs)  # type: ignore[arg-type]
-            output_shape = args.size
+            output_shape = args.shape
             input_shape = list(extract_shape(args.input))
 
             if output_shape != input_shape:

tico/serialize/operators/op_reshape.py

@@ -48,7 +48,7 @@ class ReshapeVisitor(NodeVisitor):
         )
         args = ReshapeArgs(*node.args, **node.kwargs)  # type: ignore[arg-type]
         input = args.input
-        size = args.size
+        size = args.shape
 
         if isinstance(size, int):
             raise NotYetSupportedError("scalar size conversion is not supported yet.")

tico/utils/validate_args_kwargs.py

@@ -867,7 +867,7 @@ class ReshapeArgs:
     """
 
     input: torch.fx.Node
-    size: List[int]
+    shape: List[int]
 
 
 @enforce_type

{tico-0.1.0.dev250611.dist-info → tico-0.1.0.dev250616.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tico
-Version: 0.1.0.dev250611
+Version: 0.1.0.dev250616
 Summary: Convert exported Torch module to circle
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -105,10 +105,15 @@ You can convert a torch module to a circle model with these steps.
 torch_module = AddModule()
 example_inputs = (torch.ones(4), torch.ones(4))
 
-circle_model = tico.convert(torch_module, example_inputs)
+circle_model = tico.convert(torch_module.eval(), example_inputs)
 circle_model.save('add.circle')
 ```
 
+**NOTE**
+Please make sure to call `eval()` on the PyTorch module before passing it to our API.
+This ensures the model runs in inference mode, disabling layers like dropout and
+batch normalization updates.
+
 **Compile with configuration**
 
 ```python

{tico-0.1.0.dev250611.dist-info → tico-0.1.0.dev250616.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tico/__init__.py,sha256=Hdcii4nD2OeaUn6hqhQF4rmXIBkKre9mFUyuvzdYsW4,1743
+tico/__init__.py,sha256=C7LzUiiXkYNLAtSEQwbhwUac4-sXXzDB7nzZccomIWo,1743
 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
@@ -75,7 +75,7 @@ tico/passes/decompose_grouped_conv2d.py,sha256=hPRK9D9RDjmJ3mv0Wx17zeNW4xXT2jTQD
 tico/passes/decompose_slice_scatter.py,sha256=mnfEt58PAelJdheD0cJ5C66w_3yqSngCsfM-Auy9ADU,5563
 tico/passes/extract_dtype_kwargs.py,sha256=ObpsaFlrTPYQw2hJ7UsC5CocyAtBkT_bMtzkMUqAyKc,4333
 tico/passes/fill_meta_val.py,sha256=Xbam6Aq90ZfWItZw1dgLIwH_q8RCiU5JodKNqkj-ink,1797
-tico/passes/fuse_leading_unsqueeze_reshape.py,sha256=zi_I0xISUooXqa29m5nefnYMCDMDEIAYTwHECwFA82o,4192
+tico/passes/fuse_leading_unsqueeze_reshape.py,sha256=Y5732qd0MilMQsnPiw--lxTy2lywomKVNoFt2212lLQ,4150
 tico/passes/fuse_redundant_reshape_to_mean.py,sha256=2SEZ0TfXsmUkuqUdNtS_mQzLd49su5TsQViCLMGL4f4,3691
 tico/passes/legalize_causal_mask_value.py,sha256=xKdFwwMaSFCSQpSk8xISOAqFpZ1jIhgbBIqf7KTSGuk,4017
 tico/passes/legalize_predefined_layout_operators.py,sha256=4AAqYvwMvtuAQ8mdpTJI6oIgcBL_iSMzB0pmcvJ6i3Y,15887
@@ -88,7 +88,7 @@ tico/passes/remove_nop.py,sha256=Hf91p_EJAOC6DyWNthash0_UWtEcNc_M7znamQfYQ5Y,268
 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_permute.py,sha256=ovrEpB9x_Pjdfq6N35cDyURjP8ZBlAqgFiitPxTvUlI,4204
-tico/passes/remove_redundant_reshape.py,sha256=AfY_TpcXMdelzgE9RSPwU41SKOYofr3bvsp1q-TBnLw,16402
+tico/passes/remove_redundant_reshape.py,sha256=MMd9NWf9iTNIWqizFjC7ykA-JipmMMLCfkrGinSsfno,16227
 tico/passes/remove_redundant_slice.py,sha256=Iv7TbB39fktNb4eq0VdyZnwxL_VsKLJ90diMmaf3kZk,2087
 tico/passes/remove_redundant_to_copy.py,sha256=JMDnca2f9OE58ejdMEoZ1O0l-rGAc5T91BT-dFk7FGk,3114
 tico/passes/restore_linear.py,sha256=UMMHdLmRGq9bfJx_0L9lL2UQBd51PGNP0WywO8KdrDM,4066
@@ -157,7 +157,7 @@ tico/serialize/operators/op_reciprocal.py,sha256=6b9_bxjg_0EvgAitSv1MgBi4PJSEgm-
 tico/serialize/operators/op_relu.py,sha256=WXCR_chwEUBqjFIQ_4E2avwk-Acy76pmX20rJQCBTQo,1832
 tico/serialize/operators/op_relu6.py,sha256=ZWqEolfAKjOdUC1ZCg0iuu4dBhkJRxVYR2tUzpbvKQM,1829
 tico/serialize/operators/op_repeat.py,sha256=0wTv1Mg7kg0eHz0CT6atyVAli4T4h5rYXq5opY6op20,4235
-tico/serialize/operators/op_reshape.py,sha256=PdYenXvfQxzYST3yNH6MTxUQ25TulNwiip6N2q76zfQ,2549
+tico/serialize/operators/op_reshape.py,sha256=0_bJwimiGAHaKkfwfhxUw9Gebt5tnecGaEVoKhEvV0Q,2550
 tico/serialize/operators/op_resize_nearest_neighbor.py,sha256=dXaAnZ5M_ko_tH-HolxNpHFXkDUQ8x45myskojP5XZE,2771
 tico/serialize/operators/op_rsqrt.py,sha256=yl2vd8InjhLPbE0vHIrEera6DVXlY9dLgO7yZZCH3RI,1837
 tico/serialize/operators/op_scalar_tensor.py,sha256=vDWxi4hXwyDJJhvfMR_QrBInw_No3WeU_M4gtfZqmbo,1928
@@ -191,14 +191,14 @@ tico/utils/register_custom_op.py,sha256=iRQvdqlBqrJxq_pNkvJyDIJD_SYtCUl88wwbbuvS
 tico/utils/serialize.py,sha256=AQXMBOLu-Kg2Rn-qbqsAtHndjZAZIavlKA0QFgJREHM,1420
 tico/utils/trace_decorators.py,sha256=ddLIiKQfSaQrxgF1kNpwjFTQnXENzeSfcr1kuAW4jGI,3221
 tico/utils/utils.py,sha256=fnbZ2RLH6-J-wqb32O4qsR1ce4BJU0wYNrk84QXa6_E,13158
-tico/utils/validate_args_kwargs.py,sha256=P4aMnr9EhNCtc_AgJPpuezfQbqFfDn0lhJSWqmumLZ8,25054
+tico/utils/validate_args_kwargs.py,sha256=ZX820o05hmzg3QmmduAy-rIOrDyPHYe6Uj6Yf6Hoe_c,25055
 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.dev250611.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev250611.dist-info/METADATA,sha256=42yMwo0GJBgXFJZf1DgkbdvPuXnrMTjuMVvUMhekukE,8633
-tico-0.1.0.dev250611.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev250611.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev250611.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev250611.dist-info/RECORD,,
+tico-0.1.0.dev250616.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.1.0.dev250616.dist-info/METADATA,sha256=gEVJ9bVRXMqIw5UOqPGS-j3CV7dkmpuFC35cZxg3sBk,8846
+tico-0.1.0.dev250616.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tico-0.1.0.dev250616.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
+tico-0.1.0.dev250616.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.1.0.dev250616.dist-info/RECORD,,