commit 940fdee086d77089c7669190392c92f4f2150348
author Dwight Lidman <dwight.lidman@arm.com> Thu Aug 13 13:11:48 2020 +0200
committer tim.hall <tim.hall@arm.com> Tue Aug 18 18:52:46 2020 +0000
tree baa7769b3567ebd5b71491ad3d6a97772b7efbd0
parent 29d568e73373e82ef2dd246a64ba399fe6da676a

MLBEDSW-2589: Skip weight compression for CPU ops

This commit fixes a bug where CPU ops were getting
passed on as NPU ops in weight_compressor.py due to
Operation.find_npu_op() incorrectly returning any
op with an 'npu_block_type' attribute (which every
op has) as an NPU op.

Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: I7a758f8d1b1237907816bc1be7b77aff765ae688

ethosu/vela/tensor.py

diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 312e8f3..c41a7eb 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -626,7 +626,7 @@
         for op in self.consumers():
             if op.type == "DMA":
                 return op.outputs[0].find_npu_op()
-            if "npu_block_type" in op.attrs:
+            if op.run_on_npu:
                 return op
             return None