From 895ee4c6ceb7685806804dc811a55a5d44ab4615 Mon Sep 17 00:00:00 2001 From: Ray Smith Date: Thu, 3 Oct 2024 07:54:21 -0700 Subject: [PATCH] Moved Internal code around to simplify PiperOrigin-RevId: 681877329 --- gemma/weights.h | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/gemma/weights.h b/gemma/weights.h index 73c1e22..b24b0c6 100644 --- a/gemma/weights.h +++ b/gemma/weights.h @@ -167,7 +167,7 @@ struct CompressedLayerPointers { c_vit_layers; }; -template +template struct CompressedWeights { // Must be allocated via AllocateAligned and initialized with placement new. 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. CompressedLayerPointers 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. void Reshape(hwy::ThreadPool& pool) {