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 --- 1.2/ArmnnDriverImpl.cpp | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to '1.2') diff --git a/1.2/ArmnnDriverImpl.cpp b/1.2/ArmnnDriverImpl.cpp index 4571fe0e..6d713e06 100644 --- a/1.2/ArmnnDriverImpl.cpp +++ b/1.2/ArmnnDriverImpl.cpp @@ -130,6 +130,17 @@ Return ArmnnDriverImpl::prepareArmnnModel_1_2( 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