feat: add CLI flag to control communication and computation logging

2025-08-02 12:55:12 +00:00
parent e00cef66ac
commit 96191d3b6f
6 changed files with 36 additions and 6 deletions
@@ -2066,6 +2066,13 @@ gpt_params_context gpt_params_parser_init(gpt_params & params, llama_example ex,
            params.dump_folder = value;
        }
    ));
+    add_opt(llama_arg(
+        {"--enable-comm-compute-log"},
+        "enable communication and computation logging for gantt chart analysis",
+        [](gpt_params & params) {
+            params.enable_comm_compute_log = true;
+        }
+    ));
    add_opt(llama_arg(
        {"--positive-file"}, "FNAME",
        format("positive prompts file, one prompt per line (default: '%s')", params.cvector_positive_file.c_str()),