Add batch warmup to sweep-bench

This commit is contained in:
Iwan Kawrakow
2025-05-04 11:21:19 +03:00
parent ce2b0292e1
commit a3975acd4c
3 changed files with 22 additions and 1 deletions

View File

@@ -1457,6 +1457,10 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
params.warmup = false;
return true;
}
if (arg == "--warmup-batch" || arg == "-wb") {
params.batch_warmup = true;
return true;
}
if (arg == "--output-format") {
CHECK_ARG
std::string value(argv[i]);