add warning

Signed-off-by: saienduri <saimanas.enduri@amd.com>
iree-org · Aug 6, 2024 · 083e6cd · 083e6cd
1 parent 3c8e89e
commit 083e6cd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/shark_turbine/aot/exporter.py b/shark_turbine/aot/exporter.py
@@ -252,6 +252,12 @@ def export(
             )
         )
 
+    if not strict_export:
+        warnings.warn(
+            "When strict_export == False, Python interpreter is used and code will execute exactly as it would in eager mode. "
+            "Therefore, graph optimizations and safety are not garaunteed to be at the same level as TorchDynamo."
+        )
+
     TransformedModule: Any
     current_decomps = decompositions.current_aot_decompositions()
     if isinstance(mdl, torch.export.ExportedProgram):