Add extra args

Signed-off-by: Kaiyu Xie <[email protected]>

Author: kaiyux

examples/disaggregated/slurm/benchmark/config.yaml

@@ -5,6 +5,7 @@ slurm:
   account: "<account>"
   job_time: "02:00:00"
   job_name: "<job_name>"
+  extra_args: ""
   numa_bind: true # Only enable for GB200 NVL72
 
 # Benchmark Mode

examples/disaggregated/slurm/benchmark/submit.py

@@ -138,6 +138,7 @@ def submit_job(config):
         f'--ntasks={total_tasks}',
         f'--ntasks-per-node={hw_config["gpus_per_node"]}',
         f'--segment={total_nodes}',
+        slurm_config['extra_args'],
         slurm_config['script_file'],
         # Hardware configuration
         str(hw_config['gpus_per_node']),