Skip to content
Snippets Groups Projects
Commit 5e6e4d38 authored by ESCOBAR MUNOZ Juan's avatar ESCOBAR MUNOZ Juan
Browse files

Juan 15/03/2022:tensor/discretisation.f90, Cray OpenACC optimization , discard...

Juan 15/03/2022:tensor/discretisation.f90, Cray OpenACC optimization , discard collapse + do concurrent
parent 87c311f9
No related branches found
No related tags found
No related merge requests found
...@@ -794,7 +794,9 @@ end subroutine construct_vertical_coeff ...@@ -794,7 +794,9 @@ end subroutine construct_vertical_coeff
!$acc kernels !$acc kernels
iz=1 iz=1
#ifdef MNH_COMPILER_NVHPC
!$acc loop independent collapse(2) !$acc loop independent collapse(2)
#endif
do ij=ijb,ije do ij=ijb,ije
do ii=iib,iie do ii=iib,iie
zv_st(ii,ij,iz) = zd_k(iz)* ( (-zb_k(iz)-zc_k(iz))*Tij * zu_st(ii,ij,iz ) & zv_st(ii,ij,iz) = zd_k(iz)* ( (-zb_k(iz)-zc_k(iz))*Tij * zu_st(ii,ij,iz ) &
...@@ -803,7 +805,9 @@ end subroutine construct_vertical_coeff ...@@ -803,7 +805,9 @@ end subroutine construct_vertical_coeff
end do end do
! !
do iz=2,ize-1 do iz=2,ize-1
#ifdef MNH_COMPILER_NVHPC
!$acc loop independent collapse(2) !$acc loop independent collapse(2)
#endif
do ij=ijb,ije do ij=ijb,ije
do ii=iib,iie do ii=iib,iie
zv_st(ii,ij,iz) = zd_k(iz)* ( ((-zb_k(iz)-zc_k(iz))*Tij - 4.0_rl ) * zu_st(ii,ij,iz) & zv_st(ii,ij,iz) = zd_k(iz)* ( ((-zb_k(iz)-zc_k(iz))*Tij - 4.0_rl ) * zu_st(ii,ij,iz) &
...@@ -819,14 +823,15 @@ end subroutine construct_vertical_coeff ...@@ -819,14 +823,15 @@ end subroutine construct_vertical_coeff
end do end do
! !
iz=ize iz=ize
#ifdef MNH_COMPILER_NVHPC
!$acc loop independent collapse(2) !$acc loop independent collapse(2)
#endif
do ij=ijb,ije do ij=ijb,ije
do ii=iib,iie do ii=iib,iie
zv_st(ii,ij,iz) = zd_k(iz)* ( (-zb_k(iz)-zc_k(iz))*Tij * zu_st(ii,ij,iz) & zv_st(ii,ij,iz) = zd_k(iz)* ( (-zb_k(iz)-zc_k(iz))*Tij * zu_st(ii,ij,iz) &
+zc_k(iz) *Tij * zu_st(ii,ij,iz-1) ) +zc_k(iz) *Tij * zu_st(ii,ij,iz-1) )
end do end do
end do end do
!$acc end kernels !$acc end kernels
endif endif
...@@ -1600,7 +1605,7 @@ end subroutine construct_vertical_coeff ...@@ -1600,7 +1605,7 @@ end subroutine construct_vertical_coeff
zSut0_st => Sut0%st zSut0_st => Sut0%st
zu_st => u%st zu_st => u%st
!$acc kernels !$acc kernels present(zSut0_st,zu_st)
zSut0_st(:,:,:) = zu_st(:,:,:) zSut0_st(:,:,:) = zu_st(:,:,:)
!$acc end kernels !$acc end kernels
...@@ -2108,7 +2113,9 @@ end subroutine line_Jacobi_mnh ...@@ -2108,7 +2113,9 @@ end subroutine line_Jacobi_mnh
! !
!$acc loop seq !$acc loop seq
do iz=2,nz-1 do iz=2,nz-1
#ifdef MNH_COMPILER_NVHPC
!$acc loop independent collapse(2) !$acc loop independent collapse(2)
#endif
do ij=ijb,ije do ij=ijb,ije
do ii=iib,iie do ii=iib,iie
zSu_out_st(ii,ij,iz) = (zSr_st(ii,ij,iz) / (Tijs*zd_k(iz)) & zSu_out_st(ii,ij,iz) = (zSr_st(ii,ij,iz) / (Tijs*zd_k(iz)) &
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment