tico 0.1.0.dev250528__py3-none-any.whl → 0.1.0.dev250529__py3-none-any.whl

tico/__init__.py

@@ -18,14 +18,25 @@ import torch
 from packaging.version import Version
 
 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.dev250528"
+__version__ = "0.1.0.dev250529"
+
+MINIMUM_SUPPORTED_VERSION = "2.5.0"
+SECURE_TORCH_VERSION = "2.6.0"
 
+if Version(torch.__version__) < Version(MINIMUM_SUPPORTED_VERSION):
+    warnings.warn(
+        f"TICO officially supports torch>={MINIMUM_SUPPORTED_VERSION}. "
+        f"You are using a lower version of torch ({torch.__version__}). "
+        f"We highly recommend to upgrade torch>={MINIMUM_SUPPORTED_VERSION} to avoid unexpected behaviors."
+    )
 
-if Version(torch.__version__) < Version("2.5.0"):
+if Version(torch.__version__) < Version(SECURE_TORCH_VERSION):
     warnings.warn(
-        f"TICO officially supports torch>=2.5. You are using a lower version of torch ({torch.__version__}). We highly recommend to upgrade torch >= 2.5 to avoid unexpected behaviors."
+        f"Detected PyTorch version {torch.__version__}, which may include known security vulnerabilities. "
+        f"We recommend upgrading to {SECURE_TORCH_VERSION} or later for better security.\n"
+        "Upgrade command: pip install --upgrade torch\n"
+        "For more details, see: https://pytorch.org/security"
     )

tico/serialize/circle_serializer.py

@@ -190,6 +190,10 @@ def build_circle(edge_program: ExportedProgram) -> bytes:
     # Register outputs
     logger.debug("---------------Register outputs--------------")
     for user_output in edge_program.graph_signature.user_outputs:
+        if user_output == None:
+            logger.debug(f"Ignore 'None' output")
+            continue
+
         graph.add_output(user_output)
         logger.debug(f"Registered output: {user_output}")
 

{tico-0.1.0.dev250528.dist-info → tico-0.1.0.dev250529.dist-info}/METADATA

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

{tico-0.1.0.dev250528.dist-info → tico-0.1.0.dev250529.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tico/__init__.py,sha256=ESRk_QvvGAAqKS2h0vDY7LZoWVc0pYWSNbXIkcz2WSo,1181
+tico/__init__.py,sha256=Kb5IbIgf231eEaTOQPwejW7YylbiTAvFDd422yMVae0,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
@@ -95,7 +95,7 @@ tico/passes/segment_index_select.py,sha256=ifXOIFC12lNwsB-s3k1cJcMHP3UEijPpkMAbw
 tico/serialize/__init__.py,sha256=IO6FP_xYbGy0dW0HL26GXD3ouxARaxCK7bz9dn4blPQ,26
 tico/serialize/circle_graph.py,sha256=l9fcvV4x3L5Bh2WmEseo--0KFjSVTrlSBaBrqbrmXgg,9498
 tico/serialize/circle_mapping.py,sha256=C9C3ORACQOdvBdnt5KRzlT8zao_TvzQklIxH794OhP0,5719
-tico/serialize/circle_serializer.py,sha256=AWpLRgwjp1Xm70xMPY3cxWZTOtjV_DKzzIxJXLVKBdg,8755
+tico/serialize/circle_serializer.py,sha256=gJP2QEY7hFrlP_I4JRnnMehThr9ay9SUyoOQ2ppJBl0,8859
 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
@@ -194,9 +194,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.dev250528.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev250528.dist-info/METADATA,sha256=Ql9a9NvnYDJ7WHA_qx2TljzHnyvLfNoGEeCu6rByX6s,8633
-tico-0.1.0.dev250528.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev250528.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev250528.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev250528.dist-info/RECORD,,
+tico-0.1.0.dev250529.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.1.0.dev250529.dist-info/METADATA,sha256=YO_KwNVCiSctvJUCj2KBt-D3DfPRQBv84EstPesXEmE,8633
+tico-0.1.0.dev250529.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tico-0.1.0.dev250529.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
+tico-0.1.0.dev250529.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.1.0.dev250529.dist-info/RECORD,,