diff options
Diffstat (limited to 'ethosu/vela/register_command_stream_generator.py')
| -rw-r--r-- | ethosu/vela/register_command_stream_generator.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py index 42ae99d..9d9a1e6 100644 --- a/ethosu/vela/register_command_stream_generator.py +++ b/ethosu/vela/register_command_stream_generator.py @@ -531,6 +531,7 @@ def generate_weights(emit: CommandStreamEmitter, weights: List[NpuAddressRange], emit.cmd1_with_offset(length, weights[core].length) elif core < arch.ncores: check_alignment(weights[0].address, 16) + check_length(weights[0].length, 16) emit.cmd1_with_address(addr, weights[0].address) emit.cmd1_with_offset(length, 0) |