Skip to content

Commit

Permalink
Fix tflite flags test (#23)
Browse files Browse the repository at this point in the history
Update mlir_dump_dir to ir_dump_dir since it has recently changed in
tflite.
  • Loading branch information
majiddadashi authored Jun 3, 2024
1 parent 124a20d commit ee85139
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions ai_edge_torch/convert/test/test_convert.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,7 +212,6 @@ def __init__(self):
self.assertTrue(flags["key1"], "new_value1")
self.assertTrue(flags["key2"]["subkey2"], "new_subvalue2")

@unittest.skip("https://b.corp.google.com/issues/331463544")
def test_convert_add_backdoor_flags(self):
"""Tests conversion of an add module setting a tflite converter flag."""

Expand All @@ -228,13 +227,13 @@ def forward(self, a, b):
torch_module = Add().eval()

with tempfile.TemporaryDirectory() as tmp_dir_path:
mlir_dump_path = os.path.join(
ir_dump_path = os.path.join(
tmp_dir_path, "test_convert_add_backdoor_flags_mlir_dump"
)
ai_edge_torch.convert(
torch_module, args, _ai_edge_converter_flags={"mlir_dump_dir": mlir_dump_path}
torch_module, args, _ai_edge_converter_flags={"ir_dump_dir": ir_dump_path}
)
self.assertTrue(os.path.isdir(mlir_dump_path))
self.assertTrue(os.path.isdir(ir_dump_path))

def test_convert_model_with_dynamic_batch(self):
"""
Expand Down

0 comments on commit ee85139

Please sign in to comment.