diff options
Diffstat (limited to 'ethosu/vela/register_command_stream_util.py')
| -rw-r--r-- | ethosu/vela/register_command_stream_util.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/register_command_stream_util.py b/ethosu/vela/register_command_stream_util.py index ca7e6bc6..ce49fc29 100644 --- a/ethosu/vela/register_command_stream_util.py +++ b/ethosu/vela/register_command_stream_util.py @@ -46,7 +46,7 @@ from ethosu.vela.range_set import MemoryRangeSet BASE_PTR_INDEX_MEM2MEM = int((1 << 8) | (3 << 0)) -UNARY_ELEMWISE_OPS = set((NpuElementWiseOp.ABS, NpuElementWiseOp.LRELU, NpuElementWiseOp.CLZ,)) +UNARY_ELEMWISE_OPS = (NpuElementWiseOp.ABS, NpuElementWiseOp.LRELU, NpuElementWiseOp.CLZ) def to_npu_kernel(kernel: Kernel) -> NpuKernel: |