sweep_bench: set number of repetions

2026-02-02 04:29:53 +00:00 · 2026-01-21 08:33:42 +00:00
parent 987651e54c
commit a2fb4cefda
3 changed files with 77 additions and 54 deletions
--- a/common/common.cpp
+++ b/common/common.cpp
@@ -786,6 +786,11 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
        params.max_extra_alloc_MiB = std::stoi(argv[i]);
        return true;
    }
+    if (arg == "-nrep" || arg == "--n-repetitions") {
+        CHECK_ARG
+        params.nrep = std::stoi(argv[i]);
+        return true;
+    }
    if (arg == "--samplers") {
        CHECK_ARG
        const auto sampler_names = string_split(argv[i], ";");