guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-05-15 04:21:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

format

Browse source
This commit is contained in:
Wassim KABALAN 2024-07-18 13:15:39 +02:00
parent 82be56836a
commit 4f508b7cb6
3 changed files with 54 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

1
jaxpm/distributed.py
View file

@ -131,7 +131,6 @@ def get_local_shape(mesh_shape):
return [
mesh_shape[0] // pdims[0], mesh_shape[1] // pdims[1], mesh_shape[2]
]
def normal_field(mesh_shape, seed=None):

1
jaxpm/pm.py
View file

@ -38,6 +38,7 @@ def pm_forces(positions, mesh_shape=None, delta=None, r_split=0, halo_size=0):
return forces
def lpt(cosmo, initial_conditions, a, halo_size=0):
"""
Computes first order LPT displacement