Correctly handle --threads
param
This commit is contained in:
parent
4c687844d3
commit
6d1d360ec0
3
main.cpp
3
main.cpp
@ -421,7 +421,6 @@ int main(int argc, char* argv[]) {
|
||||
Mode mode = Mode::SEQ;
|
||||
bool use_gpu = false;
|
||||
int num_generations = 0;
|
||||
int num_threads = 1;
|
||||
bool measure = false;
|
||||
|
||||
if (argc < 8) {
|
||||
@ -463,7 +462,7 @@ int main(int argc, char* argv[]) {
|
||||
}
|
||||
} else if (std::string(argv[i]) == "--threads") {
|
||||
if (i + 1 < argc) {
|
||||
num_threads = std::stoi(argv[i+1]);
|
||||
omp_set_num_threads(std::stoi(argv[i+1]));
|
||||
} else {
|
||||
print_usage();
|
||||
return 1;
|
||||
|
Loading…
x
Reference in New Issue
Block a user