Commit eb3ecf2f authored by Reimar Leike's avatar Reimar Leike
Browse files

Reintroduce dtype inference

parent b5d17329
Pipeline #75960 passed with stages
in 13 minutes and 44 seconds
......@@ -301,5 +301,7 @@ def check_jacobian_consistency(op, loc, tol=1e-8, ntries=100, perf_check=True,
raise ValueError("gradient and value seem inconsistent")
loc = locnext
ddtype = loc.values()[0].dtype if isinstance(loc, MultiField) else loc.dtype
tdtype = dirder.values()[0].dtype if isinstance(dirder, MultiField) else dirder.dtype
consistency_check(linmid.jac, domain_dtype=ddtype, target_dtype=tdtype,
only_r_linear=only_r_differentiable)
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