commit e22f96b9d1a1a87398b901d34cd52f11a5647f96
author Patrik Gustavsson <patrik.gustavsson@arm.com> Thu Apr 15 17:01:17 2021 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Fri Apr 16 13:09:58 2021 +0000
tree faa0734e5428ff57b892e713ab3a9033632e9d24
parent ad656a820394580d9e1b6d79597905074d599464

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

ethosu/vela/graph_optimiser.py

diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py
index b708b62..f59b685 100644
--- a/ethosu/vela/graph_optimiser.py
+++ b/ethosu/vela/graph_optimiser.py
@@ -263,9 +263,10 @@
     ):
         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: