jaxDecomp pfft3d promotes to complex automatically

This commit is contained in:
Wassim KABALAN 2024-10-25 10:15:43 +02:00
parent 5d4f438e92
commit 8e8e8964be

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):