diff options
| author | Tim Hall <tim.hall@arm.com> | 2020-06-25 15:04:31 +0100 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2020-06-25 15:07:58 +0100 |
| commit | f7e810a695c1426799d945d61671126543efc123 (patch) | |
| tree | dad3bfd9120bdcd92d3fe216a01a4e798276a080 /ethosu/vela/driver_actions.py | |
| parent | eca2e95e1fea150d8a942f8b5f0a4d9d7aefebc1 (diff) | |
| download | ethos-u-vela-f7e810a695c1426799d945d61671126543efc123.tar.gz | |
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>
Diffstat (limited to 'ethosu/vela/driver_actions.py')
| -rw-r--r-- | ethosu/vela/driver_actions.py | 4 |
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 |