From 34b1bd2a3575082b3ea10f21ac506a5e3a39a444 Mon Sep 17 00:00:00 2001
From: Kristofer Jonsson <kristofer.jonsson@arm.com>
Date: Mon, 19 Sep 2022 14:14:39 +0200
Subject: Move LSTM kernel to BRAM

Change-Id: Id0244bb35236d0b04732e3fddbaa472686c63483
---
 targets/corstone-300/platform.ld      | 6 +++---
 targets/corstone-300/platform.scatter | 5 +++++
 2 files changed, 8 insertions(+), 3 deletions(-)

(limited to 'targets')

diff --git a/targets/corstone-300/platform.ld b/targets/corstone-300/platform.ld
index d22b786..ebf7021 100644
--- a/targets/corstone-300/platform.ld
+++ b/targets/corstone-300/platform.ld
@@ -135,7 +135,7 @@ SECTIONS
   .text :
   {
     KEEP(*(.vectors))
-    *(.text*)
+    *(EXCLUDE_FILE(lstm_eval.*) .text*)
 
     KEEP(*(.init))
     KEEP(*(.fini))
@@ -219,8 +219,7 @@ SECTIONS
   {
     __data_start__ = .;
     *(vtable)
-    *(.data)
-    *(.data.*)
+    *(EXCLUDE_FILE(lstm_eval.*) .data .data.*)
 
     . = ALIGN(4);
     /* preinit data */
@@ -289,6 +288,7 @@ SECTIONS
   .sram.data : {
     __sram_data_start__ = .;
     *(.sram.data)
+    lstm_eval.*(.text* .data .data.*)
     __sram_data_end__ = .;
   } > BRAM AT >DDR :rom_dram
 
diff --git a/targets/corstone-300/platform.scatter b/targets/corstone-300/platform.scatter
index d683100..85c6549 100644
--- a/targets/corstone-300/platform.scatter
+++ b/targets/corstone-300/platform.scatter
@@ -182,6 +182,11 @@ LOAD_REGION_BRAM BRAM_START BRAM_SIZE
         * (.sram.data)
     }
 #endif
+
+    TFLM +0
+    {
+        lstm_eval.o
+    }
 }
 
 LOAD_REGION_SRAM SRAM_START SRAM_SIZE
-- 
cgit v1.2.1