guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-14 18:01:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update tolerance :)

Browse source
This commit is contained in:
Wassim Kabalan 2025-06-08 11:35:57 +02:00
parent 995cc4c78c
commit f0b849cf5f
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/test_distributed_pm.py
View file

@ -22,7 +22,7 @@ from jaxpm.distributed import fft3d, ifft3d
from jaxpm.painting import cic_paint, cic_paint_dx # noqa : E402 from jaxpm.painting import cic_paint, cic_paint_dx # noqa : E402
from jaxpm.pm import lpt, make_diffrax_ode, pm_forces # noqa : E402 from jaxpm.pm import lpt, make_diffrax_ode, pm_forces # noqa : E402
_TOLERANCE = 1e-1 # 🙃🙃 _TOLERANCE = 1e-8 # 🎉🎉🎉
pdims = [(1, 8), (8, 1), (4, 2), (2, 4)] pdims = [(1, 8), (8, 1), (4, 2), (2, 4)]