From 4caa1ec617546d74c18d7ce89cd747c90946e6d0 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 19 Oct 2017 01:02:17 -0500
Subject: [PATCH] Use kernel assumptions in static_max_of_pw_aff for grid size
finding (closes #46 on gitlab)
---
loopy/kernel/__init__.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/loopy/kernel/__init__.py b/loopy/kernel/__init__.py
index 642c82c4b..9e423b99f 100644
--- a/loopy/kernel/__init__.py
+++ b/loopy/kernel/__init__.py
@@ -958,7 +958,8 @@ class LoopKernel(ImmutableRecordWithoutPickling):
try:
# insist block size is constant
size = static_max_of_pw_aff(size,
- constants_only=isinstance(tag, LocalIndexTag))
+ constants_only=isinstance(tag, LocalIndexTag),
+ context=self.assumptions)
except ValueError:
pass
--
GitLab