Homec4science
Diffusion wireles-gpu (master) Browse

wireles-gpu/src/forcing.cuf
105cd9b56f1cmaster

Last Change
Raw File

forcing.cuf
View Options

module forcing
use precision
use dimen
contains
attributes(global) subroutine rhs_assemble(rhs,c,divt)
!---------------------------------------------------------------------------
! declaration
!---------------------------------------------------------------------------
implicit none
real(fp), dimension(nx,ny,nz2), intent(out) :: rhs
real(fp), dimension(nx,ny,nz2), intent(in) :: c,divt
integer :: i,j,k
i = (blockIdx%x - 1) * blockDim%x + threadIdx%x
j = (blockIdx%y - 1) * blockDim%y + threadIdx%y
k = (blockIdx%z - 1) * blockDim%z + threadIdx%z
!---------------------------------------------------------------------------
! main code
!---------------------------------------------------------------------------
! do k=2,nzb+1
if (k>1 .and. k<nz2) rhs(i,j,k)=-c(i,j,k)-divt(i,j,k)
! end do
end subroutine rhs_assemble
attributes(global) subroutine const_pres_grad(rhs)
!---------------------------------------------------------------------------
! declaration
!---------------------------------------------------------------------------
implicit none
real(fp), dimension(nx,ny,nz2), intent(inout) :: rhs
integer :: i,j,k
i = (blockIdx%x - 1) * blockDim%x + threadIdx%x
j = (blockIdx%y - 1) * blockDim%y + threadIdx%y
k = (blockIdx%z - 1) * blockDim%z + threadIdx%z
!---------------------------------------------------------------------------
! main code
!---------------------------------------------------------------------------
! do k=2,nzb+1
rhs(i,j,k)=rhs(i,j,k)+pres_force
! end do
end subroutine const_pres_grad
attributes(global) subroutine turbine_force(rhs,turb_f)
!---------------------------------------------------------------------------
! declaration
!---------------------------------------------------------------------------
implicit none
real(fp), dimension(nx,ny,nz2), intent(inout) :: rhs
real(fp), dimension(nx,ny,nz2), intent(in) :: turb_f
integer :: i,j,k
i = (blockIdx%x - 1) * blockDim%x + threadIdx%x
j = (blockIdx%y - 1) * blockDim%y + threadIdx%y
k = (blockIdx%z - 1) * blockDim%z + threadIdx%z
!---------------------------------------------------------------------------
! main code
!---------------------------------------------------------------------------
rhs(i,j,k)=rhs(i,j,k) - turb_f(i,j,k)
end subroutine turbine_force
end module forcing
c4science · Help