From 7ed56dd5808d483ab8e294f36e3ecf8e8ddd84f1 Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Wed, 30 Sep 2020 20:22:56 +0100 Subject: MLCE-233 Added ability to enable fast_math through the NNAPI driver * Added -a command line option to enable fast_math on Neon/CL when launching the NNAPI driver * The invocation will look something like: /vendor/bin/hw/android.hardware.neuralnetworks@1.3-service-armnn -v -c GpuAcc -n arm-armnn -a * Enabling fast_math may cause an increase in VTS or CTS test failures due to a change in precision Signed-off-by: Mike Kelly Change-Id: Ifd8931f95efd65d3ae3aca9d9d61bd381e4e24a5 --- ArmnnDriverImpl.cpp | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'ArmnnDriverImpl.cpp') diff --git a/ArmnnDriverImpl.cpp b/ArmnnDriverImpl.cpp index 9c6d51fd..936cc6e5 100644 --- a/ArmnnDriverImpl.cpp +++ b/ArmnnDriverImpl.cpp @@ -53,7 +53,6 @@ Return FailPrepareModel(V1_0::ErrorStatus error, return error; } - } // namespace namespace armnn_driver @@ -105,6 +104,17 @@ Return ArmnnDriverImpl::prepareModel( armnn::OptimizerOptions OptOptions; OptOptions.m_ReduceFp32ToFp16 = float32ToFloat16; + armnn::BackendOptions gpuAcc("GpuAcc", + { + { "FastMathEnabled", options.IsFastMathEnabled() } + }); + armnn::BackendOptions cpuAcc("CpuAcc", + { + { "FastMathEnabled", options.IsFastMathEnabled() } + }); + OptOptions.m_ModelOptions.push_back(gpuAcc); + OptOptions.m_ModelOptions.push_back(cpuAcc); + std::vector errMessages; try { -- cgit v1.2.1