From 013e6546902cc2e2b1e508d03c52f394a891df37 Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Mon, 5 May 2025 12:24:10 +0200
Subject: [PATCH] Broadcast reduction init val as neutral element

---
 src/pystencils/backend/transformations/loop_vectorizer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/pystencils/backend/transformations/loop_vectorizer.py b/src/pystencils/backend/transformations/loop_vectorizer.py
index 1c48b4b33..b029b95a7 100644
--- a/src/pystencils/backend/transformations/loop_vectorizer.py
+++ b/src/pystencils/backend/transformations/loop_vectorizer.py
@@ -163,7 +163,8 @@ class LoopVectorizer:
                                 PsDeclaration(
                                     PsSymbolExpr(vector_symb),
                                     PsVecBroadcast(
-                                        self._lanes, PsSymbolExpr(local_symbol)
+                                        self._lanes,
+                                        reduction_info.init_val.clone(),
                                     ),
                                 )
                             )
-- 
GitLab