Fix OMP version of loop in LDFOU2

Co-authored-by: Paul Mullowney <[email protected]> There was a typo in ABORT_TRANS. Thanks to Paul Mullowney for implementing OpenMP-offload version of OpenACC directive.
ecmwf-ifs · Mar 27, 2024 · 7fb1d17 · 7fb1d17
1 parent 9c89429
commit 7fb1d17
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/trans/gpu/internal/ldfou2_mod.F90 b/src/trans/gpu/internal/ldfou2_mod.F90
@@ -75,7 +75,9 @@ SUBROUTINE LDFOU2(KF_UV, P_FOURIER_FIELDS)
 IF (KF_UV > 0) THEN
 
 #ifdef OMPGPU
-  CALL ABOR_TRANS("LDFOU2: OpenMP offload not implemented yet for LDFOU2")
+  !$OMP TARGET TEAMS DISTRIBUTE PARALLEL DO COLLAPSE(3) DEFAULT(NONE) &
+  !$OMP& PRIVATE(KMLOC,JGL,J,KM,ISL) MAP(TO:KF_UV,F_RACTHE,D_MYMS,G_NDGLU,P_FOURIER_FIELDS) &
+  !$OMP& SHARED(D_NUMP,R_NDGNH,KF_UV,F_RACTHE,D_MYMS,G_NDGLU,P_FOURIER_FIELDS)
 #endif
 #ifdef ACCGPU
   !$ACC PARALLEL LOOP COLLAPSE(3) DEFAULT(NONE) &