Skip to content

GitLab

N
NIFTy
Commit 1c1a6b51 authored by Philipp Arras's avatar Philipp Arras
Browse files
Options

MPI fixups

parent 820ddd40
Pipeline #52936 passed with stages
        in 8 minutes and 10 seconds
        Hide whitespace changes
        Inline Side-by-side
        Showing with 4 additions and 4 deletions
        test/test_linearization.py
        View file @ 1c1a6b51
        ......@@ -25,7 +25,7 @@ pmp = pytest.mark.parametrize
        def _lin2grad(lin):
        return lin.jac(ift.full(lin.domain, 1.)).local_data
        return lin.jac(ift.full(lin.domain, 1.)).to_global_data()
        def jt(lin, check):
        ......@@ -36,7 +36,7 @@ def test_special_gradients():
        dom = ift.UnstructuredDomain((1,))
        f = ift.full(dom, 2.4)
        var = ift.Linearization.make_var(f)
        s = f.local_data
        s = f.to_global_data()
        jt(var.clip(0, 10), np.ones_like(s))
        jt(var.clip(-1, 0), np.zeros_like(s))
        ......@@ -62,8 +62,8 @@ def test_actual_gradients(f):
        eps = 1e-8
        var0 = ift.Linearization.make_var(fld)
        var1 = ift.Linearization.make_var(fld + eps)
        f0 = getattr(var0, f)().val.local_data
        f1 = getattr(var1, f)().val.local_data
        f0 = getattr(var0, f)().val.to_global_data()
        f1 = getattr(var1, f)().val.to_global_data()
        df0 = (f1 - f0)/eps
        df1 = _lin2grad(getattr(var0, f)())
        assert_allclose(df0, df1, rtol=100*eps)
        Markdown is supported
        0% or
        You are about to add 0 people to the discussion. Proceed with caution.
        Finish editing this message first!
        Please register or to comment