ai-edge-torch-nightly 0.6.0.dev20250821__py3-none-any.whl → 0.6.0.dev20250822__py3-none-any.whl

ai_edge_torch/odml_torch/export.py

@@ -124,9 +124,11 @@ class LoweringInterpreter(torch.fx.Interpreter):
   def run_node(self, node: torch.fx.Node):
     loc = self._build_loc(node)
     with loc:
-      self.lctx = self.lctx.replace(ir_location=loc, node=node)
+      self.lctx.ir_location = loc
+      self.lctx.node = node
       res = super().run_node(node)
-      self.lctx = self.lctx.replace(ir_location=None, node=None)
+      self.lctx.ir_location = None
+      self.lctx.node = None
     return res
 
   def call_function(self, target, args, kwargs):

ai_edge_torch/odml_torch/export_utils.py

@@ -63,7 +63,10 @@ def inline(
   while True:
     is_changed = False
     for op in block.operations:
-      if op.OPERATION_NAME != func.CallOp.OPERATION_NAME:
+      if (
+          not hasattr(op, "OPERATION_NAME")
+          or op.OPERATION_NAME != func.CallOp.OPERATION_NAME
+      ):
         continue
 
       call_op = cast(func.CallOp, op)
@@ -96,7 +99,10 @@ def clone_func_body_ops(func_op: func.FuncOp, ir_inputs: Sequence[ir.Value]):
 
   for op in list(func_op.entry_block.operations):
     cloned_operands = [value_mapping[val] for val in op.operands]
-    if op.OPERATION_NAME == func.ReturnOp.OPERATION_NAME:
+    if (
+        hasattr(op, "OPERATION_NAME")
+        and op.OPERATION_NAME == func.ReturnOp.OPERATION_NAME
+    ):
       return cloned_operands
 
     cloned = cast(ir.Operation, op.operation.clone())

ai_edge_torch/odml_torch/lowerings/context.py

@@ -37,6 +37,3 @@ class LoweringContext:
   def loc(self):
     """Shortcut for ir_location."""
     return self.ir_location
-
-  def replace(self, **kwargs):
-    return dataclasses.replace(self, **kwargs)

ai_edge_torch/version.py

@@ -15,4 +15,4 @@
 
 # The next version of ai-edge-torch.
 # The minor version code should be bumped after every release.
-__version__ = "0.6.0.dev20250821"
+__version__ = "0.6.0.dev20250822"

{ai_edge_torch_nightly-0.6.0.dev20250821.dist-info → ai_edge_torch_nightly-0.6.0.dev20250822.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-edge-torch-nightly
-Version: 0.6.0.dev20250821
+Version: 0.6.0.dev20250822
 Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
 Home-page: https://github.com/google-ai-edge/ai-edge-torch
 Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI

{ai_edge_torch_nightly-0.6.0.dev20250821.dist-info → ai_edge_torch_nightly-0.6.0.dev20250822.dist-info}/RECORD

@@ -2,7 +2,7 @@ ai_edge_torch/__init__.py,sha256=lemyLCNoGYRnJsmDuGZu7qOqLbLqG6CGDFtu3ue1syU,129
 ai_edge_torch/_config.py,sha256=AiqhbcheF7j_ozIGDLC89k1we95aVgFDa-tR6h7UI0s,2529
 ai_edge_torch/conftest.py,sha256=r0GTrhMRhlmOGrrkvumHN8hkmyug6WvF60vWq8wRIBI,758
 ai_edge_torch/model.py,sha256=A7loFu8jE9CsXsfMmHYZ-KDFJiaD8Kkqwm_9d3IVzk0,5638
-ai_edge_torch/version.py,sha256=KCQW3wwoSg8zjssrnlyquD4aO-OeN7u1jWFHk_-trYY,806
+ai_edge_torch/version.py,sha256=3opXPBORBcfxsu8h1LE4OMIWY6oXA1l6xSP65lCMRog,806
 ai_edge_torch/_convert/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
 ai_edge_torch/_convert/conversion.py,sha256=iQk3R-pLq4c1nfLqPB4xTRj78gghxPGzJCJtILLdg5o,6123
 ai_edge_torch/_convert/conversion_utils.py,sha256=Sr8qXVcTwc-ZnZmK7yxVrIOOp1S_vNrwzC0zUvLTI2o,2160
@@ -235,8 +235,8 @@ ai_edge_torch/lowertools/translate_recipe.py,sha256=Oavs0ENKVnIeB-WidXvokTPqNlFf
 ai_edge_torch/odml_torch/__init__.py,sha256=S8jOzE9nLof-6es3XDiGJRN-9H_XTxsVm9dE7lD3RWo,812
 ai_edge_torch/odml_torch/_torch_future.py,sha256=jSYHf1CMTJzMizPMbu2b39hAt0ZTR6gQLq67GMe9KTo,2336
 ai_edge_torch/odml_torch/_torch_library.py,sha256=Lw1gqL2HWNRspdTwNhIkYAHDyafHedHtkXyKKxn-Wss,805
-ai_edge_torch/odml_torch/export.py,sha256=FDseiAOOcgN8HOuwv6HT9sALZgXGj7vZi6kwZPWPF14,14935
-ai_edge_torch/odml_torch/export_utils.py,sha256=QeA37Irlty6AiIBuqmHmJgn3lqahBQ5xsh6IKRoKm1g,4774
+ai_edge_torch/odml_torch/export.py,sha256=WPsATMFsc62aETExBsb1v6Ec1fL3r0CEvxwE7r2C_CA,14931
+ai_edge_torch/odml_torch/export_utils.py,sha256=Eax4QUefIzpmVuQxo1y9FqJ6g0qXjg4C0IVZ5uYPscs,4899
 ai_edge_torch/odml_torch/optimization_barrier.py,sha256=2lmSiu5iXWLFWpupZHvsVeNYNzG5AVGSK3K_CNhS5Sk,2290
 ai_edge_torch/odml_torch/tf_integration.py,sha256=NN29WeXmHZ0S1RPDFHUnBi2DEjMvAtwczStPYIsQ1w8,4849
 ai_edge_torch/odml_torch/composite/__init__.py,sha256=71GM_gDZxJyo38ZSoYSwhZX3xKA9rknO93JS9kw9w_c,778
@@ -257,7 +257,7 @@ ai_edge_torch/odml_torch/lowerings/_jax_lowerings.py,sha256=xUuQjoR0NJhuwG36Guyc
 ai_edge_torch/odml_torch/lowerings/_layer_norm.py,sha256=khJIvDVk2s332Nd2Be-5dM6-wp5DGff61HCV5lskHmQ,3011
 ai_edge_torch/odml_torch/lowerings/_quantized_decomposed.py,sha256=XDZ0zLej_XaQDJnaAAxhNFAd7NfQm5SOVEp_nno_krA,6178
 ai_edge_torch/odml_torch/lowerings/_rand.py,sha256=g6SuqDkuC6hD35lyP1-5H7ASDIzPSmKukeNT5naZSv8,4133
-ai_edge_torch/odml_torch/lowerings/context.py,sha256=jslcCv7r_HtImSRTxJwHAUV_QCu9Jub51lovmoBkmFA,1295
+ai_edge_torch/odml_torch/lowerings/context.py,sha256=dhGS9oFjKLKvc-aBFbNUhb4LqggP_AmPrgpT1EnOw0w,1216
 ai_edge_torch/odml_torch/lowerings/registry.py,sha256=Tp2h11l5uTifO0aIkuUOWAF_ibEjmd65Xx99w3EXuGE,1924
 ai_edge_torch/odml_torch/lowerings/utils.py,sha256=YagY2IDqtJTpPxjPz2dhb3eyCFTpTSu3ptEPSvEuDtk,10574
 ai_edge_torch/odml_torch/passes/__init__.py,sha256=AVwIwUTMx7rXacKjGy4kwrtMd3XB2v_ncdc40KOjUqQ,1245
@@ -269,8 +269,8 @@ ai_edge_torch/testing/__init__.py,sha256=_yGgvnBZWb7T3IN3mc4x1sS4vM96HZwM8pwIcPG
 ai_edge_torch/testing/export.py,sha256=k5mGDGzwc23Z4zaIVDs8CNh-oOt64gsf9MS9NjhbPy4,3293
 ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
 ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=UPB448aMDUyC0HNYVqio2rcJPnDN0tBQMP08J6vPYew,4718
-ai_edge_torch_nightly-0.6.0.dev20250821.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-ai_edge_torch_nightly-0.6.0.dev20250821.dist-info/METADATA,sha256=HNSeR4DnS8ax0bUi9CSIOzakw0lJk1ZA8JM889_0rSg,2074
-ai_edge_torch_nightly-0.6.0.dev20250821.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_torch_nightly-0.6.0.dev20250821.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
-ai_edge_torch_nightly-0.6.0.dev20250821.dist-info/RECORD,,
+ai_edge_torch_nightly-0.6.0.dev20250822.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+ai_edge_torch_nightly-0.6.0.dev20250822.dist-info/METADATA,sha256=m65Q6FH51OLricsQOL0tRYM50M0Mp3NKniIlZqfdhhc,2074
+ai_edge_torch_nightly-0.6.0.dev20250822.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ai_edge_torch_nightly-0.6.0.dev20250822.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
+ai_edge_torch_nightly-0.6.0.dev20250822.dist-info/RECORD,,