[None][feat] Add export data to build and run script for AD (#10299)

Signed-off-by: Tal Cherckez <[email protected]>

Author: tcherckez-nvidia

examples/auto_deploy/build_and_run_ad.py

@@ -277,7 +277,11 @@ def main(config: Optional[ExperimentConfig] = None):
         config.prompt.queries,
         sampling_params=SamplingParams(**config.prompt.sp_kwargs),
     )
-    results = {"prompts_and_outputs": print_outputs(outs)}
+    results = {
+        "prompts_and_outputs": print_outputs(outs),
+    }
+    # Add config values so they get logged to JET extra
+    results.update(config.model_dump(mode="json"))
 
     # run a benchmark for the model with batch_size == config.benchmark_bs
     if config.benchmark.enabled and config.args.runtime != "trtllm":

tensorrt_llm/_torch/auto_deploy/utils/benchmark.py

@@ -122,4 +122,4 @@ def store_benchmark_results(results: dict, results_path: str):
     results_path = pathlib.Path(results_path)
     results_path.parent.mkdir(parents=True, exist_ok=True)
     with results_path.open("w") as results_file:
-        json.dump(results, results_file)
+        json.dump(results, results_file, indent=2)