Add OpenMP to ext_acc

ecmwf-ifs · Jan 23, 2025 · af93699 · af93699
1 parent c2783f5
commit af93699
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/trans/gpu/algor/ext_acc.F90 b/src/trans/gpu/algor/ext_acc.F90
@@ -290,6 +290,7 @@ subroutine ext_acc_create(ptrs, stream)
       !$acc enter data create(pp) async(stream_act)
 #endif
 #ifdef OMPGPU
+      !$omp target enter data map(alloc:pp)
 #endif
     enddo
   end subroutine
@@ -327,6 +328,7 @@ subroutine ext_acc_copyin(ptrs, stream)
       !$acc enter data copyin(pp) async(stream_act)
 #endif
 #ifdef OMPGPU
+      !$omp target enter data map(to:pp)
 #endif
     enddo
   end subroutine
@@ -363,6 +365,7 @@ subroutine ext_acc_copyout(ptrs, stream)
       !$acc exit data copyout(pp) async(stream_act)
 #endif
 #ifdef OMPGPU
+      !$omp target exit data map(from:pp)
 #endif
     enddo
   end subroutine
@@ -399,6 +402,7 @@ subroutine ext_acc_delete(ptrs, stream)
       !$acc exit data delete(pp) async(stream_act)
 #endif
 #ifdef OMPGPU
+      !$omp target exit data map(delete:pp)
 #endif
     enddo
   end subroutine