diff --git a/src/pystencils_autodiff/_autodiff.py b/src/pystencils_autodiff/_autodiff.py index 7fb78f8fb16dc1a176d0ea066d683bc4d6f28f2c..a003cd5eaed998eca92e6b58da696df5807a12cd 100644 --- a/src/pystencils_autodiff/_autodiff.py +++ b/src/pystencils_autodiff/_autodiff.py @@ -134,7 +134,7 @@ def _create_backward_assignments_tf_mad(self, diff_fields_prefix): ] += sp.diff(forward_assignment.rhs, ra) * diff_write_field[inverted_offset] for index in range(diff_read_field.index_shape[0]): - if True: + 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_vector[index],