This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new c9d0d253f0 [microNPU] increase workspace sizes for network tests
(#11943)
c9d0d253f0 is described below
commit c9d0d253f0f966d490f4a438b2469b593a0a0497
Author: Manupa Karunaratne <[email protected]>
AuthorDate: Wed Jun 29 12:26:44 2022 +0100
[microNPU] increase workspace sizes for network tests (#11943)
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
---
tests/python/contrib/test_ethosu/test_networks.py | 12 +++++++++++-
1 file changed, 11 insertions(+), 1 deletion(-)
diff --git a/tests/python/contrib/test_ethosu/test_networks.py
b/tests/python/contrib/test_ethosu/test_networks.py
index 9b09132a9e..075565cd92 100644
--- a/tests/python/contrib/test_ethosu/test_networks.py
+++ b/tests/python/contrib/test_ethosu/test_networks.py
@@ -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):
