diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2021-02-11 16:16:42 +0000 |
|---|---|---|
| committer | KeithARM <keith.davis@arm.com> | 2021-02-15 15:04:31 +0000 |
| commit | 406463269f55a5baefb941b51e10f423f6d3250a (patch) | |
| tree | 2299b75c47cae44d9c8be4bf670f30c9d65a01a8 /tests/InferenceModel.hpp | |
| parent | b55d366a93d8d3699ce5e906134c695c8040c27e (diff) | |
| download | armnn-406463269f55a5baefb941b51e10f423f6d3250a.tar.gz | |
IVGCVSW-5686 Add GpuAcc MLGO tuning file configuration argument
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I3f320499c379162f9d1b00cc8816bd144cd7eee4
Diffstat (limited to 'tests/InferenceModel.hpp')
| -rw-r--r-- | tests/InferenceModel.hpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index d20bb2271f..79962623cb 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -100,6 +100,8 @@ struct Params bool m_SaveCachedNetwork; std::string m_CachedNetworkFilePath; unsigned int m_NumberOfThreads; + std::string m_MLGOTuningFilePath; + Params() : m_ComputeDevices{} @@ -115,6 +117,7 @@ struct Params , m_SaveCachedNetwork(false) , m_CachedNetworkFilePath("") , m_NumberOfThreads(0) + , m_MLGOTuningFilePath("") {} }; @@ -434,8 +437,10 @@ public: { { "FastMathEnabled", params.m_EnableFastMath }, { "SaveCachedNetwork", params.m_SaveCachedNetwork }, - { "CachedNetworkFilePath", params.m_CachedNetworkFilePath } + { "CachedNetworkFilePath", params.m_CachedNetworkFilePath }, + { "MLGOTuningFilePath", params.m_MLGOTuningFilePath } }); + armnn::BackendOptions cpuAcc("CpuAcc", { { "FastMathEnabled", params.m_EnableFastMath }, |