Skip to content

Commit

Permalink
[microNPU] increase workspace sizes for network tests (apache#11943)
Browse files Browse the repository at this point in the history
The network tests with striping were reported to be flaky.
This commit increases the workspace size to be generous and
also repeats the test case to make sure its not flaky.

Change-Id: I134f504250c8fa0bbbcf5f673acec7ffa2ec2f55
  • Loading branch information
manupak authored Jun 29, 2022
1 parent 9ea681f commit c9d0d25
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion tests/python/contrib/test_ethosu/test_networks.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,8 +142,18 @@ def test_networks_with_usmp_and_cascader_wo_striping(accel_type, model_url, work
@pytest.mark.parametrize(
"accel_type, model_url, workspace_size",
[
# Checks the same test case multiple times to make sure its not flaky
("ethos-u55-256", MOBILENET_V1_URL, 1010000),
("ethos-u55-256", MOBILENET_V2_URL, 1180000),
("ethos-u55-256", MOBILENET_V1_URL, 1010000),
("ethos-u55-256", MOBILENET_V1_URL, 1010000),
("ethos-u55-256", MOBILENET_V1_URL, 1010000),
("ethos-u55-256", MOBILENET_V1_URL, 1010000),
# Checks the same test case multiple times to make sure its not flaky
("ethos-u55-256", MOBILENET_V2_URL, 1400000),
("ethos-u55-256", MOBILENET_V2_URL, 1400000),
("ethos-u55-256", MOBILENET_V2_URL, 1400000),
("ethos-u55-256", MOBILENET_V2_URL, 1400000),
("ethos-u55-256", MOBILENET_V2_URL, 1400000),
],
)
def test_networks_with_usmp_and_cascader_with_striping(accel_type, model_url, workspace_size):
Expand Down

0 comments on commit c9d0d25

Please sign in to comment.