diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-11-01 15:59:50 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-05-08 13:16:01 +0000 |
commit | 1fe6c8170ae2fe90b53fb71b7570aec9dfe75c45 (patch) | |
tree | bbb846edda64445c1e033b182e5a079c8d5728d8 /src/backends/neon/NeonLayerSupport.cpp | |
parent | c52190a7e80cf238ba1d8630e5cc36ec7c7849e2 (diff) | |
download | armnn-1fe6c8170ae2fe90b53fb71b7570aec9dfe75c45.tar.gz |
IVGCVSW-7307 Add CpuAcc Batch MatMul Workload
* Call dedicated MatMul kernel in ACL
* Add int8 tests
* Add int8 to documentation
* Force tensors to be dynamic (nonConst) as per request of ACL
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I992ae9aae1174214607bf29305f21cdeaf3fdc1b
Diffstat (limited to 'src/backends/neon/NeonLayerSupport.cpp')
-rw-r--r-- | src/backends/neon/NeonLayerSupport.cpp | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp index 4e4d7fa73d..cd4dca8edb 100644 --- a/src/backends/neon/NeonLayerSupport.cpp +++ b/src/backends/neon/NeonLayerSupport.cpp @@ -760,12 +760,28 @@ bool NeonLayerSupport::IsBatchMatMulSupported(const TensorInfo& inputX, const BatchMatMulDescriptor& descriptor, Optional<std::string&> reasonIfUnsupported) const { + bool isFastMathEnabled = false; +#if defined(ARMCOMPUTENEON_ENABLED) + if (m_ModelContextPtr) + { + if (m_ModelContextPtr.get() != nullptr) + { + auto modelOptions = dynamic_cast<NeonBackendModelContext*>(m_ModelContextPtr.get()); + if (modelOptions) + { + isFastMathEnabled = modelOptions->IsFastMathEnabled(); + } + } + } +#endif FORWARD_WORKLOAD_VALIDATE_FUNC(NeonBatchMatMulValidate, reasonIfUnsupported, inputX, inputY, output, - descriptor); + descriptor, + isFastMathEnabled, + nullptr); } bool NeonLayerSupport::IsBatchNormalizationSupported(const TensorInfo& input, |