Skip to content

Commit

Permalink
Bump pinned IREE version to 20241206.
Browse files Browse the repository at this point in the history
Note that the mobilenet program being tested was miscompiling and producing different output compared to onnxruntime prior to iree-org/iree@d48071d.
  • Loading branch information
ScottTodd committed Jan 7, 2025
1 parent ab29d88 commit 1034f8a
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions requirements-iree-pinned.txt
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,8 @@
# Keep these versions synced with SHORTFIN_IREE_GIT_TAG in shortfin/CMakeLists.txt
--pre
--find-links https://iree.dev/pip-release-links.html
iree-base-compiler==3.1.0rc20241204
iree-base-runtime==3.1.0rc20241204
iree-base-compiler==3.1.0rc20241206
iree-base-runtime==3.1.0rc20241206

# TODO(#760): include iree-turbine in this requirements file too?
# iree-turbine==3.1.0rc20241205
# iree-turbine==3.1.0rc20241206
2 changes: 1 addition & 1 deletion shortfin/tests/invocation/mobilenet_program_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ async def assert_mobilenet_ref_output(device, device_output):
absmean = functools.reduce(
lambda x, y: x + abs(y) / len(flat_output), flat_output, 0.0
)
assert absmean == pytest.approx(5.01964943873882)
assert absmean == pytest.approx(0.81196929)


# Tests that a single invocation on a single fiber works.
Expand Down

0 comments on commit 1034f8a

Please sign in to comment.