diff options
Diffstat (limited to 'ethosu/vela/high_level_command_stream_generator.py')
-rw-r--r-- | ethosu/vela/high_level_command_stream_generator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/high_level_command_stream_generator.py b/ethosu/vela/high_level_command_stream_generator.py index d34fb75b..d5a6341b 100644 --- a/ethosu/vela/high_level_command_stream_generator.py +++ b/ethosu/vela/high_level_command_stream_generator.py @@ -32,7 +32,7 @@ def dma_if_necessary(ps, box, tensor): if tensor.needs_dma(): dma_op = tensor.ops[0] in_tensor = dma_op.inputs[0] - yield DMA(in_tensor, tensor, box) + yield DMA(ps, in_tensor, tensor, box) def match_tensor(source, derived): |