Skip to content

Commit cab026e

Browse files
author
Tanush Prathi
committed
Fixed Cylindrical on OpenMP Frontier
1 parent 5a97a92 commit cab026e

File tree

1 file changed

+8
-16
lines changed

1 file changed

+8
-16
lines changed

src/simulation/m_fftw.fpp

Lines changed: 8 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -130,8 +130,6 @@ contains
130130
impure subroutine s_apply_fourier_filter(q_cons_vf)
131131

132132
type(scalar_field), dimension(sys_size), intent(inout) :: q_cons_vf
133-
real(c_double), pointer :: p_real(:)
134-
complex(c_double_complex), pointer :: p_cmplx(:), p_fltr_cmplx(:)
135133
integer :: i, j, k, l !< Generic loop iterators
136134
integer :: ierr !< Generic flag used to identify and report GPU errors
137135

@@ -159,16 +157,12 @@ contains
159157
end do
160158
#:endcall GPU_PARALLEL_LOOP
161159

162-
p_real => data_real_gpu
163-
p_cmplx => data_cmplx_gpu
164-
p_fltr_cmplx => data_fltr_cmplx_gpu
165160

166-
#:call GPU_DATA(attach='[p_real, p_cmplx, p_fltr_cmplx]')
167-
#:call GPU_HOST_DATA(use_device_ptr='[p_real, p_cmplx, p_fltr_cmplx]')
161+
#:call GPU_HOST_DATA(use_device_addr='[data_real_gpu, data_cmplx_gpu, data_fltr_cmplx_gpu]')
168162
#if defined(__PGI)
169163
ierr = cufftExecD2Z(fwd_plan_gpu, data_real_gpu, data_cmplx_gpu)
170164
#else
171-
ierr = hipfftExecD2Z(fwd_plan_gpu, c_loc(p_real), c_loc(p_cmplx))
165+
ierr = hipfftExecD2Z(fwd_plan_gpu, data_real_gpu, data_cmplx_gpu)
172166
call hipCheck(hipDeviceSynchronize())
173167
#endif
174168
#:endcall GPU_HOST_DATA
@@ -185,11 +179,11 @@ contains
185179
end do
186180
#:endcall GPU_PARALLEL_LOOP
187181

188-
#:call GPU_HOST_DATA(use_device_ptr='[p_real, p_fltr_cmplx]')
182+
#:call GPU_HOST_DATA(use_device_addr='[data_real_gpu, data_cmplx_gpu, data_fltr_cmplx_gpu]')
189183
#if defined(__PGI)
190184
ierr = cufftExecZ2D(bwd_plan_gpu, data_fltr_cmplx_gpu, data_real_gpu)
191185
#else
192-
ierr = hipfftExecZ2D(bwd_plan_gpu, c_loc(p_fltr_cmplx), c_loc(p_real))
186+
ierr = hipfftExecZ2D(bwd_plan_gpu, data_fltr_cmplx_gpu, data_real_gpu)
193187
call hipCheck(hipDeviceSynchronize())
194188
#endif
195189
#:endcall GPU_HOST_DATA
@@ -227,11 +221,11 @@ contains
227221
end do
228222
#:endcall GPU_PARALLEL_LOOP
229223

230-
#:call GPU_HOST_DATA(use_device_ptr='[p_real, p_cmplx]')
224+
#:call GPU_HOST_DATA(use_device_addr='[data_real_gpu, data_cmplx_gpu, data_fltr_cmplx_gpu]')
231225
#if defined(__PGI)
232226
ierr = cufftExecD2Z(fwd_plan_gpu, data_real_gpu, data_cmplx_gpu)
233227
#else
234-
ierr = hipfftExecD2Z(fwd_plan_gpu, c_loc(p_real), c_loc(p_cmplx))
228+
ierr = hipfftExecD2Z(fwd_plan_gpu, data_real_gpu, data_cmplx_gpu)
235229
call hipCheck(hipDeviceSynchronize())
236230
#endif
237231
#:endcall GPU_HOST_DATA
@@ -249,11 +243,11 @@ contains
249243
end do
250244
#:endcall GPU_PARALLEL_LOOP
251245

252-
#:call GPU_HOST_DATA(use_device_ptr='[p_real, p_fltr_cmplx]')
246+
#:call GPU_HOST_DATA(use_device_addr='[data_real_gpu, data_cmplx_gpu, data_fltr_cmplx_gpu]')
253247
#if defined(__PGI)
254248
ierr = cufftExecZ2D(bwd_plan_gpu, data_fltr_cmplx_gpu, data_real_gpu)
255249
#else
256-
ierr = hipfftExecZ2D(bwd_plan_gpu, c_loc(p_fltr_cmplx), c_loc(p_real))
250+
ierr = hipfftExecZ2D(bwd_plan_gpu, data_fltr_cmplx_gpu, data_real_gpu)
257251
call hipCheck(hipDeviceSynchronize())
258252
#endif
259253
#:endcall GPU_HOST_DATA
@@ -268,9 +262,7 @@ contains
268262
end do
269263
end do
270264
#:endcall GPU_PARALLEL_LOOP
271-
272265
end do
273-
#:endcall GPU_DATA
274266

275267
#else
276268
Nfq = 3

0 commit comments

Comments
 (0)