Commit bfd678e9 authored by Lukas Platz's avatar Lukas Platz

change PoissonianEnergy to detect whether there are INFs but no NaNs

BROKEN FOR MPI if there are NaNs in the input
parent fb34b7ce
Pipeline #47552 passed with stages
in 8 minutes and 18 seconds
......@@ -191,9 +191,21 @@ class PoissonianEnergy(EnergyOperator):
def apply(self, x):
res = x.sum() - x.log().vdot(self._d)
if isinstance(x, Linearization):
inp_vals = x.val.local_data
inp_vals = x.local_data
fix_inf = False
if np.any(inp_vals==np.inf):
if not np.any(np.isnan(inp_vals)):
fix_inf=True #Note: This will break for MPI if there are NaNs in some threads but not others
if not isinstance(x, Linearization):
if fix_inf:
res = np.inf
return Field.scalar(res)
res = x.sum() - x.log().vdot(self._d)
if fix_inf:
res = Linearization(Field.scalar(np.inf), res.jac)
if not x.want_metric:
return res
metric = SandwichOperator.make(x.jac, makeOp(1./x.val))
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment