aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorRuomei Yan <ruomei.yan@arm.com>2022-07-04 11:46:46 +0100
committerRuomei Yan <ruomei.yan@arm.com>2022-07-04 11:46:50 +0100
commitba2c7fcccf37e8c81946f0776714c64f73191787 (patch)
treea3c3672642828a48437d45e8c315a28fbae87973 /tests
parent0ee201dfa023a9f42e1c8a282e96c231c6769e07 (diff)
downloadmlia-ba2c7fcccf37e8c81946f0776714c64f73191787.tar.gz
MLIA-316 Specify cache arena size in the Vela memory profiles
When running the e2e test of Wav2Letter, the tensor allocation failed. To resolve this, we changed the arena cache size to 2096768. This is due to the requirement in Vela. We need to make it aware of the memory limitation on the target we are deploying for. Change-Id: I192790033954058fa5db249452d998764df52663
Diffstat (limited to 'tests')
-rw-r--r--tests/mlia/test_devices_ethosu_reporters.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/mlia/test_devices_ethosu_reporters.py b/tests/mlia/test_devices_ethosu_reporters.py
index 2d5905c..0da50e0 100644
--- a/tests/mlia/test_devices_ethosu_reporters.py
+++ b/tests/mlia/test_devices_ethosu_reporters.py
@@ -238,7 +238,7 @@ def test_report_operators(
Const mem area Axi1
Arena mem area Axi0
Cache mem area Axi0
- Arena cache size 4,294,967,296 bytes
+ Arena cache size 2,096,768 bytes
System config Ethos_U55_High_End_Embedded
Accelerator clock 500,000,000 Hz
@@ -282,7 +282,7 @@ def test_report_operators(
"const_mem_area": "Axi1",
"arena_mem_area": "Axi0",
"cache_mem_area": "Axi0",
- "arena_cache_size": {"value": 4294967296, "unit": "bytes"},
+ "arena_cache_size": {"value": 2096768, "unit": "bytes"},
},
"system_config": {
"accelerator_clock": {"value": 500000000.0, "unit": "Hz"},
@@ -355,7 +355,7 @@ def test_report_operators(
"Axi1",
"Axi0",
"Axi0",
- 4294967296,
+ 2096768,
"bytes",
"Ethos_U55_High_End_Embedded",
500000000.0,