Refresh OpenMP in UVTVD

ecmwf-ifs · Feb 10, 2025 · 8aaf383 · 8aaf383
1 parent e4fd8de
commit 8aaf383
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/trans/gpu/internal/uvtvd_mod.F90 b/src/trans/gpu/internal/uvtvd_mod.F90
@@ -84,6 +84,9 @@ SUBROUTINE UVTVD(KF_UV,PU,PV,PVOR,PDIV)
 !*       1.    COMPUTE U V FROM VORTICITY AND DIVERGENCE.
 !              ------------------------------------------
 
+#ifdef OMPGPU
+!$OMP TARGET DATA MAP(PRESENT,ALLOC:D,D_MYMS,D_NUMP,R,R_NTMAX,FG,ZEPSNM,PU,PV,PVOR,PDIV)
+#endif
 #ifdef ACCGPU
 !$ACC DATA &
 !$ACC& PRESENT(D,D_MYMS,D_NUMP,R,R_NTMAX) &
@@ -171,6 +174,9 @@ SUBROUTINE UVTVD(KF_UV,PU,PV,PVOR,PDIV)
 #ifdef ACCGPU
 !$ACC END DATA
 #endif
+#ifdef OMPGPU
+!$OMP END TARGET DATA
+#endif
 !     ------------------------------------------------------------------
 END ASSOCIATE