From ddba65e55fd45e4c23bdd3eceef8734b6fe2243f Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 17 Dec 2019 15:10:30 +0100
Subject: [PATCH] Add additional assertions
---
src/pystencils_autodiff/_autodiff.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/pystencils_autodiff/_autodiff.py b/src/pystencils_autodiff/_autodiff.py
index 5e11401..9c3d0ac 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):
--
GitLab