aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/driver_actions.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/driver_actions.py')
-rw-r--r--ethosu/vela/driver_actions.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/driver_actions.py b/ethosu/vela/driver_actions.py
index 79ac11a1..29c2b181 100644
--- a/ethosu/vela/driver_actions.py
+++ b/ethosu/vela/driver_actions.py
@@ -65,9 +65,9 @@ def build_id_word():
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