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

format

Browse source
This commit is contained in:
Wassim Kabalan 2025-06-12 14:51:24 +02:00
parent e1a8134b8e
commit 67a80e1041
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
jaxpm/pm.py
View file

@ -131,7 +131,7 @@ def linear_field(mesh_shape, box_size, pk, seed, sharding=None):
Generate initial conditions.
"""
# Initialize a random field with one slice on each gpu
field = normal_field(seed=seed , shape=mesh_shape, sharding=sharding)
field = normal_field(seed=seed, shape=mesh_shape, sharding=sharding)
field = fft3d(field)
kvec = fftk(field)
kmesh = sum((kk / box_size[i] * mesh_shape[i])**2