diff options
Diffstat (limited to 'tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp')
| -rw-r--r-- | tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp index c6ffe3d989..c136672c48 100644 --- a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp +++ b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp @@ -22,7 +22,8 @@ int main(int argc, char* argv[]) { return make_unique<YoloTestCaseProvider<YoloInferenceModel>>( [&] - (typename YoloInferenceModel::CommandLineOptions modelOptions) + (const InferenceTestOptions &commonOptions, + typename YoloInferenceModel::CommandLineOptions modelOptions) { if (!ValidateDirectory(modelOptions.m_ModelDir)) { @@ -39,7 +40,7 @@ int main(int argc, char* argv[]) modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel; modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode; - return std::make_unique<YoloInferenceModel>(modelParams); + return std::make_unique<YoloInferenceModel>(modelParams, commonOptions.m_EnableProfiling); }); }); } |