diff --git a/pystencils/gpu/kernelcreation.py b/pystencils/gpu/kernelcreation.py index b6f317b9e6202d3bbf3738f679b194ba0303e7b5..7f420ca85570828b07065c4d305549d659261a2c 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"