Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Bump pinned IREE version to 20241206. #778

Merged
merged 8 commits into from
Jan 7, 2025
Merged
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
5 changes: 4 additions & 1 deletion shortfin/tests/invocation/mobilenet_program_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,10 @@ 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)
# Note: this value was just copied from a sample run of the test.
# Comparison against a reference backend for this model is tested upstream
# in https://github.com/iree-org/iree-test-suites/tree/main/onnx_models.
assert absmean == pytest.approx(0.81196929)
stbaione marked this conversation as resolved.
Show resolved Hide resolved


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