From 20f2ee40ec71a947ed481afbb0a7d7c02295fa0c Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Mon, 2 Dec 2019 14:41:59 +0100
Subject: [PATCH] Fixup: allow time_constant_fields == None
---
src/pystencils_autodiff/_autodiff.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/pystencils_autodiff/_autodiff.py b/src/pystencils_autodiff/_autodiff.py
index f072e5e..4fe67dd 100644
--- a/src/pystencils_autodiff/_autodiff.py
+++ b/src/pystencils_autodiff/_autodiff.py
@@ -85,7 +85,7 @@ def _create_backward_assignments_tf_mad(self, diff_fields_prefix):
w in zip(ra.offsets, forward_assignment.lhs.offsets))
diff_read_field_sum += sp.diff(forward_assignment.rhs, ra) * \
diff_write_field[inverted_offset](*diff_write_index)
- if forward_read_field in self._time_constant_fields and self.time_constant_fields is not None:
+ if self.time_constant_fields is not None and forward_read_field in self._time_constant_fields:
# Accumulate in case of time_constant_fields
assignment = ps.Assignment(
diff_read_field.center(), diff_read_field.center() + diff_read_field_sum)
@@ -368,7 +368,7 @@ Backward:
rhs = rhs[0, 0]
# if field is constant over we time we can accumulate in assignment
- if read_access.field in self._time_constant_fields and self.time_constant_fields is not None:
+ if self.time_constant_fields is not None and read_access.field in self._time_constant_fields:
backward_assignments.append(ps.Assignment(lhs, lhs + rhs))
else:
backward_assignments.append(ps.Assignment(lhs, rhs))
--
GitLab