aboutsummaryrefslogtreecommitdiff
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
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
-rw-r--r--src/mlia/resources/vela/vela.ini2
-rw-r--r--tests/mlia/test_devices_ethosu_reporters.py6
2 files changed, 5 insertions, 3 deletions
diff --git a/src/mlia/resources/vela/vela.ini b/src/mlia/resources/vela/vela.ini
index 382820d..29a5179 100644
--- a/src/mlia/resources/vela/vela.ini
+++ b/src/mlia/resources/vela/vela.ini
@@ -58,6 +58,7 @@ Dram_write_latency=250
const_mem_area=Axi0
arena_mem_area=Axi0
cache_mem_area=Axi0
+arena_cache_size=2096768
; Shared SRAM: the SRAM is shared between the Ethos-U and the Cortex-M software
; The non-SRAM memory is assumed to be read-only
@@ -65,6 +66,7 @@ cache_mem_area=Axi0
const_mem_area=Axi1
arena_mem_area=Axi0
cache_mem_area=Axi0
+arena_cache_size=2096768
; Dedicated SRAM: the SRAM (384KB) is only for use by the Ethos-U
; The non-SRAM memory is assumed to be read-writeable
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,