Try to generate more swaps

src/pystencils_autodiff/graph_datahandling.py

@@ -93,9 +93,18 @@ class Communication(DataTransfer):
 
 class KernelCall:
     def __init__(self, kernel: pystencils.kernel_wrapper.KernelWrapper, kwargs, tmp_field_swaps=[]):
+        tmp = None
+        src = None
+        for f in kernel.ast.fields_accessed:
+            if 'pdfTmp' in f.name:
+                tmp = f
+            if 'pdfSrc' in f.name:
+                src = f
         self.kernel = kernel
         self.kwargs = kwargs
         self.tmp_field_swaps = tmp_field_swaps
+        if tmp and src:
+            self.tmp_field_swaps.append((src, tmp))
 
     def __str__(self):
         return "Call " + str(self.kernel.ast.function_name)