aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/register_command_stream_generator.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/register_command_stream_generator.py')
-rw-r--r--ethosu/vela/register_command_stream_generator.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/register_command_stream_generator.py b/ethosu/vela/register_command_stream_generator.py
index 733be590..fb705b96 100644
--- a/ethosu/vela/register_command_stream_generator.py
+++ b/ethosu/vela/register_command_stream_generator.py
@@ -176,7 +176,8 @@ class CommandStreamEmitter:
self.offset += CommandStreamEmitter.WORD_SIZE
def cmd1_with_offset(self, cmd: cmd1, offset, param=0x0):
- offset = int(offset) & 0xFFFFFFFFF
+ offset = int(offset) & 0xFFFFFFFF
+ param = int(param) & 0xFFFF
command = cmd.value | CmdMode.Payload32.value | (param << 16)
if not self.get_reg_machine(cmd).set_register(cmd, (command, offset)):