From 631f600eeaeda4814bda720b3979f8aef33edbea Mon Sep 17 00:00:00 2001 From: William Isaksson Date: Wed, 2 Aug 2023 11:37:05 +0000 Subject: MLBEDSW-7865: Vela duplicates outputs We now don't rewrite tensors if the tensor is already an output tensor of the current subgraph Signed-off-by: William Isaksson Change-Id: I9cb36d830616a69d35180326437ff53bcaa62d71 --- ethosu/vela/extract_npu_subgraphs.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/ethosu/vela/extract_npu_subgraphs.py b/ethosu/vela/extract_npu_subgraphs.py index dcc86878..c4f66b8f 100644 --- a/ethosu/vela/extract_npu_subgraphs.py +++ b/ethosu/vela/extract_npu_subgraphs.py @@ -246,12 +246,13 @@ def extract_subgraph(nng, orig_sg, arch): need_rewrite = True break for orig_out_tens in orig_sg.output_tensors: - if tens == orig_out_tens: - need_rewrite = True - elif tens.equivalence_id == orig_out_tens.equivalence_id: - need_rewrite = True - multiple_npu_sg_have_same_cpu_out_tens = True - output_tensor = orig_out_tens + if tens not in curr_sg.output_tensors: + if tens == orig_out_tens: + need_rewrite = True + elif tens.equivalence_id == orig_out_tens.equivalence_id: + need_rewrite = True + multiple_npu_sg_have_same_cpu_out_tens = True + output_tensor = orig_out_tens if need_rewrite: rewrite_tensor_npu_producer_cpu_consumers( -- cgit v1.2.1