diff --git a/deploy/cpp_infer/src/main.cpp b/deploy/cpp_infer/src/main.cpp index 51e90544f70d33257661c72fc9e11b9b9ba8555a..3f3cae9a24b4a1ded3a0a6a4f2a505a18c5819ca 100644 --- a/deploy/cpp_infer/src/main.cpp +++ b/deploy/cpp_infer/src/main.cpp @@ -69,17 +69,10 @@ int main(int argc, char **argv) { config.char_list_file); #ifdef USE_MKL - int i; - std::cout << ">>>>>>>>>>>>>>> Testing MKL <<<<<<<<<<<<<<<<< " << std::endl; -#pragma omp parallel for private(i) - for (i = 0; i < 50; i++) { - std::cout << "omp_get_num_threads(): " << omp_get_num_threads() - << std::endl; - int tid = omp_get_thread_num(); - std::cout << "Log from omp thread " << tid << std::endl; +#pragma omp parallel + for (auto i = 0; i < 10; i++) { + assert(config.cpu_math_library_num_threads == omp_get_num_threads()); } - std::cout << ">>>>>>>>>>>>>>> Testing MKL Finished <<<<<<<<<<<<<<<<< " - << std::endl; #endif auto start = std::chrono::system_clock::now();