From 5d9f76d64185b75f0beb1e81c06ae75ef3ca8f03 Mon Sep 17 00:00:00 2001 From: Nils Kohl <nils.kohl@fau.de> Date: Mon, 3 Jun 2024 19:20:06 +0200 Subject: [PATCH] Vectorizer did not seem to properly handle some loops that are smaller than the vector width. --- pystencils/cpu/vectorization.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pystencils/cpu/vectorization.py b/pystencils/cpu/vectorization.py index 3b724124a..aabae71aa 100644 --- a/pystencils/cpu/vectorization.py +++ b/pystencils/cpu/vectorization.py @@ -172,6 +172,8 @@ def vectorize_inner_loops_and_adapt_load_stores(ast_node, assume_aligned, nontem assert len(loop_nodes) in (0, 1, 2) # 2 for main and tail loop, 1 if loop range divisible by vector width if len(loop_nodes) == 0: continue + if len(loop_nodes) == 1 and loop_nodes[0].stop - loop_nodes[0].start < vector_width: + continue for loop in loop_nodes: loop.parent = loop_node.parent loop_node = loop_nodes[0] -- GitLab