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

Juan 21/09/2021:ZSOLVER/tridiag_thermo.f90 , use acc ... gang, vector , or...

Juan 21/09/2021:ZSOLVER/tridiag_thermo.f90 , use acc ... gang, vector , or vectorisation in only 1 dim on GPU
parent 5ee27151
No related branches found
No related tags found
No related merge requests found
......@@ -380,19 +380,15 @@ END DO !CONCURRENT
!
!$acc loop seq
DO JK = IKB+KKL,IKE-KKL,KKL
!$acc loop independent collapse(2)
! acc loop gang, vector collapse(2)
! gang+vector needed or parallisation vector only
!$acc loop independent gang, vector collapse(2)
DO CONCURRENT ( JI=1:JIU,JJ=1:JJU)
!DO JJ=1,JJU
! DO JI=1,JIU
ZGAM(JI,JJ,JK) = ZC(JI,JJ,JK-KKL) / ZBET(JI,JJ)
! gam(k) = c(k-1) / bet
ZBET(JI,JJ) = ZB(JI,JJ,JK) - ZA(JI,JJ,JK) * ZGAM(JI,JJ,JK)
! bet = b(k) - a(k)* gam(k)
PVARP(JI,JJ,JK)= ( ZY(JI,JJ,JK) - ZA(JI,JJ,JK) * PVARP(JI,JJ,JK-KKL) ) / ZBET(JI,JJ)
! res(k) = (y(k) -a(k)*res(k-1))/ bet
! END DO
!END DO
END DO !CONCURRENT
END DO
! special treatment for the last level
......@@ -411,8 +407,8 @@ END DO !CONCURRENT
!
!$acc loop seq
DO JK = IKE-KKL,IKB,-1*KKL
!$acc loop independent collapse(2)
! acc loop gang, vector collapse(2)
! gang+vector needed or parallisation vector only
!$acc loop independent gang, vector collapse(2)
DO CONCURRENT ( JI=1:JIU,JJ=1:JJU)
PVARP(JI,JJ,JK) = PVARP(JI,JJ,JK) - ZGAM(JI,JJ,JK+KKL) * PVARP(JI,JJ,JK+KKL)
END DO !CONCURRENT
......
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