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

Use explicit ctor for hwy divisor

This commit is contained in:
Nanubala Gnana Sai 2024-11-05 22:43:07 +05:30
parent 719098fd3e
commit 28063743fc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gemma/gemma-inl.h
View File

@ -306,8 +306,8 @@ class GemmaAttention {
}
// Self-extension
const hwy::Divisor& div_grp_size{
static_cast<uint32_t>(layer_config_.grp_size)};
const hwy::Divisor div_grp_size(
static_cast<uint32_t>(layer_config_.grp_size));
// Apply positional encodings for K (and copy KV to cache if MHA).
pool_.Run(0, layer_config_.kv_heads * num_interleaved,
[&](uint64_t task, size_t /*thread*/) HWY_ATTR {