From 354877bbc7b04b4c2a92ec49a7e3218b18a3ffca Mon Sep 17 00:00:00 2001
From: Maddin <martin.geitner@fau.de>
Date: Mon, 16 Dec 2019 13:16:37 +0100
Subject: [PATCH] add index Field parameter to ast

---
 pystencils/kernelcreation.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/pystencils/kernelcreation.py b/pystencils/kernelcreation.py
index 43b239ef5..613050544 100644
--- a/pystencils/kernelcreation.py
+++ b/pystencils/kernelcreation.py
@@ -76,6 +76,9 @@ def create_kernel(assignments,
                [0., 4., 4., 4., 0.],
                [0., 0., 0., 0., 0.]])
     """
+    #save the original assignments
+    assign = assignments
+    
     # ----  Normalizing parameters
     split_groups = ()
     if isinstance(assignments, AssignmentCollection):
@@ -130,7 +133,8 @@ def create_kernel(assignments,
     if use_auto_for_assignments:
         for a in ast.atoms(SympyAssignment):
             a.use_auto = True
-    ast.assignments = assignments
+    ast.assignments = assign
+    ast.indexFields = False
 
     return ast
 
@@ -181,6 +185,9 @@ def create_indexed_kernel(assignments,
                [0. , 0. , 0. , 4.3, 0. ],
                [0. , 0. , 0. , 0. , 0. ]])
     """
+    assign = assignments
+    indF = index_fields
+    
     if isinstance(assignments, Assignment):
         assignments = [assignments]
     elif isinstance(assignments, AssignmentCollection):
@@ -213,7 +220,8 @@ def create_indexed_kernel(assignments,
     else:
         raise ValueError("Unknown target %s. Has to be either 'cpu' or 'gpu'" % (target,))
 
-    ast.assignments = assignments
+    ast.assignments = assign
+    ast.indexFields = indF
     return ast
 
 
-- 
GitLab