diff --git a/src/pystencils_autodiff/_autodiff.py b/src/pystencils_autodiff/_autodiff.py index 7735dccbb357eadd23a5882d832f832b38731838..4edd071b606b6c733046306b85e048b992e37696 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