@@ -69,6 +69,8 @@ def generate_report(tp_size, eval_config, report_data, report_dir, env_config):
69
69
if model_args .get ('enable_expert_parallel' , False ):
70
70
parallel_mode += " + EP"
71
71
72
+ execution_model = f"{ 'Eager' if model_args .get ('enforce_eager' , False ) else 'ACLGraph' } "
73
+
72
74
report_content = template .render (
73
75
vllm_version = env_config .vllm_version ,
74
76
vllm_commit = env_config .vllm_commit ,
@@ -77,6 +79,7 @@ def generate_report(tp_size, eval_config, report_data, report_dir, env_config):
77
79
cann_version = env_config .cann_version ,
78
80
torch_version = env_config .torch_version ,
79
81
torch_npu_version = env_config .torch_npu_version ,
82
+ hardware = eval_config .get ("hardware" , "unknown" ),
80
83
model_name = eval_config ["model_name" ],
81
84
model_args = f"'{ ',' .join (f'{ k } ={ v } ' for k , v in model_args .items ())} '" ,
82
85
model_type = eval_config .get ("model" , "vllm" ),
@@ -87,7 +90,8 @@ def generate_report(tp_size, eval_config, report_data, report_dir, env_config):
87
90
batch_size = eval_config .get ("batch_size" , "auto" ),
88
91
num_fewshot = eval_config .get ("num_fewshot" , "N/A" ),
89
92
rows = report_data ["rows" ],
90
- parallel_mode = parallel_mode )
93
+ parallel_mode = parallel_mode ,
94
+ execution_model = execution_model )
91
95
92
96
report_output = os .path .join (
93
97
report_dir , f"{ os .path .basename (eval_config ['model_name' ])} .md" )
0 commit comments