From 7bd1171a0dd2dd7c33366f456d5cff48c238c82f Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Wed, 6 Sep 2023 18:28:48 +0200
Subject: [PATCH] Added fixed arg again
---
pystencils/gpu/kernelcreation.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/pystencils/gpu/kernelcreation.py b/pystencils/gpu/kernelcreation.py
index b6f317b9..7f420ca8 100644
--- a/pystencils/gpu/kernelcreation.py
+++ b/pystencils/gpu/kernelcreation.py
@@ -133,6 +133,11 @@ def created_indexed_cuda_kernel(assignments: NodeCollection, config: CreateKerne
assignments = assignments.all_assignments
assignments = add_types(assignments, config)
+ for index_field in index_fields:
+ index_field.field_type = FieldType.INDEXED
+ assert FieldType.is_indexed(index_field)
+ assert index_field.spatial_dimensions == 1, "Index fields have to be 1D"
+
def get_coordinate_symbol_assignment(name):
for ind_f in index_fields:
assert isinstance(ind_f.dtype, StructType), "Index fields have to have a struct data type"
--
GitLab