From 1034f8a4575149feabd368b09ae9b14d95808dda Mon Sep 17 00:00:00 2001 From: Scott Todd Date: Tue, 7 Jan 2025 13:41:08 -0800 Subject: [PATCH] Bump pinned IREE version to 20241206. Note that the mobilenet program being tested was miscompiling and producing different output compared to onnxruntime prior to https://github.com/iree-org/iree/commit/d48071df356eec533554699ac1e658f6809aacbe. --- requirements-iree-pinned.txt | 6 +++--- shortfin/tests/invocation/mobilenet_program_test.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/requirements-iree-pinned.txt b/requirements-iree-pinned.txt index 4ca1246ee..11d7e4258 100644 --- a/requirements-iree-pinned.txt +++ b/requirements-iree-pinned.txt @@ -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 diff --git a/shortfin/tests/invocation/mobilenet_program_test.py b/shortfin/tests/invocation/mobilenet_program_test.py index ff7b9bbf2..4cce54a4a 100644 --- a/shortfin/tests/invocation/mobilenet_program_test.py +++ b/shortfin/tests/invocation/mobilenet_program_test.py @@ -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.