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

jaxDecomp pfft3d promotes to complex automatically

Browse source
This commit is contained in:
Wassim KABALAN 2024-10-25 10:15:43 +02:00
parent 5d4f438e92
commit 8e8e8964be
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
jaxpm/distributed.py
View file

@ -30,7 +30,7 @@ def autoshmap(
def fft3d(x):
return jaxdecomp.pfft3d(x.astype(jnp.complex64))
return jaxdecomp.pfft3d(x)
def ifft3d(x):