From 0b9c9a3873da3d368e184308f4f9a4c202e3fb67 Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Tue, 15 Sep 2020 14:05:38 +0200 Subject: MLBEDSW-2337: Intermediate feature maps in fast storage Attempts to use fast storage for feature maps used in between cascaded passes. This is only relevant for system configurations where feature maps are by default not placed in SRAM, but there is SRAM for fast storage. Change-Id: I207b7cf32cfcb5bea3e6b93c2da1161c4af5221d Signed-off-by: Louis Verhaard --- ethosu/vela/nn_graph.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'ethosu/vela/nn_graph.py') diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index 21cd80b9..58aab611 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -141,6 +141,8 @@ class Subgraph: self.placement = placement self.command_stream_tensor = None self.flash_tensor = None + # Scratch information locally used in the scheduler + self.scheduling_info = {} self.memory_used = {} self.memory_used_per_type = {} -- cgit v1.2.1