diff options
| author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-08-27 13:43:32 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-08-27 15:27:46 +0100 |
| commit | a14ac50275fabb53682648f1f14e7fe7d425b4c0 (patch) | |
| tree | fec505a86811719208d9746dcb860feb02592bbd | |
| parent | 5337a29b6179c730080f4df068a4804ee26d5464 (diff) | |
| download | armnn-a14ac50275fabb53682648f1f14e7fe7d425b4c0.tar.gz | |
IVGCVSW-3277 Fix RequiresCopy in case a tensor handle factory is not available
* Added null checks prior of using either the source or the destination
handle factory when checking if the tensor requires to be copied
Change-Id: Icc6737562849be5e7d1a62243327366e06312eb9
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
| -rw-r--r-- | src/armnn/Network.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 932f9eb49e..0ef14075b6 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -441,7 +441,8 @@ bool RequiresCopy(ITensorHandleFactory::FactoryId src, ITensorHandleFactory* srcFactory = registry.GetFactory(src); ITensorHandleFactory* dstFactory = registry.GetFactory(dst); - if ((srcFactory->GetExportFlags() & dstFactory->GetImportFlags()) != 0) + if (srcFactory && dstFactory && + (srcFactory->GetExportFlags() & dstFactory->GetImportFlags()) != 0) { return false; } |