ai-edge-litert-nightly 2.0.3.dev20250921__cp39-cp39-manylinux_2_27_x86_64.whl → 2.0.4.dev20250922__cp39-cp39-manylinux_2_27_x86_64.whl

ai_edge_litert/__init__.py

@@ -1 +1 @@
-__version__ = "2.0.3.dev20250921"
+__version__ = "2.0.4.dev20250922"

ai_edge_litert/aot/core/apply_plugin.py

@@ -119,7 +119,9 @@ class ApplyPlugin(components.ApplyPluginT):
         env=env,
     )
     if result.returncode:
-      log_file = tempfile.NamedTemporaryFile(mode="w", delete=False)
+      log_file = tempfile.NamedTemporaryFile(
+          suffix=".error", mode="w", delete=False
+      )
       log_file.write(result.stdout)
       log_file.close()
       raise ValueError(

ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py

@@ -31,6 +31,11 @@ COMPILER_PLUGIN_LIB_PATH = pathlib.Path(
 )
 
 
+# Returns true if the flag is a google_tensor flag.
+def _is_google_tensor_flag(flag: str) -> bool:
+  return flag.startswith("google_tensor_")
+
+
 @import_vendor.register_backend
 class GoogleTensorBackend(types.Backend):
   """Backend implementation for the Google Tensor compiler plugin."""
@@ -120,6 +125,16 @@ def _apply_plugin(
     extra_kwargs = {"libs": lib_dir, "sdk_libs_path": sdk_libs_dir}
   except FileNotFoundError:
     extra_kwargs = {}
+
+  # Add google_tensor specific flags from the backend config.
+  for flag, value in backend.config.items():
+    if _is_google_tensor_flag(flag):
+      extra_kwargs[flag] = value
+
+  for flag, value in backend.config.get("compilation_config", {}).items():
+    if _is_google_tensor_flag(flag):
+      extra_kwargs[flag] = value
+
   return component(
       input_model,
       output_model,

{ai_edge_litert_nightly-2.0.3.dev20250921.dist-info → ai_edge_litert_nightly-2.0.4.dev20250922.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-edge-litert-nightly
-Version: 2.0.3.dev20250921
+Version: 2.0.4.dev20250922
 Summary: LiteRT is for mobile and embedded devices.
 Home-page: https://www.tensorflow.org/lite/
 Author: Google AI Edge Authors

{ai_edge_litert_nightly-2.0.3.dev20250921.dist-info → ai_edge_litert_nightly-2.0.4.dev20250922.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ai_edge_litert/__init__.py,sha256=4sh1cuVzHnJR8PD53q-Qjim5rqtA96-wB-MrG5MAY7s,34
+ai_edge_litert/__init__.py,sha256=3pjXRM6TSC_2kLCpU-YzC-j2OkecxD7ofpH29qBPvuA,34
 ai_edge_litert/_pywrap_analyzer_wrapper.so,sha256=-yAgVLhtV8rlXFqOGzAUfHBOuSKZ5SpeZfaRczRIUiw,3808
 ai_edge_litert/_pywrap_litert_compiled_model_wrapper.so,sha256=tZBPPyNssub7lcsf0ikJ2c5LtAvJ5xV4jp3aVMC00os,3832
 ai_edge_litert/_pywrap_litert_tensor_buffer_wrapper.so,sha256=zhUbsJbVkwnqs5WRHT5PfBnHBEPuUnzayjeSbY1l7w0,3832
@@ -17,8 +17,8 @@ ai_edge_litert/field_mask_pb2.py,sha256=YVi1IasKjEQAlcIwM6HMgmW0wT6TMAXjiaiysrEY
 ai_edge_litert/format_converter_wrapper_pybind11.so,sha256=s1H_tXr8BCVuJ6oRnOhIWe5fL6_L3HjZw2EVbEv2M58,3872
 ai_edge_litert/hardware_accelerator.py,sha256=FgouirT1Cs80mqvMvLbO9O7wiPBnQaWDB5vjFpJhtJo,800
 ai_edge_litert/interpreter.py,sha256=4u9FfiaYkDxf_IxCjzeLWyTyxwG184uVxqg0fbj6tYg,40650
-ai_edge_litert/libLiteRtRuntimeCApi.so,sha256=lOxX1HEOp4Tnx5uqy71aIuh-D07T8tNv_Y9TlPr6lA0,6740520
-ai_edge_litert/libpywrap_litert_common.so,sha256=mIXOVycCo0uPIZKaocQxafamhNK7vUyk3sm3dVVflHI,14991432
+ai_edge_litert/libLiteRtRuntimeCApi.so,sha256=pqT71fKMTMlo72G0sNZo5WpwYpTT33Jlc3k-ecaaLuI,6802248
+ai_edge_litert/libpywrap_litert_common.so,sha256=GnUqMD_oh3Ia0hrXKMjByclJoFQkjdPEm9Yc0-L2wdc,15052280
 ai_edge_litert/metrics_interface.py,sha256=dVu6SmbnQUntPgE5o6BxHVMyemwli-7F6tDfVMGrlYI,1542
 ai_edge_litert/metrics_portable.py,sha256=KKvR9ZOe8j2ZeBtDo_6gWJ8kENKoOawPK3LPkevnZa8,2039
 ai_edge_litert/model_runtime_info_pb2.py,sha256=E93kYJtWnsChrdegZJbKzeFpplssBGEURTdOOfjtOxg,6370
@@ -38,7 +38,7 @@ ai_edge_litert/aot/prepare_for_npu.py,sha256=bghtIcLXw5P6u_lHOh-oBv4u-jCoUi-kI3w
 ai_edge_litert/aot/ai_pack/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_edge_litert/aot/ai_pack/export_lib.py,sha256=KVzfU2zZ0c6hx0dC_nLMPYUr8NROacknUH6e-MFMWfE,10237
 ai_edge_litert/aot/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_edge_litert/aot/core/apply_plugin.py,sha256=DAWj1I9k94WCkwophgz5Nrj8A0kSO2v8X3nqN_1KJFk,4554
+ai_edge_litert/aot/core/apply_plugin.py,sha256=5hzMhJCv4kaCs3pLI60EDUbxJTEEXIrjo0bZWo6PqIs,4589
 ai_edge_litert/aot/core/common.py,sha256=5imHTXV2abVwd5RJqFD0iw6WlLZ73njOPtCUccYtrWM,3103
 ai_edge_litert/aot/core/components.py,sha256=DcdEDEiA6U0HT5Tt10aw_j_M1I-XtRiXTwE5hGBMEiI,2415
 ai_edge_litert/aot/core/mlir_transforms.py,sha256=cW1DWZX1FDEI87Ihf3CfJ31lQmfZEBLLdptQCsK4sjM,1289
@@ -50,7 +50,7 @@ ai_edge_litert/aot/vendors/import_vendor.py,sha256=HNDkx9vdmDGtX1SupfChKdW4UtCA8
 ai_edge_litert/aot/vendors/example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_edge_litert/aot/vendors/example/example_backend.py,sha256=dgwTivJQzL1z3M_w4ixmOe_681E6cZW0bupgRPM_zOM,4182
 ai_edge_litert/aot/vendors/google_tensor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py,sha256=KNzphhLmkzUjzMndLVtm1V-uGA7CMYHq2kI0-CUuPUw,4595
+ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py,sha256=8JHTlEE_FAsk3wLJ40gmsN-mOFfza65nklUr118XGqE,5064
 ai_edge_litert/aot/vendors/google_tensor/target.py,sha256=EBHk2gfw2WaGERNB9GWhFngQuGH8ZitGqSz_yhC39UU,2264
 ai_edge_litert/aot/vendors/mediatek/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_edge_litert/aot/vendors/mediatek/mediatek_backend.py,sha256=uFSciDUollsKciJooxn5Fo7FWdAidSG0Q0BRBxE8TQI,6074
@@ -64,11 +64,11 @@ ai_edge_litert/internal/litertlm_core.py,sha256=twP4-6kfqmOaqyPnJ5np8CMEn8loVmk-
 ai_edge_litert/internal/litertlm_header_schema_py_generated.py,sha256=xnijnbk5eipnsDwr2pDc8qfLG_k5S3o9UlhvVawnypg,42860
 ai_edge_litert/internal/llm_metadata_pb2.py,sha256=ABbevvRn5VDBDG39zq-rsvH21qdO6OoK_ahjPqjXLZo,3028
 ai_edge_litert/internal/sampler_params_pb2.py,sha256=2sk5oGGpSre-rPEgRICwJOg7reJTuxVtwGVOjyrAriE,2069
-ai_edge_litert/tools/apply_plugin_main,sha256=LZazEPUUN18OuSmOI3sCpPxbo117a6ddACldaGe1AP0,3579784
+ai_edge_litert/tools/apply_plugin_main,sha256=ivFdQdEsLMYtEdAaZFFmUABb6MoOBstg6FDzD3lXJW4,3638248
 ai_edge_litert/vendors/google_tensor/compiler/libLiteRtCompilerPlugin_google_tensor.so,sha256=CFuOUllPZ30Ypjld0qT1S635uSfe0lJIA_UybfLOmOU,225400
-ai_edge_litert/vendors/mediatek/compiler/libLiteRtCompilerPlugin_MediaTek.so,sha256=W8h6J9y137VvlNiUg-rQoOxITE0x2UFoTfmCm6tnCao,508864
+ai_edge_litert/vendors/mediatek/compiler/libLiteRtCompilerPlugin_MediaTek.so,sha256=uji5Ld23CHgGxADIGlQFMiy2DN90_l2jG3vomF6aLZc,508864
 ai_edge_litert/vendors/qualcomm/compiler/libLiteRtCompilerPlugin_Qualcomm.so,sha256=Ti_55JUSuM8WYP73n3OsAtiqqeHL7KIFKhaZIxK1ojw,627048
-ai_edge_litert_nightly-2.0.3.dev20250921.dist-info/METADATA,sha256=XBTn9Rx37w7whTdujuAnHO5i-a_mIZrtVLBirTLZes8,1935
-ai_edge_litert_nightly-2.0.3.dev20250921.dist-info/WHEEL,sha256=J0VjzUNosQe9-CVDjrToM9MO2MyGYGeyaSVeSw3oFFM,111
-ai_edge_litert_nightly-2.0.3.dev20250921.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
-ai_edge_litert_nightly-2.0.3.dev20250921.dist-info/RECORD,,
+ai_edge_litert_nightly-2.0.4.dev20250922.dist-info/METADATA,sha256=WyJTpeG2K18Lp_-aFZKD10OoxOvzLR11ZO0P19fetaQ,1935
+ai_edge_litert_nightly-2.0.4.dev20250922.dist-info/WHEEL,sha256=J0VjzUNosQe9-CVDjrToM9MO2MyGYGeyaSVeSw3oFFM,111
+ai_edge_litert_nightly-2.0.4.dev20250922.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
+ai_edge_litert_nightly-2.0.4.dev20250922.dist-info/RECORD,,