1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
!
! Copyright (C) Quantum ESPRESSO group
!
! This file is distributed under the terms of the
! GNU General Public License. See the file `License'
! in the root directory of the present distribution,
! or http://www.gnu.org/copyleft/gpl.txt .
!
!=---------------------------------------------------------------------------=!
MODULE fft_interfaces
IMPLICIT NONE
PRIVATE
PUBLIC :: fwfft, invfft, fft_interpolate
INTERFACE invfft
!! invfft is the interface to both the standard fft **invfft_x**,
!! and to the "box-grid" version **invfft_b**, used only in CP
!! (the latter has an additional argument)
SUBROUTINE invfft_y( fft_kind, f, dfft, howmany )
USE fft_types, ONLY: fft_type_descriptor
USE fft_param, ONLY :DP
IMPLICIT NONE
CHARACTER(LEN=*), INTENT(IN) :: fft_kind
TYPE(fft_type_descriptor), INTENT(IN) :: dfft
INTEGER, OPTIONAL, INTENT(IN) :: howmany
COMPLEX(DP) :: f(:)
END SUBROUTINE invfft_y
!
SUBROUTINE invfft_b( f, dfft, ia )
USE fft_smallbox_type, ONLY: fft_box_descriptor
USE fft_param, ONLY :DP
IMPLICIT NONE
INTEGER, INTENT(IN) :: ia
TYPE(fft_box_descriptor), INTENT(IN) :: dfft
COMPLEX(DP) :: f(:)
END SUBROUTINE invfft_b
#if defined(__CUDA)
SUBROUTINE invfft_y_gpu( grid_type, f_d, dfft, howmany, stream )
USE fft_types, ONLY: fft_type_descriptor
USE fft_param, ONLY: DP
USE cudafor
IMPLICIT NONE
CHARACTER(LEN=*), INTENT(IN) :: grid_type
TYPE(fft_type_descriptor), INTENT(IN) :: dfft
INTEGER, OPTIONAL, INTENT(IN) :: howmany
INTEGER(kind = cuda_stream_kind), OPTIONAL, INTENT(IN) :: stream
COMPLEX(DP), DEVICE :: f_d(:)
END SUBROUTINE invfft_y_gpu
#endif
END INTERFACE
INTERFACE fwfft
SUBROUTINE fwfft_y( fft_kind, f, dfft, howmany )
USE fft_types, ONLY: fft_type_descriptor
USE fft_param, ONLY :DP
IMPLICIT NONE
CHARACTER(LEN=*), INTENT(IN) :: fft_kind
TYPE(fft_type_descriptor), INTENT(IN) :: dfft
INTEGER, OPTIONAL, INTENT(IN) :: howmany
COMPLEX(DP) :: f(:)
END SUBROUTINE fwfft_y
#if defined(__CUDA)
SUBROUTINE fwfft_y_gpu( grid_type, f_d, dfft, howmany, stream )
USE fft_types, ONLY: fft_type_descriptor
USE fft_param, ONLY: DP
USE cudafor
IMPLICIT NONE
CHARACTER(LEN=*), INTENT(IN) :: grid_type
TYPE(fft_type_descriptor), INTENT(IN) :: dfft
INTEGER, OPTIONAL, INTENT(IN) :: howmany
INTEGER(kind = cuda_stream_kind), OPTIONAL, INTENT(IN) :: stream
COMPLEX(DP), DEVICE :: f_d(:)
END SUBROUTINE fwfft_y_gpu
#endif
END INTERFACE
INTERFACE fft_interpolate
!! fft_interpolate is the interface to utility that fourier interpolate
!! real/complex arrays between two grids
SUBROUTINE fft_interpolate_real( dfft_in, v_in, dfft_out, v_out )
USE fft_param, ONLY :DP
USE fft_types, ONLY: fft_type_descriptor
IMPLICIT NONE
TYPE(fft_type_descriptor), INTENT(IN) :: dfft_in, dfft_out
REAL(DP), INTENT(IN) :: v_in(:)
REAL(DP), INTENT(OUT) :: v_out(:)
END SUBROUTINE fft_interpolate_real
!
SUBROUTINE fft_interpolate_complex( dfft_in, v_in, dfft_out, v_out )
USE fft_param, ONLY :DP
USE fft_types, ONLY: fft_type_descriptor
IMPLICIT NONE
TYPE(fft_type_descriptor), INTENT(IN) :: dfft_in, dfft_out
COMPLEX(DP), INTENT(IN) :: v_in(:)
COMPLEX(DP), INTENT(OUT) :: v_out(:)
END SUBROUTINE fft_interpolate_complex
END INTERFACE
END MODULE fft_interfaces
!=---------------------------------------------------------------------------=!
|
