tico 0.1.0.dev250626__py3-none-any.whl → 0.1.0.dev250629__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.dev250626"
+__version__ = "0.1.0.dev250629"
 
 MINIMUM_SUPPORTED_VERSION = "2.5.0"
 SECURE_TORCH_VERSION = "2.6.0"

tico/utils/convert.py

@@ -162,6 +162,22 @@ def check_unsupported_target(exported_program: ExportedProgram):
         raise NotYetSupportedError("NOT SUPPORTED OPERATOR IN GRAPH MODULE")
 
 
+def check_training_ops(exported_program: ExportedProgram):
+    TRAINING_OPS = {
+        torch.ops.aten.dropout.default,
+        torch.ops.aten.native_dropout.default,
+    }
+    found = set()
+    for node in exported_program.graph.nodes:
+        if node.op == "call_function" and node.target in TRAINING_OPS:
+            found.add(node.target)
+
+    if found:
+        raise RuntimeError(
+            f"Detected training-mode ops {sorted(found)}. Call `model.eval()` before export."
+        )
+
+
 def convert_exported_module_to_circle(
     exported_program: ExportedProgram,
     config: CompileConfigBase = get_default_config(),
@@ -170,6 +186,7 @@ def convert_exported_module_to_circle(
     logger.debug("Input ExportedProgram (must be core aten)")
     logger.debug(exported_program)
 
+    check_training_ops(exported_program)
     # PRE-EDGE PASSES
     #
     # Here are the passes that run before to_edge() conversion.

{tico-0.1.0.dev250626.dist-info → tico-0.1.0.dev250629.dist-info}/METADATA

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

{tico-0.1.0.dev250626.dist-info → tico-0.1.0.dev250629.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tico/__init__.py,sha256=V3kK40Sfe76REBha3SgBJMRTeeZYJFErkkr3yo2nCE0,1743
+tico/__init__.py,sha256=RfkSpTE90FwwM4kJohc9prmZ7kQXSUmZaZiUKxrn-D0,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
@@ -180,7 +180,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=gh_EcWwjZZ5IyQKtazfiZ1_FvAMXe0leu5nPN-0OcHg,11919
+tico/utils/convert.py,sha256=fJRltUkBALia6N7lKcMMvmNBU-5DYyztnNZQjcgjvXU,12452
 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
@@ -198,9 +198,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.dev250626.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
-tico-0.1.0.dev250626.dist-info/METADATA,sha256=WnpZWIQvH2xOtEbyhhWKZNI56zg_fJ3ublFR7M-t128,8846
-tico-0.1.0.dev250626.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tico-0.1.0.dev250626.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
-tico-0.1.0.dev250626.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
-tico-0.1.0.dev250626.dist-info/RECORD,,
+tico-0.1.0.dev250629.dist-info/LICENSE,sha256=kp4JLII7bzRhPb0CPD5XTDZMh22BQ7h3k3B7t8TiSbw,12644
+tico-0.1.0.dev250629.dist-info/METADATA,sha256=DSkGeB1pQ2OlIkRm5lh4m4GpX5e5tuEk9rsU0joBLWE,8846
+tico-0.1.0.dev250629.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tico-0.1.0.dev250629.dist-info/entry_points.txt,sha256=kBKYSS_IYrSXmUYevmmepqIVPScq5vF8ulQRu3I_Zf0,59
+tico-0.1.0.dev250629.dist-info/top_level.txt,sha256=oqs7UPoNSKZEwqsX8B-KAWdQwfAa7i60pbxW_Jk7P3w,5
+tico-0.1.0.dev250629.dist-info/RECORD,,