Bug fix!!!

pystencils/datahandling/graph_datahandling.py

@@ -7,13 +7,13 @@
 """
 
 """
-
 from enum import Enum
 
 import numpy as np
 
 import pystencils.datahandling
 import pystencils.kernel_wrapper
+import pystencils.timeloop
 from pystencils.field import FieldType
 
 
@@ -181,7 +181,7 @@ class GraphDataHandling(pystencils.datahandling.SerialDataHandling):
 
     def synchronization_function(self, names, stencil=None, target=None, **_):
         for name in names:
-            gpu = target == 'cpu'
+            gpu = target == 'gpu'
             self.call_queue.append(Communication(self._fields[name], stencil, gpu))
         super().synchronization_function(names, stencil=None, target=None, **_)
 

pystencils/kernelcreation.py

@@ -76,9 +76,9 @@ def create_kernel(assignments,
                [0., 4., 4., 4., 0.],
                [0., 0., 0., 0., 0.]])
     """
-    #save the original assignments
+    # save the original assignments
     assign = assignments
-    
+
     # ----  Normalizing parameters
     split_groups = ()
     if isinstance(assignments, AssignmentCollection):
@@ -187,7 +187,7 @@ def create_indexed_kernel(assignments,
     """
     assign = assignments
     indF = index_fields
-    
+
     if isinstance(assignments, Assignment):
         assignments = [assignments]
     elif isinstance(assignments, AssignmentCollection):