From 6d43d6ee192f6f5335c9a35bb73eff672a3f604b Mon Sep 17 00:00:00 2001
From: Jan Wassenberg <janwas@google.com>
Date: Fri, 9 Jan 2026 02:55:28 -0800
Subject: [PATCH] Build fix for Arm SVE (invalid template arg on op)

PiperOrigin-RevId: 854110884
---
 gemma/flash_attention.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gemma/flash_attention.cc b/gemma/flash_attention.cc
index ba985fc5..e018ab80 100644
--- a/gemma/flash_attention.cc
+++ b/gemma/flash_attention.cc
@@ -452,7 +452,7 @@ static VF4 HWY_INLINE Reduce4(DF df, VF x_0, VF x_1, VF x_2, VF x_3,
   constexpr size_t kMaxLanes = hn::MaxLanes(df);
   HWY_LANES_CONSTEXPR size_t kLanes = hn::Lanes(df);
   HWY_ALIGN T x_transposed[4 * kMaxLanes];
-  hn::StoreInterleaved4<DF>(x_0, x_1, x_2, x_3, df, x_transposed);
+  hn::StoreInterleaved4(x_0, x_1, x_2, x_3, df, x_transposed);
   VF4 result = hn::Load(df4, x_transposed);
   for (int i = 1; i < kLanes; ++i) {
     result = reducer(result, hn::Load(df4, x_transposed + i * 4));