derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
60 lines
1.9 KiB
Fortran
60 lines
1.9 KiB
Fortran
module boundary_mod
|
|
use kind_mod
|
|
use mesh_mod, only: mesh_t
|
|
implicit none
|
|
!$omp requires unified_shared_memory
|
|
|
|
private
|
|
public :: boundary_conditions
|
|
|
|
contains
|
|
|
|
subroutine boundary_conditions(mesh,u,au)
|
|
type(mesh_t), intent(inout) :: mesh
|
|
real(RK), intent(inout) :: u(:,:)
|
|
real(RK), intent(inout) :: au(:,:)
|
|
integer(IK) :: id,i,j,n_x,n_y
|
|
real(RK) :: invdx2,invdy2
|
|
|
|
n_x = mesh%n_x
|
|
n_y = mesh%n_y
|
|
invdx2 = mesh%dx**-2
|
|
invdy2 = mesh%dy**-2
|
|
|
|
!$omp target teams distribute parallel do private(i,j)
|
|
do id=1,2*n_x+2*n_y-4
|
|
if (id == 1) then
|
|
au(1,1) = (2._RK*u(1,1)-u(2,1))*invdx2 &
|
|
+ (2._RK*u(1,1)-u(1,2))*invdy2
|
|
else if (id <= n_x-1) then
|
|
i = id
|
|
au(i,1) = (-u(i-1,1)+2._RK*u(i,1)-u(i+1,1))*invdx2 &
|
|
+ (2._RK*u(i,1)-u(i,2))*invdy2
|
|
else if (id == n_x) then
|
|
au(n_x,1) = (2._RK*u(n_x,1)-u(n_x-1,1))*invdx2 &
|
|
+ (2._RK*u(n_x,1)-u(n_x,2))*invdy2
|
|
else if (id == n_x+1) then
|
|
au(1,n_y) = (2._RK*u(1,n_y)-u(2,1))*invdx2 &
|
|
+ (2._RK*u(1,n_y)-u(1,n_y-1))*invdy2
|
|
else if (id <= 2*n_x-1) then
|
|
i = id - n_x
|
|
au(i,n_y) = (-u(i-1,n_y)+2._RK*u(i,n_y)-u(i+1,n_y))*invdx2 &
|
|
+ (2._RK*u(i,n_y)-u(i,n_y-1))*invdy2
|
|
else if (id == 2*n_x) then
|
|
au(n_x,n_y) = (2._RK*u(n_x,n_y)-u(n_x-1,1))*invdx2 &
|
|
+ (2._RK*u(n_x,n_y)-u(n_x,n_y-1))*invdy2
|
|
else if (id <= 2*n_x+n_y-2) then
|
|
j = id - 2*n_x + 1
|
|
au(1,j) = (2._RK*u(1,j)-u(2,j))*invdx2 &
|
|
+ (-u(1,j-1)+2._RK*u(1,j)-u(1,j+1))*invdy2
|
|
else
|
|
j = id - 2*n_x - n_y + 3
|
|
au(n_x,j) = (2._RK*u(n_x,j)-u(n_x-1,j))*invdx2 &
|
|
+ (-u(n_x,j-1)+2._RK*u(n_x,j)-u(n_x,j+1))*invdy2
|
|
end if
|
|
end do
|
|
|
|
end subroutine boundary_conditions
|
|
|
|
end module boundary_mod
|