ai-edge-litert-nightly 2.0.4.dev20250924__cp310-cp310-manylinux_2_27_x86_64.whl → 2.0.4.dev20250926__cp310-cp310-manylinux_2_27_x86_64.whl

ai_edge_litert/__init__.py

@@ -1 +1 @@
-__version__ = "2.0.4.dev20250924"
+__version__ = "2.0.4.dev20250926"

ai_edge_litert/aot/ai_pack/export_lib.py

@@ -21,6 +21,7 @@ from typing import cast
 from ai_edge_litert.aot.core import common
 from ai_edge_litert.aot.core import types
 from ai_edge_litert.aot.vendors import fallback_backend
+from ai_edge_litert.aot.vendors.google_tensor import target as google_tensor_target
 from ai_edge_litert.aot.vendors.mediatek import mediatek_backend
 from ai_edge_litert.aot.vendors.mediatek import target as mtk_target
 from ai_edge_litert.aot.vendors.qualcomm import qualcomm_backend
@@ -194,6 +195,16 @@ def _target_to_ai_pack_info(target: types.Target) -> str | None:
         device_group_name=group_name, device_selectors=device_selector
     )
     return device_group
+  elif isinstance(target, google_tensor_target.Target):
+    group_name = str(target)
+    soc_manufacturer, soc_model = _process_google_tensor_target(target)
+    device_selector = _DEVICE_SELECTOR_TEMPLATE.format(
+        soc_man=soc_manufacturer, soc_model=soc_model
+    )
+    device_group = _DEVICE_GROUP_TEMPLATE.format(
+        device_group_name=group_name, device_selectors=device_selector
+    )
+    return device_group
   elif isinstance(target, fallback_backend.FallbackTarget):
     # Don't need to have device selector for fallback target.
     return None
@@ -222,6 +233,14 @@ def _process_mtk_target(
   ), str(target.soc_model)
 
 
+# TODO: b/407453529 - Auto-generate this function from CSVs.
+def _process_google_tensor_target(
+    target: google_tensor_target.Target,
+) -> tuple[str, str]:
+  """Returns tuple of (manufacturer, model) for the given Google Tensor target."""
+  return str(target.soc_manufacturer), str(target.soc_model).replace('_', ' ')
+
+
 def _write_targeting_config(
     compiled_models: types.CompilationResult, ai_pack_dir: pathlib.Path
 ) -> None:

ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py

@@ -46,7 +46,7 @@ class GoogleTensorBackend(types.Backend):
 
   @property
   def soc_manufacturer(self) -> target_lib.SocManufacturer:
-    return target_lib.SocManufacturer.GOOGLE_TENSOR
+    return target_lib.SocManufacturer.GOOGLE
 
   @property
   def soc_model(self) -> target_lib.SocModel:

ai_edge_litert/aot/vendors/google_tensor/target.py

@@ -32,7 +32,7 @@ else:
 # pylint: enable=g-importing-member
 
 
-_GOOGLE_TENSOR_BACKEND_ID = "google_tensor"
+_GOOGLE_TENSOR_BACKEND_ID = "GOOGLE"
 
 
 class SocModel(StrEnum):
@@ -40,15 +40,15 @@ class SocModel(StrEnum):
 
   ALL = "ALL"
 
-  G3 = "g3"
-  G4 = "g4"
-  G5 = "g5"
+  TENSOR_G3 = "Tensor_G3"
+  TENSOR_G4 = "Tensor_G4"
+  TENSOR_G5 = "Tensor_G5"
 
 
 class SocManufacturer(StrEnum):
   """Google Tensor SOC manufacturer."""
 
-  GOOGLE_TENSOR = "GoogleTensor"
+  GOOGLE = "Google"
 
 
 @dataclasses.dataclass
@@ -56,7 +56,7 @@ class Target(types.Target):
   """Compilation target for Google Tensor SOCs."""
 
   soc_model: SocModel
-  soc_manufacturer: SocManufacturer = SocManufacturer.GOOGLE_TENSOR
+  soc_manufacturer: SocManufacturer = SocManufacturer.GOOGLE
 
   @classmethod
   def backend_id(cls) -> str:

