|
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