Fix tests

apache · May 4, 2022 · ce772d6 · ce772d6
1 parent 9183e93
commit ce772d6
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/python/contrib/test_ethosu/cascader/test_memory_reduction.py b/tests/python/contrib/test_ethosu/cascader/test_memory_reduction.py
@@ -88,10 +88,10 @@ def _get_ethosu_workspace_size(mod, params, accel_type, pool_size, enable_cascad
 @pytest.mark.parametrize(
     "accel_type, expected_ws_size_without_cascader, expected_ws_size_with_cascader",
     [
-        ("ethos-u55-256", 1067408, 14096),
-        ("ethos-u55-128", 1067408, 3968),
-        ("ethos-u55-64", 1067408, 2272),
-        ("ethos-u55-32", 1067392, 2256),
+        ("ethos-u55-256", 1067520, 14208),
+        ("ethos-u55-128", 1067520, 4080),
+        ("ethos-u55-64", 1067520, 2384),
+        ("ethos-u55-32", 1067504, 2368),
     ],
 )
 def test_double_conv2d(
@@ -153,8 +153,8 @@ def tf_graph(x):
 @pytest.mark.parametrize(
     "accel_type, expected_ws_size_without_cascader, expected_ws_size_with_cascader",
     [
-        ("ethos-u55-256", 180096, 5024),
-        ("ethos-u55-128", 180096, 4832),
+        ("ethos-u55-256", 180288, 5216),
+        ("ethos-u55-128", 180288, 5024),
         pytest.param("ethos-u55-64", 180096, 4832, marks=pytest.mark.xfail),
         pytest.param("ethos-u55-32", 180096, 4832, marks=pytest.mark.xfail),
     ],