Skip to content

Commit 7be099f

Browse files
authored
llama-bench: correct argument parsing error message (ggml-org#9524)
1 parent 8b836ae commit 7be099f

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

examples/llama-bench/llama-bench.cpp

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -439,6 +439,9 @@ static cmd_params parse_cmd_params(int argc, char ** argv) {
439439
}
440440
types.push_back(gt);
441441
}
442+
if (invalid_param) {
443+
break;
444+
}
442445
params.type_k.insert(params.type_k.end(), types.begin(), types.end());
443446
} else if (arg == "-ctv" || arg == "--cache-type-v") {
444447
if (++i >= argc) {
@@ -455,6 +458,9 @@ static cmd_params parse_cmd_params(int argc, char ** argv) {
455458
}
456459
types.push_back(gt);
457460
}
461+
if (invalid_param) {
462+
break;
463+
}
458464
params.type_v.insert(params.type_v.end(), types.begin(), types.end());
459465
} else if (arg == "-t" || arg == "--threads") {
460466
if (++i >= argc) {
@@ -520,6 +526,9 @@ static cmd_params parse_cmd_params(int argc, char ** argv) {
520526
}
521527
modes.push_back(mode);
522528
}
529+
if (invalid_param) {
530+
break;
531+
}
523532
params.split_mode.insert(params.split_mode.end(), modes.begin(), modes.end());
524533
} else if (arg == "-mg" || arg == "--main-gpu") {
525534
if (++i >= argc) {

0 commit comments

Comments
 (0)