Quentin 02/07/2024: optimize perf bugfix GPU for turb

src/PHYEX/turb/turb.f90

@@ -1990,14 +1990,20 @@ END IF
 CALL ETHETA(D,CST,KRR,KRRI,PTHLT,PRT,ZLOCPEXNM,ZATHETA,PSRCT,GOCEAN,OCOMPUTE_SRC,ZETHETA)
 CALL EMOIST(D,CST,KRR,KRRI,PTHLT,PRT,ZLOCPEXNM,ZAMOIST,PSRCT,GOCEAN,ZEMOIST)
 !
-!$acc kernels present(ZWORK2D,PLM)
 IF (KRR>0) THEN
-  DO JK = IKTB+1,IKTE-1
-    DO JIJ=IIJB,IIJE
+!$acc kernels
+!$acc loop independent collapse(2)
+  DO CONCURRENT (JK=IKTB+1:IKTE-1,JIJ=IIJB:IIJE)
       ZDTHLDZ(JIJ,JK)= 0.5*((PTHLT(JIJ,JK+IKL)-PTHLT(JIJ,JK    ))/PDZZ(JIJ,JK+IKL)+ &
                               (PTHLT(JIJ,JK    )-PTHLT(JIJ,JK-IKL))/PDZZ(JIJ,JK    ))
       ZDRTDZ(JIJ,JK) = 0.5*((PRT(JIJ,JK+IKL,1)-PRT(JIJ,JK    ,1))/PDZZ(JIJ,JK+IKL)+ &
                               (PRT(JIJ,JK    ,1)-PRT(JIJ,JK-IKL,1))/PDZZ(JIJ,JK    ))
+  END DO
+!$acc end kernels
+
+!$acc kernels
+!$acc loop independent collapse(2) private(ZVAR)
+  DO CONCURRENT (JK=IKTB+1:IKTE-1,JIJ=IIJB:IIJE)
       IF (GOCEAN) THEN
         ZVAR=CST%XG*(CST%XALPHAOC*ZDTHLDZ(JIJ,JK)-CST%XBETAOC*ZDRTDZ(JIJ,JK))
       ELSE
@@ -2009,11 +2015,13 @@ IF (KRR>0) THEN
         PLM(JIJ,JK)=MAX(CST%XMNH_EPSILON,MIN(PLM(JIJ,JK), &
                       0.76* SQRT(PTKET(JIJ,JK)/ZVAR)))
       END IF
-    END DO
   END DO
+!$acc end kernels
+
 ELSE! For dry atmos or unsalted ocean runs
-  DO JK = IKTB+1,IKTE-1
-    DO JIJ=IIJB,IIJE
+!$acc kernels
+!$acc loop independent collapse(2) private(ZVAR)
+  DO CONCURRENT (JK=IKTB+1:IKTE-1,JIJ=IIJB:IIJE)
       ZDTHLDZ(JIJ,JK)= 0.5*((PTHLT(JIJ,JK+IKL)-PTHLT(JIJ,JK    ))/PDZZ(JIJ,JK+IKL)+ &
                               (PTHLT(JIJ,JK    )-PTHLT(JIJ,JK-IKL))/PDZZ(JIJ,JK    ))
       IF (GOCEAN) THEN
@@ -2026,9 +2034,10 @@ ELSE! For dry atmos or unsalted ocean runs
         PLM(JIJ,JK)=MAX(CST%XMNH_EPSILON,MIN(PLM(JIJ,JK), &
                       0.76* SQRT(PTKET(JIJ,JK)/ZVAR)))
       END IF
-    END DO
   END DO
+!$acc end kernels
 END IF
+!$acc kernels present(ZWORK2D, PLM)
 !  special case near the surface
 !$mnh_expand_array(JIJ=IIJB:IIJE)
 ZDTHLDZ(:,IKB)=(PTHLT(:,IKB+IKL)-PTHLT(:,IKB))/PDZZ(:,IKB+IKL)