ai-edge-torch-nightly 0.2.0.dev20240603__py3-none-any.whl → 0.2.0.dev20240604__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of ai-edge-torch-nightly might be problematic. Click here for more details.

Files changed (6) hide show
  1. ai_edge_torch/convert/test/test_convert.py +3 -4
  2. {ai_edge_torch_nightly-0.2.0.dev20240603.dist-info → ai_edge_torch_nightly-0.2.0.dev20240604.dist-info}/METADATA +1 -1
  3. {ai_edge_torch_nightly-0.2.0.dev20240603.dist-info → ai_edge_torch_nightly-0.2.0.dev20240604.dist-info}/RECORD +6 -6
  4. {ai_edge_torch_nightly-0.2.0.dev20240603.dist-info → ai_edge_torch_nightly-0.2.0.dev20240604.dist-info}/LICENSE +0 -0
  5. {ai_edge_torch_nightly-0.2.0.dev20240603.dist-info → ai_edge_torch_nightly-0.2.0.dev20240604.dist-info}/WHEEL +0 -0
  6. {ai_edge_torch_nightly-0.2.0.dev20240603.dist-info → ai_edge_torch_nightly-0.2.0.dev20240604.dist-info}/top_level.txt +0 -0
ai_edge_torch/convert/test/test_convert.py CHANGED
@@ -212,7 +212,6 @@ class TestConvert(unittest.TestCase):
212
212
  self.assertTrue(flags["key1"], "new_value1")
213
213
  self.assertTrue(flags["key2"]["subkey2"], "new_subvalue2")
214
214
 
215
- @unittest.skip("https://b.corp.google.com/issues/331463544")
216
215
  def test_convert_add_backdoor_flags(self):
217
216
  """Tests conversion of an add module setting a tflite converter flag."""
218
217
 
@@ -228,13 +227,13 @@ class TestConvert(unittest.TestCase):
228
227
  torch_module = Add().eval()
229
228
 
230
229
  with tempfile.TemporaryDirectory() as tmp_dir_path:
231
- mlir_dump_path = os.path.join(
230
+ ir_dump_path = os.path.join(
232
231
  tmp_dir_path, "test_convert_add_backdoor_flags_mlir_dump"
233
232
  )
234
233
  ai_edge_torch.convert(
235
- torch_module, args, _ai_edge_converter_flags={"mlir_dump_dir": mlir_dump_path}
234
+ torch_module, args, _ai_edge_converter_flags={"ir_dump_dir": ir_dump_path}
236
235
  )
237
- self.assertTrue(os.path.isdir(mlir_dump_path))
236
+ self.assertTrue(os.path.isdir(ir_dump_path))
238
237
 
239
238
  def test_convert_model_with_dynamic_batch(self):
240
239
  """
{ai_edge_torch_nightly-0.2.0.dev20240603.dist-info → ai_edge_torch_nightly-0.2.0.dev20240604.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ai-edge-torch-nightly
3
- Version: 0.2.0.dev20240603
3
+ Version: 0.2.0.dev20240604
4
4
  Summary: Supporting PyTorch models with the Google AI Edge TFLite runtime.
5
5
  Home-page: https://github.com/google-ai-edge/ai-edge-torch
6
6
  Keywords: On-Device ML,AI,Google,TFLite,PyTorch,LLMs,GenAI
{ai_edge_torch_nightly-0.2.0.dev20240603.dist-info → ai_edge_torch_nightly-0.2.0.dev20240604.dist-info}/RECORD RENAMED
@@ -21,7 +21,7 @@ ai_edge_torch/convert/fx_passes/optimize_layout_transposes_pass/layout_partition
21
21
  ai_edge_torch/convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/greedy.py,sha256=8uHJbIwPMTgeSfYVba163pkXSQkHLxFwar_8A1AhgAM,2279
22
22
  ai_edge_torch/convert/fx_passes/optimize_layout_transposes_pass/layout_partitioners/min_cut.py,sha256=FlNKt2EhIKnlVEeUWTiv5sz446YKU6Yy1H0Gd6VRgkU,6432
23
23
  ai_edge_torch/convert/test/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
24
- ai_edge_torch/convert/test/test_convert.py,sha256=USduDO6PaO3nlA82jMihTct--mCU_ugILZDin00lcJ8,8092
24
+ ai_edge_torch/convert/test/test_convert.py,sha256=2qPmmGqnfV_o1gfsSdjGq3-JR1b323ligiy5MdAv9NA,8021
25
25
  ai_edge_torch/convert/test/test_convert_composites.py,sha256=SrVn_cEMtQhYYCMOUKK0K7M57MQNQX-lOUwieln0HGA,6616
26
26
  ai_edge_torch/convert/test/test_convert_multisig.py,sha256=kMaGnHe9ylfyU68qCifYcaGwJqyejKz--QQt9jS2oUA,4537
27
27
  ai_edge_torch/debug/__init__.py,sha256=TKvmnjVk3asvYcVh6C-LPr6srgAF_nppSAupWEXqwPY,707
@@ -100,8 +100,8 @@ ai_edge_torch/quantize/quant_config.py,sha256=ExThdTXqnWmGC3-F6sdXbXr8nYzkEe_qCz
100
100
  ai_edge_torch/testing/__init__.py,sha256=hHLluseD2R0Hh4W6XZRIXY_dRQeYudjsrKGf6LZz65g,671
101
101
  ai_edge_torch/testing/model_coverage/__init__.py,sha256=5P8J6Zk5YYtDvTBucFvB9NGSRI7Gw_24WnrbhXgycEE,765
102
102
  ai_edge_torch/testing/model_coverage/model_coverage.py,sha256=EIyKz-HY70DguWuSrJal8LpYXQ5ZSEUf3ZrVl7jikFM,4286
103
- ai_edge_torch_nightly-0.2.0.dev20240603.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
104
- ai_edge_torch_nightly-0.2.0.dev20240603.dist-info/METADATA,sha256=lSrdb2AHtqNqaAx0xXSxvAU2VZBpRslbse2gWrwNEo0,1748
105
- ai_edge_torch_nightly-0.2.0.dev20240603.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
106
- ai_edge_torch_nightly-0.2.0.dev20240603.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
107
- ai_edge_torch_nightly-0.2.0.dev20240603.dist-info/RECORD,,
103
+ ai_edge_torch_nightly-0.2.0.dev20240604.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
104
+ ai_edge_torch_nightly-0.2.0.dev20240604.dist-info/METADATA,sha256=0Hjm-wBxxZRbVdxcFhmOKVwsU52_l5F6IblrflOWMWk,1748
105
+ ai_edge_torch_nightly-0.2.0.dev20240604.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
106
+ ai_edge_torch_nightly-0.2.0.dev20240604.dist-info/top_level.txt,sha256=5KXRaF2hwkApYxf7Y8y_tVb9aulGTlbOoNdbx1aKRkE,14
107
+ ai_edge_torch_nightly-0.2.0.dev20240604.dist-info/RECORD,,