Correctly handle --threads param

This commit is contained in:
karl 2020-12-16 19:22:02 +01:00
parent 4c687844d3
commit 6d1d360ec0

View File

@ -421,7 +421,6 @@ int main(int argc, char* argv[]) {
Mode mode = Mode::SEQ; Mode mode = Mode::SEQ;
bool use_gpu = false; bool use_gpu = false;
int num_generations = 0; int num_generations = 0;
int num_threads = 1;
bool measure = false; bool measure = false;
if (argc < 8) { if (argc < 8) {
@ -463,7 +462,7 @@ int main(int argc, char* argv[]) {
} }
} else if (std::string(argv[i]) == "--threads") { } else if (std::string(argv[i]) == "--threads") {
if (i + 1 < argc) { if (i + 1 < argc) {
num_threads = std::stoi(argv[i+1]); omp_set_num_threads(std::stoi(argv[i+1]));
} else { } else {
print_usage(); print_usage();
return 1; return 1;