mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-04-07 20:30:54 +00:00
jaxDecomp pfft3d promotes to complex automatically
This commit is contained in:
parent
5d4f438e92
commit
8e8e8964be
1 changed files with 1 additions and 1 deletions
|
@ -30,7 +30,7 @@ def autoshmap(
|
|||
|
||||
|
||||
def fft3d(x):
|
||||
return jaxdecomp.pfft3d(x.astype(jnp.complex64))
|
||||
return jaxdecomp.pfft3d(x)
|
||||
|
||||
|
||||
def ifft3d(x):
|
||||
|
|
Loading…
Add table
Reference in a new issue