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 --- DriverOptions.cpp | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'DriverOptions.cpp') diff --git a/DriverOptions.cpp b/DriverOptions.cpp index 1f534b6e..d179d653 100644 --- a/DriverOptions.cpp +++ b/DriverOptions.cpp @@ -35,6 +35,7 @@ DriverOptions::DriverOptions(armnn::Compute computeDevice, bool fp16Enabled) , m_ClTuningLevel(armnn::IGpuAccTunedParameters::TuningLevel::Rapid) , m_EnableGpuProfiling(false) , m_fp16Enabled(fp16Enabled) + , m_FastMathEnabled(false) { } @@ -45,6 +46,7 @@ DriverOptions::DriverOptions(const std::vector& backends, bool , m_ClTuningLevel(armnn::IGpuAccTunedParameters::TuningLevel::Rapid) , m_EnableGpuProfiling(false) , m_fp16Enabled(fp16Enabled) + , m_FastMathEnabled(false) { } @@ -54,6 +56,7 @@ DriverOptions::DriverOptions(int argc, char** argv) , m_ClTuningLevel(armnn::IGpuAccTunedParameters::TuningLevel::Rapid) , m_EnableGpuProfiling(false) , m_fp16Enabled(false) + , m_FastMathEnabled(false) { namespace po = boost::program_options; @@ -104,6 +107,10 @@ DriverOptions::DriverOptions(int argc, char** argv) "exhaustive approach " "rapid: only 3 lws values should be tested for each kernel ") + ("fast-math,a", + po::bool_switch(&m_FastMathEnabled), + "Turns FastMath on") + ("gpu-profiling,p", po::bool_switch(&m_EnableGpuProfiling), "Turns GPU profiling on") -- cgit v1.2.1