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