From 1a3ad99d15eb97c1ff54c5c629b906897ac95f97 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 18 Feb 2025 13:41:12 -0600
Subject: [PATCH] VectorizabilityChecker: simplify subscripts

---
 loopy/expression.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/loopy/expression.py b/loopy/expression.py
index 64172b88..9d48a8f0 100644
--- a/loopy/expression.py
+++ b/loopy/expression.py
@@ -122,6 +122,8 @@ class VectorizabilityChecker(Mapper[bool, []]):
 
         index = expr.index_tuple
 
+        index = tuple(simplify_using_aff(self.kernel, idx_i) for idx_i in index)
+
         from pymbolic.primitives import Variable
 
         from loopy.kernel.array import VectorArrayDimTag
-- 
GitLab