From c21d1c76b21f337d46990c2c262bda536f15e930 Mon Sep 17 00:00:00 2001 From: Louis Verhaard Date: Wed, 25 Nov 2020 20:58:51 +0100 Subject: MLBEDSW-3583: Prevent DMA for converted FullyConnected Do not use DMA for weights of a FullyConnected op that has been converted to a Conv2D. Change-Id: Ibf6710c0a1723c8b48c563ca204f274af5ca88ce Signed-off-by: Louis Verhaard --- ethosu/vela/insert_dma.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'ethosu/vela/insert_dma.py') diff --git a/ethosu/vela/insert_dma.py b/ethosu/vela/insert_dma.py index fc1e7986..f02039cb 100644 --- a/ethosu/vela/insert_dma.py +++ b/ethosu/vela/insert_dma.py @@ -77,7 +77,9 @@ def insert_dma_cmd(op, arch, nng): ): only_vector_product_consumers = True for oper in tens.consumers(): - if oper is None or oper.type.npu_block_type != NpuBlockType.VectorProduct: + if oper is None or not ( + oper.type.npu_block_type == NpuBlockType.VectorProduct or "is_converted_fc" in oper.attrs + ): only_vector_product_consumers = False break -- cgit v1.2.1