vela: MLBEDSW-828 weight/scale stream interleaving
- Multicore weight and scale stream interleaving for
multicore hardware architecture.
Change-Id: Ic82850463391c629d90d08c26cf0c48dd438286d
Signed-off-by: Tim Hall <tim.hall@arm.com>
diff --git a/ethosu/vela/driver_actions.py b/ethosu/vela/driver_actions.py
index 79ac11a..29c2b18 100644
--- a/ethosu/vela/driver_actions.py
+++ b/ethosu/vela/driver_actions.py
@@ -65,9 +65,9 @@
def build_config_word(arch):
- macs_cc = arch.config.macs
+ macs_cc = arch.ncores * arch.config.macs
log2_macs_cc = int(np.log2(macs_cc) + 0.5)
- shram_size = int(arch.shram_size_bytes / 1024)
+ shram_size = arch.ncores * int(arch.shram_size_bytes / 1024)
n = config_r()
n.set_shram_size(shram_size)
n.set_cmd_stream_version(0) # may be incremented in the future