ikawrakow/ik_llama.cpp
1
0
Fork 0
mirror of https://github.com/ikawrakow/ik_llama.cpp.git synced 2026-02-07 15:00:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

bitnet: add 2 bpw quantization

Browse Source 
The scalar dot product already chieves 37 t/s for TG!
This commit is contained in:
Kawrakow
2024-06-17 18:41:30 +03:00
parent f9ba085ef7
commit 318899c8b7
10 changed files with 216 additions and 184 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/quantize-stats/quantize-stats.cpp
View File

@@ -341,6 +341,10 @@ int main(int argc, char ** argv) {
if (!layer_included(params, kv_tensor.first)) {
continue;
}
if (kv_tensor.second->ne[0] == 1 || kv_tensor.second->ne[1] == 1) {
// we never quantize those
continue;
}
if (params.verbose) {
printf("%s: type %s, size %" PRId64 "\n", kv_tensor.first.c_str(), ggml_type_name(kv_tensor.second->type), ggml_nelements(kv_tensor.second));
}
@@ -386,6 +390,10 @@ int main(int argc, char ** argv) {
if (!layer_included(params, kv_tensor.first)) {
continue;
}
if (kv_tensor.second->ne[0] == 1 || kv_tensor.second->ne[1] == 1) {
// we never quantize those
continue;
}
if (params.verbose) {
printf(" %s ...\n", kv_tensor.first.c_str());
}