From 983dec1bc236ab43c0f33ff9bef8c960d86da091 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Mon, 16 Sep 2019 13:36:30 +0200
Subject: [PATCH] Fixup: add ASTs to AutoDiffOp

---
 src/pystencils_autodiff/_autodiff.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/pystencils_autodiff/_autodiff.py b/src/pystencils_autodiff/_autodiff.py
index 7735dcc..4edd071 100644
--- a/src/pystencils_autodiff/_autodiff.py
+++ b/src/pystencils_autodiff/_autodiff.py
@@ -62,6 +62,10 @@ Backward:
         self._time_constant_fields = time_constant_fields
         self._kwargs = kwargs
         self.op_name = op_name
+        self._forward_ast_cpu = None
+        self._backward_ast_cpu = None
+        self._forward_ast_gpu = None
+        self._backward_ast_gpu = None
         self._forward_kernel_cpu = None
         self._backward_kernel_cpu = None
         self._forward_kernel_gpu = None
-- 
GitLab