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

Moved Internal code around to simplify 

PiperOrigin-RevId: 681877329
This commit is contained in:
Ray Smith 2024-10-03 07:54:21 -07:00 committed by Copybara-Service
parent 12291e1ac0
commit 895ee4c6ce
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
gemma/weights.h
View File

@ -167,7 +167,7 @@ struct CompressedLayerPointers {
c_vit_layers; c_vit_layers;
}; };
template <class TConfig, typename = void> template <class TConfig>
struct CompressedWeights { struct CompressedWeights {
// Must be allocated via AllocateAligned and initialized with placement new. // Must be allocated via AllocateAligned and initialized with placement new.
void* operator new(size_t, void* addr) { return addr; } void* operator new(size_t, void* addr) { return addr; }
@ -207,7 +207,9 @@ struct CompressedWeights {
// Must be last so that the other arrays remain aligned. // Must be last so that the other arrays remain aligned.
CompressedLayerPointers<TConfig> c_layer_ptrs; CompressedLayerPointers<TConfig> c_layer_ptrs;
explicit CompressedWeights(hwy::ThreadPool& pool) : c_layer_ptrs(pool) {} explicit CompressedWeights(hwy::ThreadPool& pool)
: c_layer_ptrs(pool)
{}
// Called by weights.cc after ForEachTensor. // Called by weights.cc after ForEachTensor.
void Reshape(hwy::ThreadPool& pool) { void Reshape(hwy::ThreadPool& pool) {