diff options
author | Diqing Zhong <diqing.zhong@arm.com> | 2020-12-09 15:22:40 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-12-14 06:58:24 +0000 |
commit | 94457b175b8646bce089c9924e99686587de8992 (patch) | |
tree | ef83eac89ff7f828c5a517c1ba48b2358ffc8bb1 /ethosu/vela/insert_dma.py | |
parent | 9bfe0f86ea525055954b160a3c678024743030ec (diff) | |
download | ethos-u-vela-94457b175b8646bce089c9924e99686587de8992.tar.gz |
Revert "Revert "MLMBED-3450: Do not convert batched fully connected to conv""
- We have combined estimates for conv and fc, add the fix back
Change-Id: I49a29c716189b37b387df4b46efab5f4e6125994
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>
Diffstat (limited to 'ethosu/vela/insert_dma.py')
-rw-r--r-- | ethosu/vela/insert_dma.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/ethosu/vela/insert_dma.py b/ethosu/vela/insert_dma.py index f02039cb..fc1e7986 100644 --- a/ethosu/vela/insert_dma.py +++ b/ethosu/vela/insert_dma.py @@ -77,9 +77,7 @@ def insert_dma_cmd(op, arch, nng): ): only_vector_product_consumers = True for oper in tens.consumers(): - if oper is None or not ( - oper.type.npu_block_type == NpuBlockType.VectorProduct or "is_converted_fc" in oper.attrs - ): + if oper is None or oper.type.npu_block_type != NpuBlockType.VectorProduct: only_vector_product_consumers = False break |