From bcc094936fc68bf4d274729fafd4926dab02d987 Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Wed, 24 Jul 2024 18:38:40 +0200
Subject: [PATCH] test
---
src/pystencils/gpu/kernelcreation.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/pystencils/gpu/kernelcreation.py b/src/pystencils/gpu/kernelcreation.py
index c2e6143b..c7670c2d 100644
--- a/src/pystencils/gpu/kernelcreation.py
+++ b/src/pystencils/gpu/kernelcreation.py
@@ -72,9 +72,11 @@ def create_cuda_kernel(assignments: NodeCollection, config: CreateKernelConfig):
if len(indexed_elements) > 0:
common_indexed_element = get_common_indexed_element(indexed_elements)
+ index = list(common_indexed_element.indices[0].atoms(TypedSymbol))
+ assert len(index) == 1, "index expressions must only contain one symbol representing the index"
indexing = indexing_creator(iteration_space=(slice(0, common_indexed_element.shape[0], 1), *iteration_space),
data_layout=common_field.layout)
- extended_ctrs = [common_indexed_element.indices[0], *loop_counter_symbols]
+ extended_ctrs = [index[0], *loop_counter_symbols]
loop_counter_assignments = indexing.get_loop_ctr_assignments(extended_ctrs)
else:
indexing = indexing_creator(iteration_space=iteration_space, data_layout=common_field.layout)
--
GitLab