aboutsummaryrefslogtreecommitdiff
path: root/tests/InferenceTest.hpp
diff options
context:
space:
mode:
authorMatthew Bentham <matthew.bentham@arm.com>2019-04-09 13:10:46 +0100
committerMatteo Martincigh <matteo.martincigh@arm.com>2019-04-15 15:29:37 +0000
commit3e68b97946bfee3c89ec2d4363a22550a10b2e55 (patch)
treed7d9be27aff46a18ac9442092d908223f435dcd7 /tests/InferenceTest.hpp
parent200e38039cf2cef21ae9ba6f86fab6fd524e5077 (diff)
downloadarmnn-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.hpp')
-rw-r--r--tests/InferenceTest.hpp7
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/InferenceTest.hpp b/tests/InferenceTest.hpp
index 91a65ea494..3ebfdbcc3c 100644
--- a/tests/InferenceTest.hpp
+++ b/tests/InferenceTest.hpp
@@ -91,7 +91,7 @@ public:
virtual ~IInferenceTestCaseProvider() {}
virtual void AddCommandLineOptions(boost::program_options::options_description& options) {};
- virtual bool ProcessCommandLineOptions() { return true; };
+ virtual bool ProcessCommandLineOptions(const InferenceTestOptions &commonOptions) { return true; };
virtual std::unique_ptr<IInferenceTestCase> GetTestCase(unsigned int testCaseId) = 0;
virtual bool OnInferenceTestFinished() { return true; };
};
@@ -219,7 +219,7 @@ public:
ClassifierTestCaseProvider(TConstructDatabaseCallable constructDatabase, TConstructModelCallable constructModel);
virtual void AddCommandLineOptions(boost::program_options::options_description& options) override;
- virtual bool ProcessCommandLineOptions() override;
+ virtual bool ProcessCommandLineOptions(const InferenceTestOptions &commonOptions) override;
virtual std::unique_ptr<IInferenceTestCase> GetTestCase(unsigned int testCaseId) override;
virtual bool OnInferenceTestFinished() override;
@@ -227,7 +227,8 @@ private:
void ReadPredictions();
typename InferenceModel::CommandLineOptions m_ModelCommandLineOptions;
- std::function<std::unique_ptr<InferenceModel>(typename InferenceModel::CommandLineOptions)> m_ConstructModel;
+ std::function<std::unique_ptr<InferenceModel>(const InferenceTestOptions& commonOptions,
+ typename InferenceModel::CommandLineOptions)> m_ConstructModel;
std::unique_ptr<InferenceModel> m_Model;
std::string m_DataDir;