{ai_edge_litert_nightly-2.0.4.dev20250924.dist-info → ai_edge_litert_nightly-2.0.4.dev20250926.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-edge-litert-nightly
-Version: 2.0.4.dev20250924
+Version: 2.0.4.dev20250926
 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.4.dev20250924.dist-info → ai_edge_litert_nightly-2.0.4.dev20250926.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ai_edge_litert/__init__.py,sha256=_fJd6-vaLks8aTS8r5u8wYK1KUWQ1FyYVwjia8RXg5w,34
+ai_edge_litert/__init__.py,sha256=UYTfdtds6qtVLD5XCWsyEjh4tvUWoAVFh79qZHWoQyg,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=myhbESLC7R5tMfk_gIOH9k8ttwEpUHsdHz_Ia_PceGU,6810912
-ai_edge_litert/libpywrap_litert_common.so,sha256=LV-tWLiWflOwALnj8lzkMgaZcR9n74mbAtfh5Qu8OPA,15051648
+ai_edge_litert/libLiteRtRuntimeCApi.so,sha256=vk-NBzl1Ss1QFhth-RSnA8LrmInQcy7D6NPxXl4L0kE,6815664
+ai_edge_litert/libpywrap_litert_common.so,sha256=02nL2ZbQQLqbhRDIs4Dx-DGgtSSivXCzWMgenrLBBB4,15057480
 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
@@ -36,7 +36,7 @@ ai_edge_litert/aot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 ai_edge_litert/aot/aot_compile.py,sha256=CtBdFTu-a3pmPCruCZwA1FNJFLZXaSFZ_yvt8O5oBxM,5554
 ai_edge_litert/aot/prepare_for_npu.py,sha256=bghtIcLXw5P6u_lHOh-oBv4u-jCoUi-kI3wF3wA1Mow,5494
 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/ai_pack/export_lib.py,sha256=-1KhA1xK6jm7bDPg9zZ7DoIbMWRfmrc1Y4PPCOqdVJU,11068
 ai_edge_litert/aot/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_edge_litert/aot/core/apply_plugin.py,sha256=5hzMhJCv4kaCs3pLI60EDUbxJTEEXIrjo0bZWo6PqIs,4589
 ai_edge_litert/aot/core/common.py,sha256=5imHTXV2abVwd5RJqFD0iw6WlLZ73njOPtCUccYtrWM,3103
@@ -50,8 +50,8 @@ 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=8JHTlEE_FAsk3wLJ40gmsN-mOFfza65nklUr118XGqE,5064
-ai_edge_litert/aot/vendors/google_tensor/target.py,sha256=EBHk2gfw2WaGERNB9GWhFngQuGH8ZitGqSz_yhC39UU,2264
+ai_edge_litert/aot/vendors/google_tensor/google_tensor_backend.py,sha256=xWDwsw1gjb7uPoXyoaiI5cZ6uKCRK87VrVLyjvPyrcc,5057
+ai_edge_litert/aot/vendors/google_tensor/target.py,sha256=jzJVgKr2grhaazs-Wo-yHLb9QjGWDENA44uAjO7s4p8,2279
 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
 ai_edge_litert/aot/vendors/mediatek/target.py,sha256=MvwdgybSHmVYuN1txgzmDUvO5HYubx9TLzOi1TXkJx4,2214
@@ -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=cODsIYPCep9D7CYMBrmFrpPhGks8_KTr34Fu2qi569o,3636400
-ai_edge_litert/vendors/google_tensor/compiler/libLiteRtCompilerPlugin_google_tensor.so,sha256=TIqd_tQoOzLz9Cpbqc4MTA60oEUSWkYDUKUDvthLr_A,225592
-ai_edge_litert/vendors/mediatek/compiler/libLiteRtCompilerPlugin_MediaTek.so,sha256=MOVnT8r8BRV9QI5E7hQoC6S4UkzNAqwZA0X-WOwXIps,508928
-ai_edge_litert/vendors/qualcomm/compiler/libLiteRtCompilerPlugin_Qualcomm.so,sha256=qeLL2764L_hdVFkkW5QpMhsMQaTmpxIZ6OgZ0zjA2U4,627928
-ai_edge_litert_nightly-2.0.4.dev20250924.dist-info/METADATA,sha256=OL-amnKjecaz6qSZOGSJJUkB6Izur6X9_ebOPj9Ckhg,1935
-ai_edge_litert_nightly-2.0.4.dev20250924.dist-info/WHEEL,sha256=za-oMjTdcjhchGU1RZp7BTlGkA-HSjVEh4xakst1U4I,113
-ai_edge_litert_nightly-2.0.4.dev20250924.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
-ai_edge_litert_nightly-2.0.4.dev20250924.dist-info/RECORD,,
+ai_edge_litert/tools/apply_plugin_main,sha256=cLHcBIPX2nHkPvHKkyqLtRapywMxBNlciYrb2jzrUrs,3636232
+ai_edge_litert/vendors/google_tensor/compiler/libLiteRtCompilerPlugin_google_tensor.so,sha256=Lbb6h_2a56s50ik-S8z64swbu-DweNdkSQJ2obqP_J4,227472
+ai_edge_litert/vendors/mediatek/compiler/libLiteRtCompilerPlugin_MediaTek.so,sha256=DMLZ1TaYEkdwV-jf7QxqbvDT_iIRlnyOjRtjB_8rdHw,508856
+ai_edge_litert/vendors/qualcomm/compiler/libLiteRtCompilerPlugin_Qualcomm.so,sha256=I1OWzUDwph_9a8LpSN3_znWd0qh4KJd_d-2_aJbYuLo,629776
+ai_edge_litert_nightly-2.0.4.dev20250926.dist-info/METADATA,sha256=mCmGLH-x0RopOlCfK7f0i5r4FJMP9ulJnTlyfIl9p8M,1935
+ai_edge_litert_nightly-2.0.4.dev20250926.dist-info/WHEEL,sha256=za-oMjTdcjhchGU1RZp7BTlGkA-HSjVEh4xakst1U4I,113
+ai_edge_litert_nightly-2.0.4.dev20250926.dist-info/top_level.txt,sha256=WcDZgG99n0a0xDS_ipL8ZWy956g1v5fVyR3FH96VDT0,15
+ai_edge_litert_nightly-2.0.4.dev20250926.dist-info/RECORD,,