diff options
author | Andreas Nevalainen <andreas.nevalainen@arm.com> | 2020-10-14 13:55:43 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-10-19 15:01:27 +0000 |
commit | 6e827082524af57bf04833c30754384b46216e59 (patch) | |
tree | 2e7c621b061c91be19770f85640b2d5da8e6803c /ethosu/vela/register_command_stream_generator.py | |
parent | 1f951fc47abd52db0ac048802dab0c95b149d7b8 (diff) | |
download | ethos-u-vela-6e827082524af57bf04833c30754384b46216e59.tar.gz |
MLBEDSW-3194: Updated elementwise IFM banks count
Signed-off-by: Andreas Nevalainen <andreas.nevalainen@arm.com>
Change-Id: Ie404a0c13e7c7de0eff649f77e0147a0f3d73acd
Diffstat (limited to 'ethosu/vela/register_command_stream_generator.py')
-rw-r--r-- | ethosu/vela/register_command_stream_generator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py index 073b50fb..4f3fe7d4 100644 --- a/ethosu/vela/register_command_stream_generator.py +++ b/ethosu/vela/register_command_stream_generator.py @@ -585,7 +585,7 @@ def generate_register_command_stream(nng, sg, arch, verbose=False): # Set IFM2_IB_START to the latter half of the IB space ifm_ib_start = shared_buffer.bank_locations[SharedBufferArea.IFM] emit.cmd0_with_param( - cmd0.NPU_SET_IFM2_IB_START, (shram_required - ifm_ib_start) / 2 + ifm_ib_start + cmd0.NPU_SET_IFM2_IB_START, (shram_required - ifm_ib_start) // shared_buffer.ifm_count + ifm_ib_start ) emit.cmd0_with_param(cmd0.NPU_SET_IFM2_BROADCAST, ifm2_broadcast) |