@@ -205,15 +205,15 @@ process_model() {
205205 local -A results=()
206206
207207 # Collect results for all data types and modes
208- for dtype in float32 bfloat16 float16 amp_bf16 amp_fp16 ; do
209- for mode in training inference ; do
208+ for mode in training inference ; do
209+ for dtype in float32 bfloat16 float16 amp_bf16 amp_fp16 ; do
210210 local key=" ${mode} _${dtype} "
211211 results[$key ]=$( get_model_result " $results_dir " " $suite " " $model " " $dtype " " $mode " )
212212 done
213213 done
214214
215215 local row
216- row=$( generate_html_row " $suite " " $model " " ${results[@]} " )
216+ row=$( generate_html_row " $suite " " $model " )
217217
218218 if [[ " $row " =~ ${RED} | ${GREEN} | ${YELLOW} ]]; then
219219 echo " $row " | tee -a accuracy.details.html >> accuracy.regression.html
@@ -258,15 +258,20 @@ determine_color() {
258258
259259generate_html_row () {
260260 local suite=" $1 " model=" $2 "
261- shift 2
262- local results=(" $@ " )
263-
264261 cat << EOF
265262<tr>
266263 <td>$suite </td>
267264 <td>$model </td>
268- <td>${results[0]} </td><td>${results[1]} </td><td>${results[2]} </td><td>${results[3]} </td><td>${results[4]} </td>
269- <td>${results[5]} </td><td>${results[6]} </td><td>${results[7]} </td><td>${results[8]} </td><td>${results[9]} </td>
265+ <td>${results["training_float32"]} </td>
266+ <td>${results["training_bfloat16"]} </td>
267+ <td>${results["training_float16"]} </td>
268+ <td>${results["training_amp_bf16"]} </td>
269+ <td>${results["training_amp_fp16"]} </td>
270+ <td>${results["inference_float32"]} </td>
271+ <td>${results["inference_bfloat16"]} </td>
272+ <td>${results["inference_float16"]} </td>
273+ <td>${results["inference_amp_bf16"]} </td>
274+ <td>${results["inference_amp_fp16"]} </td>
270275</tr>
271276EOF
272277}
0 commit comments