Cherry-pick fix for mobilenet numerics that caused unit tests to hang.

nod-ai · Jan 7, 2025 · 2c158c4 · 2c158c4
1 parent 04ef073
commit 2c158c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.