diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-04-15 17:01:17 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-04-16 13:09:58 +0000 |
commit | e22f96b9d1a1a87398b901d34cd52f11a5647f96 (patch) | |
tree | faa0734e5428ff57b892e713ab3a9033632e9d24 /ethosu/vela/graph_optimiser.py | |
parent | ad656a820394580d9e1b6d79597905074d599464 (diff) | |
download | ethos-u-vela-e22f96b9d1a1a87398b901d34cd52f11a5647f96.tar.gz |
MLBEDSW Vela: Fix check for format restrictions
Fixed the check related to if there are any CPU
producers/consumers.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I0ed08c650d1ca34e8e148aee68a5ed09c25fdd87
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
-rw-r--r-- | ethosu/vela/graph_optimiser.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index b708b62e..f59b685a 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -263,9 +263,10 @@ def check_format_restrictions(tens, arch): ): return - if not any(cons.run_on_npu for cons in tens.consumer_list): + # Check if any of the producers/consumers is run on CPU + if not all(cons.run_on_npu for cons in tens.consumer_list): return - if not any(prod.run_on_npu for prod in tens.ops): + if not all(prod.run_on_npu for prod in tens.ops): return # "Concat" ofm exception: |