diff options
| author | Michael McGeagh <michael.mcgeagh@arm.com> | 2020-12-02 12:39:03 +0000 |
|---|---|---|
| committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-12-08 14:58:19 +0000 |
| commit | f3e3ad746de943e312c290d614184c0f6559ac1f (patch) | |
| tree | 44f6eac5a6b1aec4114ad2c2faddced820ea1e17 /ethosu/vela/high_level_command_stream.py | |
| parent | d9c2c4258c50f04ef3a6c3849508d317249e8ebf (diff) | |
| download | ethos-u-vela-f3e3ad746de943e312c290d614184c0f6559ac1f.tar.gz | |
MLBEDSW-2836 Change sets to tuples
Replace conditional checks against sets with tuples.
If not requiring uniqueness, or complex set operations, it is quicker to
use tuples instead.
Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: Ie8732c8d46067244963936c53f0ec81adda50372
Diffstat (limited to 'ethosu/vela/high_level_command_stream.py')
| -rw-r--r-- | ethosu/vela/high_level_command_stream.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/high_level_command_stream.py b/ethosu/vela/high_level_command_stream.py index 30e22ffe..d057d17e 100644 --- a/ethosu/vela/high_level_command_stream.py +++ b/ethosu/vela/high_level_command_stream.py @@ -55,8 +55,8 @@ class Box: new_start_coord[idx] += split_offset[idx] new_end_coord[idx] += split_offset[idx] - if split_offset is None and npu_block_type in set( - (NpuBlockType.ConvolutionMxN, NpuBlockType.VectorProduct, NpuBlockType.ReduceSum) + if (split_offset is None) and ( + npu_block_type in (NpuBlockType.ConvolutionMxN, NpuBlockType.VectorProduct, NpuBlockType.ReduceSum) ): # these types of operations do a "dot product" or sum over the entire IFM new_start_coord[-1] = 0 |