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