diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2019-04-09 13:10:46 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-04-15 15:29:37 +0000 |
commit | 3e68b97946bfee3c89ec2d4363a22550a10b2e55 (patch) | |
tree | d7d9be27aff46a18ac9442092d908223f435dcd7 /tests/InferenceTest.inl | |
parent | 200e38039cf2cef21ae9ba6f86fab6fd524e5077 (diff) | |
download | armnn-3e68b97946bfee3c89ec2d4363a22550a10b2e55.tar.gz |
IVGCVSW-2928 Fix issue with GPU profiling
Correctly enable GPU profiling when test profiling is enabled.
Remove extra copy of the profiling-enabled flag from InferenceModel::Params
and correctly pass around the copy that is in InferenceTestOptions.
!referencetests:180329
Change-Id: I0daa1bab2e7068fc479bf417a553183b1d922166
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
Diffstat (limited to 'tests/InferenceTest.inl')
-rw-r--r-- | tests/InferenceTest.inl | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl index 538720bd83..0112037bc3 100644 --- a/tests/InferenceTest.inl +++ b/tests/InferenceTest.inl @@ -162,7 +162,8 @@ void ClassifierTestCaseProvider<TDatabase, InferenceModel>::AddCommandLineOption } template <typename TDatabase, typename InferenceModel> -bool ClassifierTestCaseProvider<TDatabase, InferenceModel>::ProcessCommandLineOptions() +bool ClassifierTestCaseProvider<TDatabase, InferenceModel>::ProcessCommandLineOptions( + const InferenceTestOptions& commonOptions) { if (!ValidateDirectory(m_DataDir)) { @@ -171,7 +172,7 @@ bool ClassifierTestCaseProvider<TDatabase, InferenceModel>::ProcessCommandLineOp ReadPredictions(); - m_Model = m_ConstructModel(m_ModelCommandLineOptions); + m_Model = m_ConstructModel(commonOptions, m_ModelCommandLineOptions); if (!m_Model) { return false; @@ -336,7 +337,8 @@ int ClassifierInferenceTestMain(int argc, return make_unique<TestCaseProvider>(constructDatabase, [&] - (typename InferenceModel::CommandLineOptions modelOptions) + (const InferenceTestOptions &commonOptions, + typename InferenceModel::CommandLineOptions modelOptions) { if (!ValidateDirectory(modelOptions.m_ModelDir)) { @@ -358,7 +360,7 @@ int ClassifierInferenceTestMain(int argc, modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel; modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode; - return std::make_unique<InferenceModel>(modelParams); + return std::make_unique<InferenceModel>(modelParams, commonOptions.m_EnableProfiling); }); }); } |