From ae7901c3f441d70b54bb78d11d040c9a9c81ab94 Mon Sep 17 00:00:00 2001
From: Paul Chang <paulchang@google.com>
Date: Thu, 29 Feb 2024 13:07:48 -0800
Subject: [PATCH] Minor style fix

Remove some obsolete TODOs.

PiperOrigin-RevId: 611571224
---
 gemma.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/gemma.cc b/gemma.cc
index 4775f89..9f1e4a0 100644
--- a/gemma.cc
+++ b/gemma.cc
@@ -721,10 +721,10 @@ HWY_EXPORT(GetCompressedWeightsT);
 HWY_EXPORT(Generate2B);
 HWY_EXPORT(Generate7B);
 
-KVCache CreateKVCache(size_t size_cache_pos, size_t kSeqLen) {
+KVCache CreateKVCache(size_t size_cache_pos, size_t seq_len) {
   KVCache kv_cache = {};
-  kv_cache.key_cache = hwy::AllocateAligned<float>(kSeqLen * size_cache_pos);
-  kv_cache.value_cache = hwy::AllocateAligned<float>(kSeqLen * size_cache_pos);
+  kv_cache.key_cache = hwy::AllocateAligned<float>(seq_len * size_cache_pos);
+  kv_cache.value_cache = hwy::AllocateAligned<float>(seq_len * size_cache_pos);
   return kv_cache;
 }