Index: LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/lmdz_atke_exchange_coeff.f90
===================================================================
--- LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/lmdz_atke_exchange_coeff.f90	(revision 5921)
+++ LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/lmdz_atke_exchange_coeff.f90	(revision 5922)
@@ -8,4 +8,5 @@
                         wind_u,wind_v,temp,qvap,play,pinterf,cdrag_uv, &
                         tke,eps,tke_shear,tke_buoy,tke_trans,Km_out,Kh_out)
+!$gpum horizontal ngrid
 
 !========================================================================
@@ -411,4 +412,5 @@
 !===============================================================================================
 subroutine atke_vdiff_tke(ngrid,nlay,dtime,z_lay,z_interf,temp,play,l_exchange,Sm,tke,tke_trans)
+!$gpum horizontal ngrid
 
 ! routine that computes the vertical diffusion of TKE by the turbulence
Index: LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/lmdz_call_atke.f90
===================================================================
--- LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/lmdz_call_atke.f90	(revision 5921)
+++ LMDZ6/branches/PBLSURF_GPUPORT/libf/phylmd/lmdz_call_atke.f90	(revision 5922)
@@ -103,4 +103,5 @@
 
 subroutine atke_explicit_prediction(ngrid,nlay,rg,rd,dtime,pinterf,play,temp,wind1,x_in,K_in,x_surf,cdrag,x_predict)
+!$gpum horizontal ngrid
 
 INTEGER, INTENT(IN) :: ngrid ! number of horizontal index (flat grid)