diff options
Diffstat (limited to 'ethosu/vela/test/test_architecture_allocator.py')
-rw-r--r-- | ethosu/vela/test/test_architecture_allocator.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/test/test_architecture_allocator.py b/ethosu/vela/test/test_architecture_allocator.py index 94768fc1..a199e9d8 100644 --- a/ethosu/vela/test/test_architecture_allocator.py +++ b/ethosu/vela/test/test_architecture_allocator.py @@ -106,6 +106,7 @@ def test_allocate(test_data): Block.from_shape(config.ofm_block.as_list()), arch, block_type, + ofm_shape, ifm_shape, ifm2_shape, is_partkernel=config.is_partkernel, |