Add memory area to run_platform.py
Control the placement of the model and the arena for baremetal application
in SRAM/DRAM by 4 configurable options:
a. Model in SRAM and Arena in SRAM
b. Model in SRAM and Arena in DRAM
c. Model in DRAM and Arena in DRAM w/o Scratch buffer
d. Model in DRAM and Arena in DRAM with Scratch buffer in SRAM
Change-Id: Ia154be8a1c88cb13aeee62e701c2db7719a9d71c
diff --git a/targets/corstone-300/platform.scatter b/targets/corstone-300/platform.scatter
index 8a6bc88..fab12d1 100644
--- a/targets/corstone-300/platform.scatter
+++ b/targets/corstone-300/platform.scatter
@@ -65,6 +65,16 @@
* memory banks.
*/
+#ifndef ETHOSU_MODEL
+ /* default value - '1', for DRAM */
+ #define ETHOSU_MODEL 1
+#endif
+
+#ifndef ETHOSU_ARENA
+ /* default value - '1', for DRAM */
+ #define ETHOSU_ARENA 1
+#endif
+
#ifndef STACK_SIZE
#define STACK_SIZE 0x8000
#endif
@@ -157,14 +167,20 @@
}
; 2MB SSE-300 SRAM (3 cycles read latency) from M55/U55
- SRAM SRAM_START UNINIT SRAM_SIZE
+ SRAM SRAM_START SRAM_SIZE
{
-#ifndef ETHOSU_FAST_MEMORY_SIZE
- ; Place tensor arena in SRAM if we do not have a fast memory area
+ #if (ETHOSU_MODEL == 0)
+ ; Place network model in SRAM
+ * (network_model_sec)
+ #endif
+
+ #if (ETHOSU_ARENA == 0)
+ ; Place tensor arena in SRAM
* (.bss.tensor_arena)
-#else
- * (.bss.ethosu_scratch)
-#endif
+ #endif
+
+ ; Place scratch buffer in SRAM
+ * (.bss.ethosu_scratch)
}
ARM_LIB_HEAP (STACK_HEAP - STACK_SIZE - __STACKSEAL_SIZE - HEAP_SIZE) EMPTY ALIGN 8 HEAP_SIZE {}
@@ -185,19 +201,22 @@
{
}
#else //trustzone secure or non-trustzone
+ ; Place model and its affiliates in DRAM
{
+ #if (ETHOSU_MODEL == 1)
* (network_model_sec)
+ #endif
* (input_data_sec)
* (expected_output_data_sec)
* (output_data_sec)
}
-#ifdef ETHOSU_FAST_MEMORY_SIZE
- ; Place tensor arena in DRAM if we have a fast memory area
- ARENA +0 UNINIT ALIGN 16
- {
- * (.bss.tensor_arena)
- }
-#endif
+ #if (ETHOSU_ARENA == 1)
+ ; Place tensor arena in DRAM if we have a fast memory area
+ ARENA +0 UNINIT ALIGN 16
+ {
+ * (.bss.tensor_arena)
+ }
+ #endif
#endif
}