diff --git a/src/pystencils_autodiff/_autodiff.py b/src/pystencils_autodiff/_autodiff.py index 5e11401c2c49d19a3b713b33f717b32724cd8f7d..9c3d0ac461b288e24fdeda1983b1e5839abe1556 100644 --- a/src/pystencils_autodiff/_autodiff.py +++ b/src/pystencils_autodiff/_autodiff.py @@ -472,6 +472,7 @@ Backward: @property def backward_ast_cpu(self): + assert self._backward_assignments, 'No backward assignments!' if not self._backward_ast_cpu: if (self._boundary_handling == AutoDiffBoundaryHandling.ZEROS or self._boundary_handling == AutoDiffBoundaryHandling.VALID): @@ -488,6 +489,7 @@ Backward: @property def backward_ast_gpu(self): + assert self._backward_assignments, 'No backward assignments!' if not self._backward_ast_gpu: if (self._boundary_handling == AutoDiffBoundaryHandling.ZEROS or self._boundary_handling == AutoDiffBoundaryHandling.VALID):