Skip to content

Commit

Permalink
Mark TFLite (TOSA) model tests XFAIL and log a compatibility warning.
Browse files Browse the repository at this point in the history
  • Loading branch information
ScottTodd committed Jan 16, 2025
1 parent 769df34 commit b55de5b
Show file tree
Hide file tree
Showing 3 changed files with 15 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,19 @@ def main():
if args.output_format != "mlir-bytecode":
logging.warning("output-format option is deprecated, emitting MLIR bytecode")

# Log compatibility warnings for some known issues.
try:
from packaging import version
from tensorflow import __version__ as tf_version

# https://discourse.llvm.org/t/rfc-tosa-dialect-increment-to-v1-0/83708
if version.parse(tf_version) <= version.parse("2.18.0"):
logging.warning(
f"Found tensorflow version {tf_version}. Versions of tensorflow<=2.18.0 have known compatibility issues with TOSA v1.0. Consider using a newer tensorflow version or iree-base-compiler<=3.1.0"
)
except:
pass

tflite_to_tosa(
flatbuffer=args.flatbuffer,
bytecode=args.output_path,
Expand Down
1 change: 1 addition & 0 deletions integrations/tensorflow/test/iree_tfl_tests/gpt2.run
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
# RUN: %PYTHON -m iree_tfl_tests.gpt2_test --artifacts_dir=%t
# XFAIL: *
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
# RUN: %PYTHON -m iree_tfl_tests.person_detect_test --artifacts_dir=%t
# XFAIL: *

0 comments on commit b55de5b

Please sign in to comment.