diff --git a/modules/ocl/test/main.cpp b/modules/ocl/test/main.cpp index cafefa5ab..d76fa8483 100644 --- a/modules/ocl/test/main.cpp +++ b/modules/ocl/test/main.cpp @@ -55,5 +55,24 @@ #include "opencv2/ocl/private/opencl_dumpinfo.hpp" +int LOOP_TIMES = 1; -CV_TEST_MAIN(".", dumpOpenCLDevice()) +void readLoopTimes(int argc, char ** argv) +{ + const char * const command_line_keys = + "{ |test_loop_times |1 |count of iterations per each test}" + "{h |help |false |print help info}"; + + cv::CommandLineParser parser(argc, argv, command_line_keys); + if (parser.get("help")) + { + std::cout << "\nAvailable options besides google test option: \n"; + parser.printParams(); + } + + LOOP_TIMES = parser.get("test_loop_times"); + CV_Assert(LOOP_TIMES > 0); +} + +CV_TEST_MAIN(".", dumpOpenCLDevice(), + readLoopTimes(argc, argv)) diff --git a/modules/ocl/test/utility.hpp b/modules/ocl/test/utility.hpp index 7501b19fc..2ec7001ac 100644 --- a/modules/ocl/test/utility.hpp +++ b/modules/ocl/test/utility.hpp @@ -42,7 +42,7 @@ #ifndef __OPENCV_TEST_UTILITY_HPP__ #define __OPENCV_TEST_UTILITY_HPP__ -#define LOOP_TIMES 1 +extern int LOOP_TIMES; #define MWIDTH 256 #define MHEIGHT 256