Skip to content

Commit f1c2a4c

Browse files
committed
Fix printing l2 norm when calc_mode = 1
1 parent 90cb1be commit f1c2a4c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/imatrix/imatrix.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1302,7 +1302,7 @@ static bool show_statistics(const common_params & params) {
13021302
LOG_INF("%5s\t%-20s\t%11.2f\t%10.4f\t%10.4f\t%8.2f\t%8.2f\t%7d\t%12.4f\t%7.2f%%\t%6.2f%%\t%10.4f\n",
13031303
layer.c_str(),
13041304
name.c_str(),
1305-
tstat.sum_values,
1305+
tensor_calc_mode == 1 ? tstat.l2_norm : tstat.sum_values,
13061306
tstat.min_values,
13071307
tstat.max_values,
13081308
tstat.mean_values,

0 commit comments

Comments
 (0)