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

Merge pull request #422 from ufownl:bugfix/compress_weights_ce 

PiperOrigin-RevId: 685635493
This commit is contained in:
Copybara-Service 2024-10-14 02:46:33 -07:00
parent 5d0167904d ed88115e6a
commit 2892e232e2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
compression/compress_weights.cc
View File

@ -161,8 +161,8 @@ void CompressWeights(const Path& weights_path,
printf("Compressing weights from %s to %s\n", weights_path.path.c_str(), printf("Compressing weights from %s to %s\n", weights_path.path.c_str(),
compressed_weights_path.path.c_str()); compressed_weights_path.path.c_str());
using CConfig = Configs::c; using CConfig = typename Configs::c;
using UCConfig = Configs::uc; using UCConfig = typename Configs::uc;
// Allocate compressed weights. // Allocate compressed weights.
using CWeights = CompressedWeights<CConfig>; using CWeights = CompressedWeights<CConfig>;
ByteStorageT c_weights_u8 = AllocateCompressedWeights<CConfig>()(pool); ByteStorageT c_weights_u8 = AllocateCompressedWeights<CConfig>()(pool);