tico 0.1.0.dev250709__py3-none-any.whl → 0.1.0.dev250713__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.dev250709"
+__version__ = "0.1.0.dev250713"
 
 MINIMUM_SUPPORTED_VERSION = "2.5.0"
 SECURE_TORCH_VERSION = "2.6.0"

tico/passes/legalize_causal_mask_value.py

@@ -17,6 +17,7 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     import torch.fx
 import torch
+from torch._export.utils import is_lifted_tensor_constant
 from torch.export import ExportedProgram
 
 from tico.passes import ops
@@ -52,6 +53,7 @@ class LegalizeCausalMaskValue(PassBase):
         graph_module = exported_program.graph_module
         graph = graph_module.graph
         modified = False
+
         for node in graph.nodes:
             if not is_target_node(node, ops.aten.add):
                 continue
@@ -60,21 +62,19 @@ class LegalizeCausalMaskValue(PassBase):
             input = args.input
             other = args.other
 
-            if (
-                isinstance(input, torch.fx.Node)
-                and input.name
-                in exported_program.graph_signature.lifted_tensor_constants
+            if isinstance(input, torch.fx.Node) and is_lifted_tensor_constant(
+                exported_program, input
             ):
                 mask_node = input
-            elif (
-                isinstance(other, torch.fx.Node)
-                and other.name
-                in exported_program.graph_signature.lifted_tensor_constants
+            elif isinstance(other, torch.fx.Node) and is_lifted_tensor_constant(
+                exported_program, other
             ):
                 mask_node = other
             else:
                 continue
 
+            assert isinstance(mask_node, torch.fx.Node)
+
             mask_node_name = (
                 exported_program.graph_signature.inputs_to_lifted_tensor_constants[
                     mask_node.name
@@ -90,6 +90,7 @@ class LegalizeCausalMaskValue(PassBase):
             if torch.all(
                 torch.logical_or(mask_data == 0, mask_data < fp32_minus_inf_rounded)
             ):
+                # Replace the value from -inf to -120
                 exported_program.constants[mask_node_name] = torch.where(
                     mask_data < fp32_minus_inf_rounded,
                     torch.tensor(new_mask, dtype=mask_data.dtype),

tico/utils/convert.py

@@ -182,8 +182,13 @@ def check_training_ops(exported_program: ExportedProgram):
 
 def convert_exported_module_to_circle(
     exported_program: ExportedProgram,
-    config: CompileConfigBase = get_default_config(),
+    config: Optional[CompileConfigBase] = None,
 ) -> bytes:
+    if not config:
+        config = get_default_config()
+
+    assert isinstance(config, CompileConfigBase)
+
     logger = logging.getLogger(__name__)
     logger.debug("Input ExportedProgram (must be core aten)")
     logger.debug(exported_program)

{tico-0.1.0.dev250709.dist-info → tico-0.1.0.dev250713.dist-info}/METADATA

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

{tico-0.1.0.dev250709.dist-info → tico-0.1.0.dev250713.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tico/__init__.py,sha256=uUyBqMHPDHaFA-Jf9XrT1UKDQ1UbQTvgqju8rx5nxQg,1743
+tico/__init__.py,sha256=LDO-ry5VXGAZCkBH5Tb-m5CIIBwYSY1NqxdYJ6EtezQ,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
@@ -78,7 +78,7 @@ tico/passes/extract_dtype_kwargs.py,sha256=ObpsaFlrTPYQw2hJ7UsC5CocyAtBkT_bMtzkM
 tico/passes/fill_meta_val.py,sha256=Xbam6Aq90ZfWItZw1dgLIwH_q8RCiU5JodKNqkj-ink,1797
 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=xKdFwwMaSFCSQpSk8xISOAqFpZ1jIhgbBIqf7KTSGuk,4017
+tico/passes/legalize_causal_mask_value.py,sha256=0nfUKGd7XSe9Hg5TAi4dUi6Nn6-JRTWCwhULR5AEgqs,4079
 tico/passes/legalize_predefined_layout_operators.py,sha256=MNx7L2dAlsxSazb-F7c0onPqHleI17zAc7AzQAa9aJ4,18934
 tico/passes/lower_pow2_to_mul.py,sha256=nfJXa9ZTZMiLg6ownSyvkM4KF2z9tZW34Q3CCWI_vmQ,2402
 tico/passes/lower_to_resize_nearest_neighbor.py,sha256=N6F56Of8Aiv-KIiYLHnh33WX72W60ZVQSBEYWHdYqNQ,9005
@@ -182,7 +182,7 @@ 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=5C8Z2ia2XN4k3XgtJrFZYJSEejoeMllyr8YW6gwu9mw,12763
+tico/utils/convert.py,sha256=11Ps0i4-3Fmcts_PZO5Eo8rRL7FcLV33eHz_G97MnCg,12865
 tico/utils/define.py,sha256=Ypgp7YffM4pgPl4Zh6TmogSn1OxGBMRw_e09qYGflZk,1467
 tico/utils/diff_graph.py,sha256=_eDGGPDPYQD4b--MXX0DLoVgSt_wLfNPt47UlolLLR4,5272
 tico/utils/errors.py,sha256=f3csJjgbXG9W1aHhqEcou008Aor19W57X8oT5Hx8w1M,954
@@ -201,9 +201,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.dev250709.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev250709.dist-info/METADATA,sha256=zpUu_UaIWzup7BHIowx8BPnplKbnS6lJoHCYfvs_t_c,8430
-tico-0.1.0.dev250709.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev250709.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev250709.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev250709.dist-info/RECORD,,
+tico-0.1.0.dev250713.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.1.0.dev250713.dist-info/METADATA,sha256=FhV3CCFUbKieEpdRVt0MvAOV6sOVV-h3aKPBsrQttHY,8430
+tico-0.1.0.dev250713.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tico-0.1.0.dev250713.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
+tico-0.1.0.dev250713.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.1.0.dev250713.dist-info/RECORD,,