From 768b068f39e361a223baa7edaafec17a6b1c9cbd Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Wed, 2 Sep 2026 14:18:54 -0500 Subject: [PATCH] Add a missing assumption in the tc-dep-data generation kernel --- sumpy/e2e.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/sumpy/e2e.py b/sumpy/e2e.py index 0d0bdc29..04aa836c 100644 --- a/sumpy/e2e.py +++ b/sumpy/e2e.py @@ -615,7 +615,10 @@ def get_kernel(self, result_dtype): *gather_loopy_arguments([self.src_expansion, self.tgt_expansion]) ], name=self.name, - assumptions="ntranslation_classes>=1", + assumptions="ntranslation_classes>=1 and " + "translation_classes_level_start>=0 and " + "translation_classes_level_start + ntranslation_classes " + "<= ntranslation_vectors", fixed_parameters={ "dim": self.dim, "m2l_translation_classes_dependent_ndata": (