|
13 | 13 |
|
14 | 14 | import cupy as cp |
15 | 15 |
|
16 | | -fft = get_xp(cp)(_fft.fft), |
17 | | -ifft = get_xp(cp)(_fft.ifft), |
18 | | -fftn = get_xp(cp)(_fft.fftn), |
19 | | -ifftn = get_xp(cp)(_fft.ifftn), |
20 | | -rfft = get_xp(cp)(_fft.rfft), |
21 | | -irfft = get_xp(cp)(_fft.irfft), |
22 | | -rfftn = get_xp(cp)(_fft.rfftn), |
23 | | -irfftn = get_xp(cp)(_fft.irfftn), |
24 | | -hfft = get_xp(cp)(_fft.hfft), |
25 | | -ihfft = get_xp(cp)(_fft.ihfft), |
26 | | -fftfreq = get_xp(cp)(_fft.fftfreq), |
27 | | -rfftfreq = get_xp(cp)(_fft.rfftfreq), |
28 | | -fftshift = get_xp(cp)(_fft.fftshift), |
29 | | -ifftshift = get_xp(cp)(_fft.ifftshift), |
| 16 | +fft = get_xp(cp)(_fft.fft) |
| 17 | +ifft = get_xp(cp)(_fft.ifft) |
| 18 | +fftn = get_xp(cp)(_fft.fftn) |
| 19 | +ifftn = get_xp(cp)(_fft.ifftn) |
| 20 | +rfft = get_xp(cp)(_fft.rfft) |
| 21 | +irfft = get_xp(cp)(_fft.irfft) |
| 22 | +rfftn = get_xp(cp)(_fft.rfftn) |
| 23 | +irfftn = get_xp(cp)(_fft.irfftn) |
| 24 | +hfft = get_xp(cp)(_fft.hfft) |
| 25 | +ihfft = get_xp(cp)(_fft.ihfft) |
| 26 | +fftfreq = get_xp(cp)(_fft.fftfreq) |
| 27 | +rfftfreq = get_xp(cp)(_fft.rfftfreq) |
| 28 | +fftshift = get_xp(cp)(_fft.fftshift) |
| 29 | +ifftshift = get_xp(cp)(_fft.ifftshift) |
30 | 30 |
|
31 | 31 | __all__ = fft_all + _fft.__all__ |
32 | 32 |
|
|
0 commit comments