diff options
Diffstat (limited to 'src/Derivatives_4_3.F90')
| -rw-r--r-- | src/Derivatives_4_3.F90 | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/Derivatives_4_3.F90 b/src/Derivatives_4_3.F90 index 19a012e..1bc9734 100644 --- a/src/Derivatives_4_3.F90 +++ b/src/Derivatives_4_3.F90 @@ -41,6 +41,7 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) if ( bb(1) == 0 ) then il = 1 + gsize else +!$omp parallel workshare dvar(1,:,:) = ( q(1,1) * var(1,:,:) + q(2,1) * var(2,:,:) + & q(3,1) * var(3,:,:) + q(4,1) * var(4,:,:) ) * idel dvar(2,:,:) = ( q(1,2) * var(1,:,:) + q(2,2) * var(2,:,:) + & @@ -57,11 +58,13 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) q(3,5) * var(3,:,:) + q(4,5) * var(4,:,:) + & q(5,5) * var(5,:,:) + q(6,5) * var(6,:,:) + & q(7,5) * var(7,:,:) ) * idel +!$omp end parallel workshare il = 6 end if if ( bb(2) == 0 ) then ir = ni - gsize else +!$omp parallel workshare dvar(ni-4,:,:) = - ( q(1,5) * var(ni,:,:) + q(2,5) * var(ni-1,:,:) + & q(3,5) * var(ni-2,:,:) + q(4,5) * var(ni-3,:,:) + & q(5,5) * var(ni-4,:,:) + q(6,5) * var(ni-5,:,:) + & @@ -81,13 +84,16 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) dvar(ni,:,:) = - ( q(1,1) * var(ni,:,:) + q(2,1) * var(ni-1,:,:) + & q(3,1) * var(ni-2,:,:) + & q(4,1) * var(ni-3,:,:) ) * idel +!$omp end parallel workshare ir = ni - 5 end if if (il > ir+1) call CCTK_WARN (0, "domain too small") +!$omp parallel workshare dvar(il:ir,:,:) = ( a(1) * ( var(il+1:ir+1,:,:) - & var(il-1:ir-1,:,:) ) + & a(2) * ( var(il+2:ir+2,:,:) - & var(il-2:ir-2,:,:) ) ) * idel +!$omp end parallel workshare case (1) direction if ( zero_derivs_y /= 0 ) then dvar = zero @@ -95,6 +101,7 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) if ( bb(1) == 0 ) then jl = 1 + gsize else +!$omp parallel workshare dvar(:,1,:) = ( q(1,1) * var(:,1,:) + q(2,1) * var(:,2,:) + & q(3,1) * var(:,3,:) + q(4,1) * var(:,4,:) ) * idel dvar(:,2,:) = ( q(1,2) * var(:,1,:) + q(2,2) * var(:,2,:) + & @@ -111,11 +118,13 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) q(3,5) * var(:,3,:) + q(4,5) * var(:,4,:) + & q(5,5) * var(:,5,:) + q(6,5) * var(:,6,:) + & q(7,5) * var(:,7,:) ) * idel +!$omp end parallel workshare jl = 6 end if if ( bb(2) == 0 ) then jr = nj - gsize else +!$omp parallel workshare dvar(:,nj-4,:) = - ( q(1,5) * var(:,nj,:) + q(2,5) * var(:,nj-1,:) + & q(3,5) * var(:,nj-2,:) + q(4,5) * var(:,nj-3,:) + & q(5,5) * var(:,nj-4,:) + q(6,5) * var(:,nj-5,:) + & @@ -135,13 +144,16 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) dvar(:,nj,:) = - ( q(1,1) * var(:,nj,:) + q(2,1) * var(:,nj-1,:) + & q(3,1) * var(:,nj-2,:) + & q(4,1) * var(:,nj-3,:) ) * idel +!$omp end parallel workshare jr = nj - 5 end if if (jl > jr+1) call CCTK_WARN (0, "domain too small") +!$omp parallel workshare dvar(:,jl:jr,:) = ( a(1) * ( var(:,jl+1:jr+1,:) - & var(:,jl-1:jr-1,:) ) + & a(2) * ( var(:,jl+2:jr+2,:) - & var(:,jl-2:jr-2,:) ) ) * idel +!$omp end parallel workshare end if case (2) direction if ( zero_derivs_z /= 0 ) then @@ -150,6 +162,7 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) if ( bb(1) == 0 ) then kl = 1 + gsize else +!$omp parallel workshare dvar(:,:,1) = ( q(1,1) * var(:,:,1) + q(2,1) * var(:,:,2) + & q(3,1) * var(:,:,3) + q(4,1) * var(:,:,4) ) * idel dvar(:,:,2) = ( q(1,2) * var(:,:,1) + q(2,2) * var(:,:,2) + & @@ -166,11 +179,13 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) q(3,5) * var(:,:,3) + q(4,5) * var(:,:,4) + & q(5,5) * var(:,:,5) + q(6,5) * var(:,:,6) + & q(7,5) * var(:,:,7) ) * idel +!$omp end parallel workshare kl = 6 end if if ( bb(2) == 0 ) then kr = nk - gsize else +!$omp parallel workshare dvar(:,:,nk-4) = - ( q(1,5) * var(:,:,nk) + q(2,5) * var(:,:,nk-1) + & q(3,5) * var(:,:,nk-2) + q(4,5) * var(:,:,nk-3) + & q(5,5) * var(:,:,nk-4) + q(6,5) * var(:,:,nk-5) + & @@ -190,13 +205,16 @@ subroutine deriv_gf_4_3 ( var, ni, nj, nk, dir, bb, gsize, delta, dvar ) dvar(:,:,nk) = - ( q(1,1) * var(:,:,nk) + q(2,1) * var(:,:,nk-1) + & q(3,1) * var(:,:,nk-2) + & q(4,1) * var(:,:,nk-3) ) * idel +!$omp end parallel workshare kr = nk - 5 end if if (kl > kr+1) call CCTK_WARN (0, "domain too small") +!$omp parallel workshare dvar(:,:,kl:kr) = ( a(1) * ( var(:,:,kl+1:kr+1) - & var(:,:,kl-1:kr-1) ) + & a(2) * ( var(:,:,kl+2:kr+2) - & var(:,:,kl-2:kr-2) ) ) * idel +!$omp end parallel workshare end if end select direction end subroutine deriv_gf_4_3 |