@@ -230,21 +230,15 @@ def create_summary_table(test_result: Dict[str, Any]) -> pd.DataFrame:
230230 accelerators = machine .get ("accelerators" , [])
231231 if accelerators :
232232 acc = accelerators [0 ]
233- summary_data .append (
234- {"指标" : "GPU型号" , "数值" : acc .get ("model" , "Unknown" )}
235- )
236- summary_data .append (
237- {"指标" : "GPU数量" , "数值" : acc .get ("count" , "Unknown" )}
238- )
233+ summary_data .append ({"指标" : "GPU型号" , "数值" : acc .get ("model" , "Unknown" )})
234+ summary_data .append ({"指标" : "GPU数量" , "数值" : acc .get ("count" , "Unknown" )})
239235 summary_data .append (
240236 {
241237 "指标" : "显存/卡" ,
242238 "数值" : f"{ acc .get ('memory_gb_per_card' , 'Unknown' )} GB" ,
243239 }
244240 )
245- summary_data .append (
246- {"指标" : "CUDA版本" , "数值" : acc .get ("cuda" , "Unknown" )}
247- )
241+ summary_data .append ({"指标" : "CUDA版本" , "数值" : acc .get ("cuda" , "Unknown" )})
248242
249243 # Test config summary
250244 config = test_result .get ("config" , {})
@@ -387,21 +381,15 @@ def create_summary_table_infer(test_result: dict) -> pd.DataFrame:
387381 {"指标" : "模型" , "数值" : cfg .get ("model" , "" )},
388382 {
389383 "指标" : "batch" ,
390- "数值" : (cfg .get ("infer_args" , {}) or {}).get (
391- "static_batch_size" , "unknown"
392- ),
384+ "数值" : (cfg .get ("infer_args" , {}) or {}).get ("static_batch_size" , "unknown" ),
393385 },
394386 {
395387 "指标" : "prompt_tok" ,
396- "数值" : (cfg .get ("infer_args" , {}) or {}).get (
397- "prompt_token_num" , "unknown"
398- ),
388+ "数值" : (cfg .get ("infer_args" , {}) or {}).get ("prompt_token_num" , "unknown" ),
399389 },
400390 {
401391 "指标" : "output_tok" ,
402- "数值" : (cfg .get ("infer_args" , {}) or {}).get (
403- "output_token_num" , "unknown"
404- ),
392+ "数值" : (cfg .get ("infer_args" , {}) or {}).get ("output_token_num" , "unknown" ),
405393 },
406394 {"指标" : "warmup" , "数值" : cfg .get ("warmup_iterations" , "unknown" )},
407395 {"指标" : "measured" , "数值" : cfg .get ("measured_iterations" , "unknown" )},
@@ -423,9 +411,7 @@ def create_summary_table_ops(test_result: dict) -> pd.DataFrame:
423411
424412 rows .append ({"指标" : "testcase" , "数值" : test_result .get ("testcase" , "" )})
425413 # Try to get operator name from config
426- rows .append (
427- {"指标" : "算子" , "数值" : cfg .get ("operator" , cfg .get ("op_name" , "Unknown" ))}
428- )
414+ rows .append ({"指标" : "算子" , "数值" : cfg .get ("operator" , cfg .get ("op_name" , "Unknown" ))})
429415
430416 # Environment info
431417 env = test_result .get ("environment" , {})
@@ -441,9 +427,7 @@ def create_summary_table_ops(test_result: dict) -> pd.DataFrame:
441427 # Scalar metrics summary
442428 scalars = [m for m in test_result .get ("metrics" , []) if m .get ("type" ) == "scalar" ]
443429 for m in scalars :
444- rows .append (
445- {"指标" : m .get ("name" ), "数值" : f"{ m .get ('value' )} { m .get ('unit' ,'' )} " }
446- )
430+ rows .append ({"指标" : m .get ("name" ), "数值" : f"{ m .get ('value' )} { m .get ('unit' ,'' )} " })
447431
448432 # Common config fields fallback
449433 for k in [
0 commit comments