diff --git a/pystencils/gpu/kernelcreation.py b/pystencils/gpu/kernelcreation.py index e3ad451bb154c2cee279713fdfd2167a4268b3fb..b6f317b9e6202d3bbf3738f679b194ba0303e7b5 100644 --- a/pystencils/gpu/kernelcreation.py +++ b/pystencils/gpu/kernelcreation.py @@ -130,6 +130,9 @@ def created_indexed_cuda_kernel(assignments: NodeCollection, config: CreateKerne f"{spatial_coordinates}" spatial_coordinates = list(spatial_coordinates)[0] + assignments = assignments.all_assignments + assignments = add_types(assignments, config) + 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"