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 #593 from ufownl:bugfix/dc2bf16 

PiperOrigin-RevId: 767098675
This commit is contained in:
Copybara-Service 2025-06-04 05:21:54 -07:00
parent 6897313080 a82f8d5690
commit eff0213e88
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gemma/weights.cc
View File

@ -367,7 +367,7 @@ static void MapAll(const std::vector<TensorToRead>& tensors,
// Mode == kReadBF16:
template <typename T>
static void DecompressToBF16(const MatPtr& mat,
static void DecompressToBF16(MatPtr& mat,
const hwy::AlignedFreeUniquePtr<uint8_t[]>& buf) {
hwy::HWY_NAMESPACE::ScalableTag<BF16> dbf;
const size_t cols = mat.Cols();