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;
|
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;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user