Coverage for gpaw/gpu/cpupyx/fft.py: 100%

1import scipy.fft as fft 

2import gpaw.gpu.cpupy as cp 

3 

4 

5def ifftn(array_Q, 

6 shape, 

7 norm=None, 

8 overwrite_x=False): 

9 return cp.ndarray(fft.ifftn(array_Q._data, 

10 shape, 

11 norm=norm, 

12 overwrite_x=overwrite_x)) 

13 

14 

15def fftn(array_Q, 

16 shape=None, 

17 norm=None, 

18 overwrite_x=False): 

19 return cp.ndarray(fft.fftn(array_Q._data, 

20 shape, 

21 norm=norm, 

22 overwrite_x=overwrite_x)) 

23 

24 

25def rfftn(array_Q, 

26 shape=None, 

27 norm=None, 

28 overwrite_x=False): 

29 return cp.ndarray(fft.rfftn(array_Q._data, 

30 shape, 

31 norm=norm, 

32 overwrite_x=overwrite_x)) 

33 

34 

35def irfftn(array_Q, 

36 shape=None, 

37 norm=None, 

38 overwrite_x=False): 

39 return cp.ndarray(fft.irfftn(array_Q._data, 

40 shape, 

41 norm=norm, 

42 overwrite_x=overwrite_x))