happyz
/
llama.cpp
mirror of https://github.com/ggerganov/llama.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update stats output sort based on imatrix type

This commit is contained in:
Ed Addario 2025-08-15 21:12:56 +01:00
parent 2756617c3f
commit 42bfe3b2a3
No known key found for this signature in database
GPG Key ID: E7875815A3230993
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tools/imatrix/imatrix.cpp
View File

@ -1284,15 +1284,18 @@ static bool show_statistics(const common_params & params) {
}
struct tensor_comparer {
bool legacy_mode;
explicit tensor_comparer(const bool legacy) : legacy_mode(legacy) {}
bool operator()(const tensor_statistics & a, const tensor_statistics & b) const {
std::string layer, name_a, name_b;
;
process_tensor_name(a.tensor, layer, name_a);
process_tensor_name(b.tensor, layer, name_b);
return name_a < name_b || (name_a == name_b && a.sum_values > b.sum_values);
return legacy_mode ? name_a < name_b || (name_a == name_b && a.sum_values > b.sum_values) :
name_a < name_b || (name_a == name_b && a.cossim > b.cossim);
}
};
std::sort(ts.begin(), ts.end(), tensor_comparer());
std::sort(ts.begin(), ts.end(), tensor_comparer(legacy_mode));
struct layer_stats {
float lyr_sum = 0.0f;