Compare commits
33 commits
mpi_fdm
...
diffusivit
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
0849b31e7b | ||
|
|
d798fe343e | ||
|
|
8431f9cac8 | ||
|
|
2934fb0e44 | ||
|
|
f38b995827 | ||
|
|
0776f3ef6e | ||
|
|
e16ead959d | ||
|
|
f5b0550436 | ||
|
|
4a8a67d241 | ||
|
|
db7732025a | ||
|
|
a56c4d978c | ||
|
|
e659052e26 | ||
|
|
b426be89ad | ||
|
|
749e77de32 | ||
|
|
6933297130 | ||
|
|
75f0feaa02 | ||
|
|
27bf7ab6dd | ||
|
|
3ee83a9fff | ||
|
|
56fc50e9f0 | ||
|
|
b39e4cbef4 | ||
|
|
dac299c060 | ||
|
|
3225cd757b | ||
|
|
6c75483f1e | ||
|
|
383b2c5aa7 | ||
|
|
6f0c1f44fa | ||
|
|
4f60757a9a | ||
|
|
016bd9fb80 | ||
|
|
804698f717 | ||
|
|
4f7bb89461 | ||
|
|
50bc5073c1 | ||
|
|
9c48452b0f | ||
|
|
6d7b31c555 | ||
|
|
cfe19ffb17 |
12 changed files with 1695 additions and 414 deletions
86
Makefile
86
Makefile
|
|
@ -4,77 +4,37 @@
|
|||
|
||||
MPIF90 = blah
|
||||
|
||||
ifeq ($(OMPI), 1)
|
||||
FFTW_HOME = ../libs/fftw-3.2.2/lib
|
||||
OPENMPI_HOME = /home/comb/ignis/openmpi/
|
||||
MPIF90 = $(OPENMPI_HOME)/bin/mpif90
|
||||
FCFLAGS = -i8 -r8 -c -openmp $(MPI_COMPILE_FLAGS) -I$(FFTW_HOME)/include -I$(OPENMPI_HOME)/include
|
||||
LDFLAGS = -i8 -r8 -openmp $(MPI_LD_FLAGS) -L$(FFTW_HOME)/lib -lfftw3 -L$(OPENMPI_HOME)/lib -lfftw3 -lm -shared-intel
|
||||
FCFLAGS_F77 = -Mextend
|
||||
FFTW_HOME = $(FFTW_DIR)
|
||||
MPIF90 = mpif90
|
||||
FCFLAGS = -Ofast -march=native -Wall -fdefault-integer-8 -fdefault-double-8 -fdefault-real-8 -c $(MPI_COMPILE_FLAGS) -I$(FFTW_HOME)/include
|
||||
LDFLAGS = -Ofast -march=native -Wall -fdefault-integer-8 -fdefault-double-8 -fdefault-real-8 $(MPIPP_LD_FLAGS) $(MPI_LD_FLAGS) -L$(FFTW_HOME)/lib -lfftw3 -lm
|
||||
FCFLAGS_F77 =
|
||||
|
||||
MODULES += m_openmpi.o
|
||||
else
|
||||
FFTW_HOME = ../libs/fftw-3.2.2/lib
|
||||
|
||||
|
||||
# compile for mpiP MPI Profiling
|
||||
ifeq ($(PROFILER), mpiP)
|
||||
FFTW_HOME = $(FFTW_DIR)
|
||||
MPIF90 = mpif90
|
||||
FCFLAGS = -i8 -r8 -c $(MPI_COMPILE_FLAGS) -I$(FFTW_HOME)/include
|
||||
LDFLAGS = -i8 -r8 $(MPI_LD_FLAGS) -L$(FFTW_HOME)/lib -lfftw3 -lm
|
||||
FCFLAGS_F77 = -Mextend
|
||||
|
||||
MODULES += m_mpi.o
|
||||
endif
|
||||
|
||||
# Franklin cluster, NERSC
|
||||
ifeq ($(NERSC_HOST), franklin)
|
||||
MPIF90 = ftn
|
||||
|
||||
FCFLAGS = -target=linux -i8 -r8 -O4 -c
|
||||
LDFLAGS = -target=linux -i8 -r8 -O4 -lfftw3
|
||||
FCFLAGS_F77 = -Mextend
|
||||
endif
|
||||
|
||||
# Hopper cluster, NERSC
|
||||
ifeq ($(NERSC_HOST), hopper)
|
||||
MPIF90 = ftn
|
||||
|
||||
FCFLAGS = -target=linux -i8 -r8 -O4 -c
|
||||
LDFLAGS = -target=linux -i8 -r8 -O4 -lfftw3
|
||||
FCFLAGS_F77 = -Mextend
|
||||
endif
|
||||
|
||||
# Yellowrail cluster, LANL
|
||||
ifeq ($(HOSTNAME), yr-fe1.lanl.gov)
|
||||
MPIF90 = mpif90
|
||||
|
||||
FCFLAGS = -i8 -r8 -O4 -c $(MPI_COMPILE_FLAGS) -I$(FFTW_INCLUDE)
|
||||
LDFLAGS = -i8 -r8 -O4 -lfftw3 $(MPI_LD_FLAGS) -L$(FFTW_HOME)/lib
|
||||
endif
|
||||
|
||||
# Coyote cluster, LANL
|
||||
ifeq ($(HOSTNAME), cy-c2.lanl.gov)
|
||||
MPIF90 = mpif90
|
||||
FCFLAGS = -i8 -r8 -O4 -c $(MPI_COMPILE_FLAGS) -I$(FFTW_INCLUDE)
|
||||
LDFLAGS = -i8 -r8 -O4 -lfftw3 $(MPI_LD_FLAGS) -L$(FFTW_HOME)/lib
|
||||
endif
|
||||
|
||||
# Linux Cluster, CMU
|
||||
ifeq ($(HOSTNAME), karman.me.cmu.edu)
|
||||
FFTW_HOME = ../libs/fftw-3.2.2/lib
|
||||
MPIF90 = mpif90
|
||||
FCFLAGS = -i8 -r8 -c $(MPI_COMPILE_FLAGS) -I$(FFTW_HOME)/include
|
||||
LDFLAGS = -i8 -r8 $(MPI_LD_FLAGS) -L$(FFTW_HOME)/lib -lfftw3 -lm
|
||||
FCFLAGS_F77 = -Mextend
|
||||
FCFLAGS = -Ofast -march=native -Wall -fdefault-integer-8 -fdefault-double-8 -fdefault-real-8 -c $(MPI_COMPILE_FLAGS) -I$(FFTW_HOME)/include
|
||||
MPIP_LD_FLAGS = -L$(MPIP_DIR)/lib -lmpiP -lm -lbfd -liberty -lunwind
|
||||
LDFLAGS = -Ofast -march=native -Wall -fdefault-integer-8 -fdefault-double-8 -fdefault-real-8 $(MPIPP_LD_FLAGS) $(MPI_LD_FLAGS) -L$(FFTW_HOME)/lib -lfftw3 -lm
|
||||
FCFLAGS_F77 =
|
||||
|
||||
endif
|
||||
|
||||
# sparrow.stanford.edu, Mac OS X box
|
||||
ifeq ($(HOSTNAME), sparrow.stanford.edu)
|
||||
MPIF90 = mpif90
|
||||
FCFLAGS = -fdefault-real-8 -fdefault-integer-8 -finit-integer=0 -finit-real=zero -c
|
||||
FCFLAGS_F77 = -ffixed-form -ffixed-line-length-none
|
||||
FCFLAGS_F90 = -ffree-form -ffree-line-length-none
|
||||
LDFLAGS = -fdefault-real-8 -fdefault-integer-8 -finit-integer=0 -finit-real=zero -lmpi -lmpi -lfftw3 -lm
|
||||
# tau source instrumentation
|
||||
ifeq ($(PROFILER), tau_source)
|
||||
FFTW_HOME = $(FFTW_DIR)
|
||||
MPIF90 = tau_f90.sh
|
||||
FCFLAGS = -Ofast -march=native -Wall -fdefault-integer-8 -fdefault-double-8 -fdefault-real-8 -c $(MPI_COMPILE_FLAGS) -I$(FFTW_HOME)/include
|
||||
LDFLAGS = -Ofast -march=native -Wall -fdefault-integer-8 -fdefault-double-8 -fdefault-real-8 $(MPI_LD_FLAGS) -L$(FFTW_HOME)/lib -lfftw3 -lm
|
||||
FCFLAGS_F77 =
|
||||
|
||||
endif
|
||||
|
||||
|
||||
# Program name
|
||||
PROG = hit3d.x
|
||||
|
||||
|
|
|
|||
45
gather_4.f90
45
gather_4.f90
|
|
@ -23,8 +23,6 @@ subroutine gather_4
|
|||
|
||||
! number of variables to write out
|
||||
n_out = 3
|
||||
if (int_scalars) n_out = n_out + n_scalars
|
||||
if (les .and. n_les>0) n_out = n_out + n_les
|
||||
|
||||
! putting all variables in wrk array
|
||||
do k = 1,nz
|
||||
|
|
@ -44,50 +42,13 @@ subroutine gather_4
|
|||
|
||||
! velocities
|
||||
call xFFT3d(-1,1)
|
||||
fname = 'u.'//file_ext
|
||||
tmp4(1:nx,1:ny,1:nz) = wrk(1:nx,1:ny,1:nz,1)
|
||||
call gather_tmp4(u_)
|
||||
|
||||
call xFFT3d(-1,2)
|
||||
fname = 'v.'//file_ext
|
||||
tmp4(1:nx,1:ny,1:nz) = wrk(1:nx,1:ny,1:nz,2)
|
||||
call gather_tmp4(v_)
|
||||
|
||||
call xFFT3d(-1,3)
|
||||
fname = 'w.'//file_ext
|
||||
tmp4(1:nx,1:ny,1:nz) = wrk(1:nx,1:ny,1:nz,3)
|
||||
call gather_tmp4(w_)
|
||||
|
||||
! scalars
|
||||
if (int_scalars) then
|
||||
do n = 1,n_scalars
|
||||
call xFFT3d(-1,3+n)
|
||||
write(fname,"('sc',i2.2,'.',a6)") n,file_ext
|
||||
tmp4(1:nx,1:ny,1:nz) = wrk(1:nx,1:ny,1:nz,3+n)
|
||||
call write_tmp4
|
||||
u_(1:nx,1:ny,1:nz) = wrk(1:nx,1:ny,1:nz,1)
|
||||
v_(1:nx,1:ny,1:nz) = wrk(1:nx,1:ny,1:nz,2)
|
||||
w_(1:nx,1:ny,1:nz) = wrk(1:nx,1:ny,1:nz,3)
|
||||
|
||||
end do
|
||||
end if
|
||||
|
||||
! LES quantities
|
||||
if (les) then
|
||||
! turbulent viscosity
|
||||
if (allocated(turb_visc)) then
|
||||
write(fname,"('nu_t.',a6)") file_ext
|
||||
tmp4 = turb_visc
|
||||
call write_tmp4
|
||||
end if
|
||||
|
||||
if (n_les > 0) then
|
||||
do n = 1, n_les
|
||||
call xFFT3d(-1,3+n_scalars+n)
|
||||
write(fname,"('les',i1,'.',a6)") n,file_ext
|
||||
tmp4(1:nx,1:ny,1:nz) = wrk(1:nx,1:ny,1:nz,3+n_scalars+n)
|
||||
call write_tmp4
|
||||
end do
|
||||
end if
|
||||
|
||||
end if
|
||||
|
||||
return
|
||||
end subroutine gather_4
|
||||
|
|
|
|||
|
|
@ -266,8 +266,10 @@ subroutine init_velocity
|
|||
do i = 1,nx+2
|
||||
|
||||
n_shell = nint(sqrt(real(akx(i)**2 + aky(k)**2 + akz(j)**2, 4)))
|
||||
if (n_shell .gt. 0 .and. n_shell .le. kmax .and. e_spec(n_shell) .gt. zip) then
|
||||
if (n_shell .gt. 0 .and. n_shell .le. kmax ) then
|
||||
if (e_spec(n_shell) .gt. zip) then
|
||||
fields(i,j,k,1:3) = fields(i,j,k,1:3) * sqrt(e_spec1(n_shell)/e_spec(n_shell))
|
||||
endif
|
||||
else
|
||||
fields(i,j,k,1:3) = zip
|
||||
end if
|
||||
|
|
|
|||
852
m_compact.f90
852
m_compact.f90
|
|
@ -1,14 +1,20 @@
|
|||
MODULE m_compact
|
||||
|
||||
use m_openmpi
|
||||
|
||||
IMPLICIT NONE
|
||||
PRIVATE
|
||||
REAL*8, DIMENSION(:), ALLOCATABLE :: lxf,lxs,wxf,wxs, &
|
||||
lyf,lys,wyf,wys, &
|
||||
lzf,lzs,wzf,wzs
|
||||
! lyzf,lyzs,wyzf,wyzs
|
||||
|
||||
REAL*8, DIMENSION(:), ALLOCATABLE :: zz1, zz2, vz1, vz2
|
||||
|
||||
INTEGER :: nxc,nyc,nzc
|
||||
REAL*8, PARAMETER :: ezero = 1.0e-14
|
||||
|
||||
PUBLIC :: ludcmp,dfnonp,d2fnonp,dfp,d2fp
|
||||
PUBLIC :: ludcmp,dfnonp,d2fnonp,dfp,d2fp,par_dfp,par_d2fp,pdfp,pd2fp
|
||||
|
||||
CONTAINS
|
||||
|
||||
|
|
@ -65,6 +71,22 @@
|
|||
call nonp_lud(3,nzc)
|
||||
ENDIF
|
||||
|
||||
ALLOCATE(vz1(nzc),STAT=ierr)
|
||||
IF(ierr /= 0) PRINT*, 'work array for lud allocation failed'
|
||||
|
||||
ALLOCATE(zz1(nzc),STAT=ierr)
|
||||
IF(ierr /= 0) PRINT*, 'work array for lud allocation failed'
|
||||
|
||||
ALLOCATE(vz2(nzc),STAT=ierr)
|
||||
IF(ierr /= 0) PRINT*, 'work array for lud allocation failed'
|
||||
|
||||
ALLOCATE(zz2(nzc),STAT=ierr)
|
||||
IF(ierr /= 0) PRINT*, 'work array for lud allocation failed'
|
||||
|
||||
call ctdlu (3.0, nzc, vz1, zz1)
|
||||
|
||||
call ctdlu (11.0/2.0, nzc, vz2, zz2)
|
||||
|
||||
! CALL x_lud
|
||||
! CALL yz_lud
|
||||
|
||||
|
|
@ -141,6 +163,280 @@
|
|||
l(n)=1./d
|
||||
END SUBROUTINE ptdlu
|
||||
|
||||
SUBROUTINE ctdlu(lambda,n,v,z)
|
||||
INTEGER, INTENT(IN) :: n
|
||||
REAL*8, INTENT(IN) :: lambda
|
||||
REAL*8, INTENT(OUT) :: v(n)
|
||||
REAL*8, INTENT(OUT) :: z(n)
|
||||
INTEGER :: i
|
||||
REAL*8 :: sigma, alpha
|
||||
|
||||
v(1)=0.0
|
||||
v(2)=lambda
|
||||
DO i=3,n
|
||||
v(i)=lambda - 1./v(i-1)
|
||||
ENDDO
|
||||
|
||||
if (lambda > 0.0) then
|
||||
alpha = (-lambda + sqrt(lambda**2 - 4.0))/2.0
|
||||
else
|
||||
alpha = (-lambda - sqrt(lambda**2 - 4.0))/2.0
|
||||
endif
|
||||
|
||||
sigma = (1.0 + alpha**2) / (lambda * (1.0 - alpha**2) * (1.0 - alpha**n))
|
||||
|
||||
DO i=1,n
|
||||
z(i) = sigma * (alpha**(i-1) + alpha**(n-i+1))
|
||||
ENDDO
|
||||
END SUBROUTINE ctdlu
|
||||
|
||||
|
||||
SUBROUTINE pdfp(x,xu,xl,dx,h,nd,n,nall,dir)
|
||||
|
||||
INTEGER,INTENT(IN) :: nall,n,nd,dir
|
||||
REAL*8,INTENT(IN) :: h
|
||||
REAL*8,INTENT(IN),DIMENSION(nd,n) :: x
|
||||
REAL*8,INTENT(OUT),DIMENSION(nd,2) :: xu, xl
|
||||
REAL*8,INTENT(OUT),DIMENSION(nd,n) :: dx
|
||||
|
||||
INTEGER :: i,j
|
||||
REAL*8 :: r1,r2,h1
|
||||
|
||||
integer (kind=MPI_INTEGER_KIND) :: idx
|
||||
integer (kind=MPI_INTEGER_KIND),dimension(4) :: requests
|
||||
logical :: recvLow, recvUpp
|
||||
|
||||
integer (kind=MPI_INTEGER_KIND), parameter :: rhstag = 0
|
||||
|
||||
h1=1./h
|
||||
r1=7./3.
|
||||
r2=1./12.
|
||||
|
||||
!if (myid.eq.0) write(*,*) "parallel dfp commuication"
|
||||
if (myid.eq.master) then
|
||||
|
||||
call MPI_SEND_INIT (x(1,n-1), 2*nd, MPI_REAL8, myid+1, rhstag, &
|
||||
MPI_COMM_TASK, requests(1), mpi_err)
|
||||
|
||||
call MPI_SEND_INIT (x(1,1), 2*nd, MPI_REAL8, numprocs-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(2), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xl, 2*nd, MPI_REAL8, numprocs-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(3), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xu, 2*nd, MPI_REAL8, myid+1, rhstag, &
|
||||
MPI_COMM_TASK, requests(4), mpi_err)
|
||||
|
||||
elseif (myid.eq.numprocs-1) then
|
||||
|
||||
call MPI_SEND_INIT (x(1,n-1), 2*nd, MPI_REAL8, 0, rhstag, &
|
||||
MPI_COMM_TASK, requests(1), mpi_err)
|
||||
|
||||
call MPI_SEND_INIT (x(1,1), 2*nd, MPI_REAL8, myid-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(2), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xl, 2*nd, MPI_REAL8, myid-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(3), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xu, 2*nd, MPI_REAL8, 0, rhstag, &
|
||||
MPI_COMM_TASK, requests(4), mpi_err)
|
||||
|
||||
else
|
||||
|
||||
call MPI_SEND_INIT (x(1,n-1), 2*nd, MPI_REAL8, myid+1, rhstag, &
|
||||
MPI_COMM_TASK, requests(1), mpi_err)
|
||||
|
||||
call MPI_SEND_INIT (x(1,1), 2*nd, MPI_REAL8, myid-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(2), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xl, 2*nd, MPI_REAL8, myid-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(3), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xu, 2*nd, MPI_REAL8, myid+1, rhstag, &
|
||||
MPI_COMM_TASK, requests(4), mpi_err)
|
||||
|
||||
endif
|
||||
|
||||
call MPI_STARTALL (4, requests, mpi_err)
|
||||
|
||||
DO i=3,n-2
|
||||
DO j=1,nd
|
||||
dx(j,i)=(r1*(x(j,i+1)-x(j,i-1))+r2*(x(j,i+2)-x(j,i-2)))*h1
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
recvLow = .true.
|
||||
recvUpp = .true.
|
||||
|
||||
do while (recvLow.or.recvUpp)
|
||||
|
||||
call MPI_WAITANY (2, requests(3:), idx, mpi_status, mpi_err)
|
||||
|
||||
if (idx.eq.1) then
|
||||
DO j=1,nd
|
||||
dx(j,1) =(r1*( x(j,2)-xl(j,2)) +r2*( x(j,3)-xl(j,1))) *h1
|
||||
ENDDO
|
||||
|
||||
DO j=1,nd
|
||||
dx(j,2) =(r1*( x(j,3)- x(j,1)) +r2*( x(j,4)-xl(j,2))) *h1
|
||||
ENDDO
|
||||
|
||||
recvLow = .false.
|
||||
endif
|
||||
|
||||
if (idx.eq.2) then
|
||||
DO j=1,nd
|
||||
dx(j,n-1)=(r1*( x(j,n)- x(j,n-2))+r2*(xu(j,1)- x(j,n-3)))*h1
|
||||
ENDDO
|
||||
|
||||
DO j=1,nd
|
||||
dx(j,n) =(r1*(xu(j,1)- x(j,n-1))+r2*(xu(j,2)- x(j,n-2)))*h1
|
||||
ENDDO
|
||||
|
||||
recvUpp = .false.
|
||||
endif
|
||||
enddo
|
||||
|
||||
IF (dir.eq.1) CALL ctdslv(dx,vz1,zz1,nd,n,nall) ! x-direction
|
||||
IF (dir.eq.2) CALL ctdslv(dx,vz1,zz1,nd,n,nall) ! x-direction
|
||||
IF (dir.eq.3) CALL ctdslv(dx,vz1,zz1,nd,n,nall) ! x-direction
|
||||
|
||||
END SUBROUTINE pdfp
|
||||
|
||||
|
||||
SUBROUTINE pd2fp(x,xu,xl,dx,h,nd,n,nall,dir)
|
||||
INTEGER,INTENT(IN) :: nall,n,nd,dir
|
||||
REAL*8,INTENT(IN) :: h
|
||||
REAL*8,INTENT(IN),DIMENSION(nd,n) :: x
|
||||
REAL*8,INTENT(OUT),DIMENSION(nd,n) :: dx
|
||||
INTEGER :: i,j
|
||||
REAL*8 :: h2,r1,r2,t1,t2
|
||||
|
||||
REAL*8,DIMENSION(nd,2) :: xu, xl
|
||||
|
||||
integer (kind=MPI_INTEGER_KIND) :: idx
|
||||
integer (kind=MPI_INTEGER_KIND),dimension(4) :: requests
|
||||
logical :: recvLow, recvUpp
|
||||
|
||||
integer (kind=MPI_INTEGER_KIND), parameter :: rhstag = 0
|
||||
|
||||
h2=1./(h*h)
|
||||
r1=6.
|
||||
r2=3./8.
|
||||
|
||||
!if (myid.eq.0) write(*,*) "parallel d2fp commuication"
|
||||
if (myid.eq.master) then
|
||||
|
||||
call MPI_SEND_INIT (x(1,n-1), 2*nd, MPI_REAL8, myid+1, rhstag, &
|
||||
MPI_COMM_TASK, requests(1), mpi_err)
|
||||
|
||||
call MPI_SEND_INIT (x(1,1), 2*nd, MPI_REAL8, numprocs-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(2), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xl, 2*nd, MPI_REAL8, numprocs-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(3), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xu, 2*nd, MPI_REAL8, myid+1, rhstag, &
|
||||
MPI_COMM_TASK, requests(4), mpi_err)
|
||||
|
||||
elseif (myid.eq.numprocs-1) then
|
||||
|
||||
call MPI_SEND_INIT (x(1,n-1), 2*nd, MPI_REAL8, 0, rhstag, &
|
||||
MPI_COMM_TASK, requests(1), mpi_err)
|
||||
|
||||
call MPI_SEND_INIT (x(1,1), 2*nd, MPI_REAL8, myid-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(2), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xl, 2*nd, MPI_REAL8, myid-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(3), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xu, 2*nd, MPI_REAL8, 0, rhstag, &
|
||||
MPI_COMM_TASK, requests(4), mpi_err)
|
||||
|
||||
else
|
||||
|
||||
call MPI_SEND_INIT (x(1,n-1), 2*nd, MPI_REAL8, myid+1, rhstag, &
|
||||
MPI_COMM_TASK, requests(1), mpi_err)
|
||||
|
||||
call MPI_SEND_INIT (x(1,1), 2*nd, MPI_REAL8, myid-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(2), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xl, 2*nd, MPI_REAL8, myid-1, rhstag, &
|
||||
MPI_COMM_TASK, requests(3), mpi_err)
|
||||
|
||||
call MPI_RECV_INIT (xu, 2*nd, MPI_REAL8, myid+1, rhstag, &
|
||||
MPI_COMM_TASK, requests(4), mpi_err)
|
||||
|
||||
endif
|
||||
|
||||
call MPI_STARTALL (4, requests, mpi_err)
|
||||
|
||||
DO i=3,n-2
|
||||
DO j=1,nd
|
||||
|
||||
t1 = (x(j,i+1)-2.*x(j,i)+x(j,i-1))
|
||||
t2 = (x(j,i+2)-2.*x(j,i)+x(j,i-2))
|
||||
dx(j,i)=(r1*t1+r2*t2)*h2
|
||||
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
recvLow = .true.
|
||||
recvUpp = .true.
|
||||
|
||||
do while (recvLow.or.recvUpp)
|
||||
|
||||
call MPI_WAITANY (2, requests(3:), idx, mpi_status, mpi_err)
|
||||
|
||||
if (idx.eq.1) then
|
||||
|
||||
DO j=1,nd
|
||||
|
||||
t1 = (x(j,2)-2.*x(j,1)+xl(j,2))
|
||||
t2 = (x(j,3)-2.*x(j,1)+xl(j,1))
|
||||
dx(j,1)=(r1*t1+r2*t2)*h2
|
||||
|
||||
ENDDO
|
||||
|
||||
DO j=1,nd
|
||||
|
||||
t1 = (x(j,3)-2.*x(j,2)+ x(j,1))
|
||||
t2 = (x(j,4)-2.*x(j,2)+xl(j,2))
|
||||
dx(j,2)=(r1*t1+r2*t2)*h2
|
||||
|
||||
ENDDO
|
||||
|
||||
recvLow = .false.
|
||||
endif
|
||||
|
||||
if (idx.eq.2) then
|
||||
|
||||
DO j=1,nd
|
||||
|
||||
t1 = ( x(j,n)-2.*x(j,n-1)+x(j,n-2))
|
||||
t2 = (xu(j,1)-2.*x(j,n-1)+x(j,n-3))
|
||||
dx(j,n-1)=(r1*t1+r2*t2)*h2
|
||||
|
||||
ENDDO
|
||||
|
||||
DO j=1,nd
|
||||
|
||||
t1 = (xu(j,1)-2.*x(j,n)+x(j,n-1))
|
||||
t2 = (xu(j,2)-2.*x(j,n)+x(j,n-2))
|
||||
dx(j,n)=(r1*t1+r2*t2)*h2
|
||||
|
||||
ENDDO
|
||||
|
||||
recvUpp = .false.
|
||||
endif
|
||||
enddo
|
||||
|
||||
IF (dir.eq.1) CALL ctdslv(dx,vz2,zz2,nd,n,nall) ! x-direction
|
||||
IF (dir.eq.2) CALL ctdslv(dx,vz2,zz2,nd,n,nall) ! y-direction
|
||||
IF (dir.eq.3) CALL ctdslv(dx,vz2,zz2,nd,n,nall) ! z-direction
|
||||
END SUBROUTINE pd2fp
|
||||
|
||||
|
||||
SUBROUTINE dfnonp(n,h,x,dx,nd,dir)
|
||||
INTEGER,INTENT(IN) :: n,nd,dir
|
||||
REAL*8,INTENT(IN) :: h
|
||||
|
|
@ -219,6 +515,85 @@
|
|||
|
||||
END SUBROUTINE dfp
|
||||
|
||||
SUBROUTINE par_dfp(x,xu,xl,dx,h,nd,n,nall,dir)
|
||||
INTEGER,INTENT(IN) :: nall,n,nd,dir
|
||||
REAL*8,INTENT(IN) :: h
|
||||
REAL*8,INTENT(IN),DIMENSION(nd,n) :: x
|
||||
REAL*8,INTENT(OUT),DIMENSION(nd,n) :: dx
|
||||
INTEGER :: i,j
|
||||
REAL*8 :: r1,r2,h1
|
||||
|
||||
REAL*8,DIMENSION(nd,2) :: xu, xl
|
||||
|
||||
h1=1./h
|
||||
r1=7./3.
|
||||
r2=1./12.
|
||||
|
||||
!if (myid.eq.0) write(*,*) "parallel dfp commuication"
|
||||
if (myid.eq.master) then
|
||||
|
||||
call MPI_ISEND (x(1,1), 2*nd, MPI_REAL8, numprocs-1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xu, 2*nd, MPI_REAL8, myid+1, myid+1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
call MPI_ISEND (x(1,n-1), 2*nd, MPI_REAL8, myid+1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xl, 2*nd, MPI_REAL8, numprocs-1, numprocs-1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
elseif (myid.eq.numprocs-1) then
|
||||
|
||||
call MPI_ISEND (x(1,1), 2*nd, MPI_REAL8, myid-1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xu, 2*nd, MPI_REAL8, 0, 0, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
call MPI_ISEND (x(1,n-1), 2*nd, MPI_REAL8, 0, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xl, 2*nd, MPI_REAL8, myid-1, myid-1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
else
|
||||
|
||||
call MPI_ISEND (x(1,1), 2*nd, MPI_REAL8, myid-1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xu, 2*nd, MPI_REAL8, myid+1, myid+1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
call MPI_ISEND (x(1,n-1), 2*nd, MPI_REAL8, myid+1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xl, 2*nd, MPI_REAL8, myid-1, myid-1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
endif
|
||||
|
||||
DO j=1,nd
|
||||
dx(j,n-1)=(r1*( x(j,n)- x(j,n-2))+r2*(xu(j,1)- x(j,n-3)))*h1
|
||||
dx(j,n)= (r1*(xu(j,1)- x(j,n-1))+r2*(xu(j,2)- x(j,n-2)))*h1
|
||||
dx(j,1)= (r1*( x(j,2)-xl(j,2)) +r2*( x(j,3)-xl(j,1)))*h1
|
||||
dx(j,2)= (r1*( x(j,3)- x(j,1)) +r2*( x(j,4)-xl(j,2)))*h1
|
||||
ENDDO
|
||||
|
||||
DO i=3,n-2
|
||||
DO j=1,nd
|
||||
dx(j,i)=(r1*(x(j,i+1)-x(j,i-1))+r2*(x(j,i+2)-x(j,i-2)))*h1
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
|
||||
IF (dir.eq.1) CALL par_ptdslv(dx,lxf,wxf,nd,n,nall) ! x-direction
|
||||
IF (dir.eq.2) CALL par_ptdslv(dx,lyf,wyf,nd,n,nall) ! x-direction
|
||||
IF (dir.eq.3) CALL par_ptdslv(dx,lzf,wzf,nd,n,nall) ! x-direction
|
||||
|
||||
END SUBROUTINE par_dfp
|
||||
|
||||
SUBROUTINE ptdslv(r,n,l,w,nd)
|
||||
INTEGER,INTENT(IN) :: n,nd
|
||||
REAL*8,INTENT(INOUT),DIMENSION(nd,n) :: r
|
||||
|
|
@ -247,6 +622,362 @@
|
|||
ENDDO
|
||||
END SUBROUTINE ptdslv
|
||||
|
||||
SUBROUTINE ctdslv(r,va,za,nd,n,nall)
|
||||
INTEGER,PARAMETER :: nb = 256
|
||||
INTEGER,PARAMETER :: nb2 = 64
|
||||
INTEGER,INTENT(IN) :: n,nd,nall
|
||||
REAL*8,INTENT(INOUT),DIMENSION(nd,n) :: r
|
||||
REAL*8,INTENT(IN),DIMENSION(nall) :: va
|
||||
REAL*8,INTENT(IN),DIMENSION(nall) :: za
|
||||
REAL*8,DIMENSION(n) :: v,z
|
||||
INTEGER i,j
|
||||
INTEGER ii,jj
|
||||
|
||||
REAL*8, DIMENSION(nd) :: x1, x0
|
||||
|
||||
INTEGER npart, nbase, nlow, nupp
|
||||
INTEGER :: pid, np
|
||||
REAL*8, DIMENSION(nb) :: r0, r1
|
||||
|
||||
pid = myid
|
||||
np = numprocs
|
||||
|
||||
npart = nall / np
|
||||
nbase = pid * npart
|
||||
nlow = pid * npart + 1
|
||||
nupp = (pid + 1) * npart
|
||||
|
||||
v = va(nlow:nupp)
|
||||
z = za(nlow:nupp)
|
||||
|
||||
if (npart.lt.4) then
|
||||
! assertion fail
|
||||
endif
|
||||
|
||||
x1 = 0.0
|
||||
x0 = 0.0
|
||||
|
||||
do i=1,n
|
||||
x0 = x0 + z(i)*r(:,i)
|
||||
enddo
|
||||
|
||||
call MPI_REDUCE(x0, x1, nd, MPI_REAL8, MPI_SUM, master, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
! first process
|
||||
if (myid.eq.0) then
|
||||
|
||||
CALL MPI_SSEND(x1, nd, MPI_REAL8, np-1, 100, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
r(:,1) = x1
|
||||
r(:,2) = r(:,2) - x1
|
||||
|
||||
! last process
|
||||
elseif (pid.eq.(np-1)) then
|
||||
|
||||
CALL MPI_RECV(x1, nd, MPI_REAL8, master, 100, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
r(:,n) = r(:,n) - x1
|
||||
|
||||
! intermediate process
|
||||
else
|
||||
|
||||
endif
|
||||
|
||||
! first process
|
||||
if (myid.eq.0) then
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,2)=r(j,2)/v(2)
|
||||
ENDDO
|
||||
|
||||
DO i=3,n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=(r(j,i)-r(j,i-1))/v(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
r0 = r(jj:jj+nb-1,n)
|
||||
CALL MPI_SEND(r0, nb, MPI_REAL8, pid+1, jj, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
ENDDO
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
CALL MPI_RECV(r1, nb, MPI_REAL8, pid+1, nd+jj, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
i=n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i) = r(j,i) - r1(j-jj+1)/v(i)
|
||||
ENDDO
|
||||
|
||||
DO i=n-1,2,-1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i) = r(j,i) - r(j,i+1)/v(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
ENDDO
|
||||
|
||||
! last process
|
||||
elseif (pid.eq.(np-1)) then
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
CALL MPI_RECV(r1, nb, MPI_REAL8, pid-1, jj, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
i=1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=(r(j,i) - r1(j-jj+1))/v(i)
|
||||
ENDDO
|
||||
|
||||
DO i=2,n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=(r(j,i) - r(j,i-1))/v(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
ENDDO
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
DO i=n-1,1,-1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-r(j,i+1)/v(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
r0 = r(jj:jj+nb-1,1)
|
||||
CALL MPI_SEND(r0, nb, MPI_REAL8, pid-1, nd+jj, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
ENDDO
|
||||
|
||||
! intermediate process
|
||||
else
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
CALL MPI_RECV(r1, nb, MPI_REAL8, pid-1, jj, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
i=1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=(r(j,i)-r1(j-jj+1))/v(i)
|
||||
ENDDO
|
||||
|
||||
DO i=2,n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=(r(j,i)-r(j,i-1))/v(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
r0 = r(jj:jj+nb-1,n)
|
||||
CALL MPI_SEND(r0, nb, MPI_REAL8, pid+1, jj, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
ENDDO
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
CALL MPI_RECV(r1, nb, MPI_REAL8, pid+1, nd+jj, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
i=n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-r1(j-jj+1)/v(i)
|
||||
ENDDO
|
||||
|
||||
DO i=n-1,1,-1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-r(j,i+1)/v(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
r0 = r(jj:jj+nb-1,1)
|
||||
CALL MPI_SEND(r0, nb, MPI_REAL8, pid-1, nd+jj, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
ENDDO
|
||||
|
||||
endif
|
||||
|
||||
END SUBROUTINE ctdslv
|
||||
|
||||
|
||||
SUBROUTINE par_ptdslv(r,la,wa,nd,n,nall)
|
||||
INTEGER,PARAMETER :: nb = 64
|
||||
INTEGER,INTENT(IN) :: n,nd,nall
|
||||
REAL*8,INTENT(INOUT),DIMENSION(nd,n) :: r
|
||||
REAL*8,INTENT(IN),DIMENSION(nall) :: la,wa
|
||||
REAL*8,DIMENSION(n) :: l,w
|
||||
INTEGER i,j
|
||||
INTEGER ii,jj
|
||||
REAL*8, DIMENSION(nd) :: sum
|
||||
|
||||
INTEGER npart, nbase, nlow, nupp
|
||||
INTEGER :: pid, np
|
||||
REAL*8, DIMENSION(nb) :: r0, r1, sum0
|
||||
REAL*8, DIMENSION(nb,2) :: buf
|
||||
|
||||
pid = myid
|
||||
np = numprocs
|
||||
|
||||
npart = nall / np
|
||||
nbase = pid * npart
|
||||
nlow = pid * npart + 1
|
||||
nupp = (pid + 1) * npart
|
||||
|
||||
l = la(nlow:nupp)
|
||||
w = wa(nlow:nupp)
|
||||
|
||||
if (npart.lt.4) then
|
||||
! assertion fail
|
||||
endif
|
||||
|
||||
call MPI_BARRIER(MPI_COMM_WORLD, mpi_err)
|
||||
|
||||
! first process
|
||||
if (myid.eq.0) then
|
||||
|
||||
DO jj=1,nd,nb
|
||||
DO j=jj,jj+nb-1
|
||||
sum(j)=w(1)*r(j,1)
|
||||
r(j,1)=r(j,1)*l(1)
|
||||
ENDDO
|
||||
|
||||
DO i=2,n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-r(j,i-1)
|
||||
sum(j)=sum(j)+w(i)*r(j,i)
|
||||
r(j,i)=r(j,i)*l(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
buf(:,1) = r(jj:jj+nb-1,n)
|
||||
buf(:,2) = sum(jj:jj+nb-1)
|
||||
CALL MPI_SEND(buf, 2*nb, MPI_REAL8, pid+1, 1, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
ENDDO
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
CALL MPI_RECV(buf, 2*nb, MPI_REAL8, pid+1, 2, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
r0 = buf(:,1)
|
||||
r1 = buf(:,2)
|
||||
|
||||
i=n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-l(i)*r0(j-jj+1)-w(i)*r1(j-jj+1)
|
||||
ENDDO
|
||||
DO i=n-1,1,-1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-l(i)*r(j,i+1)-w(i)*r1(j-jj+1)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
ENDDO
|
||||
|
||||
! last process
|
||||
elseif (pid.eq.(np-1)) then
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
CALL MPI_RECV(buf, 2*nb, MPI_REAL8, pid-1, 1, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
r0 = buf(:,1)
|
||||
sum0 = buf(:,2)
|
||||
|
||||
i=1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-r0(j-jj+1)
|
||||
sum(j)=sum0(j-jj+1)+w(i)*r(j,i)
|
||||
r(j,i)=r(j,i)*l(i)
|
||||
ENDDO
|
||||
|
||||
DO i=2,n-1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-r(j,i-1)
|
||||
sum(j)=sum(j)+w(i)*r(j,i)
|
||||
r(j,i)=r(j,i)*l(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,n)=l(n)*(r(j,n)-sum(j))
|
||||
ENDDO
|
||||
|
||||
ENDDO
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,n-1)=r(j,n-1)-w(n-1)*r(j,n)
|
||||
ENDDO
|
||||
|
||||
DO i=n-2,1,-1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-l(i)*r(j,i+1)-w(i)*r(j,n)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
buf(:,1) = r(jj:jj+nb-1,1)
|
||||
buf(:,2) = r(jj:jj+nb-1,n)
|
||||
CALL MPI_SEND(buf, 2*nb, MPI_REAL8, pid-1, 2, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
ENDDO
|
||||
|
||||
! intermediate process
|
||||
else
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
CALL MPI_RECV(buf, 2*nb, MPI_REAL8, pid-1, 1, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
r0 = buf(:,1)
|
||||
sum0 = buf(:,2)
|
||||
|
||||
i=1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-r0(j-jj+1)
|
||||
sum(j)=sum0(j-jj+1)+w(i)*r(j,i)
|
||||
r(j,i)=r(j,i)*l(i)
|
||||
ENDDO
|
||||
|
||||
DO i=2,n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-r(j,i-1)
|
||||
sum(j)=sum(j)+w(i)*r(j,i)
|
||||
r(j,i)=r(j,i)*l(i)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
buf(:,1) = r(jj:jj+nb-1,n)
|
||||
buf(:,2) = sum(jj:jj+nb-1)
|
||||
CALL MPI_SEND(buf, 2*nb, MPI_REAL8, pid+1, 1, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
ENDDO
|
||||
|
||||
DO jj=1,nd,nb
|
||||
|
||||
CALL MPI_RECV(buf, 2*nb, MPI_REAL8, pid+1, 2, MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
r0 = buf(:,1)
|
||||
r1 = buf(:,2)
|
||||
|
||||
i=n
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-l(i)*r0(j-jj+1)-w(i)*r1(j-jj+1)
|
||||
ENDDO
|
||||
DO i=n-1,1,-1
|
||||
DO j=jj,jj+nb-1
|
||||
r(j,i)=r(j,i)-l(i)*r(j,i+1)-w(i)*r1(j-jj+1)
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
buf(:,1) = r(jj:jj+nb-1,1)
|
||||
!buf(:,2) = r1
|
||||
CALL MPI_SEND(buf, 2*nb, MPI_REAL8, pid-1, 2, MPI_COMM_TASK, mpi_err)
|
||||
|
||||
ENDDO
|
||||
|
||||
endif
|
||||
|
||||
END SUBROUTINE par_ptdslv
|
||||
|
||||
SUBROUTINE d2fp(n,h,x,dx,nd,dir)
|
||||
INTEGER,INTENT(IN) :: n,nd,dir
|
||||
REAL*8,INTENT(IN) :: h
|
||||
|
|
@ -260,30 +991,22 @@
|
|||
DO j=1,nd
|
||||
t1 = (x(j,n)-2.*x(j,n-1)+x(j,n-2))
|
||||
t2 = (x(j,1)-2.*x(j,n-1)+x(j,n-3))
|
||||
IF (x(j,n).eq.x(j,n-1).and.x(j,n-1).eq.x(j,n-2)) t1=0.
|
||||
IF (x(j,1).eq.x(j,n-1).and.x(j,n-1).eq.x(j,n-3)) t2=0.
|
||||
dx(j,n-1)=(r1*t1+r2*t2)
|
||||
|
||||
t1 = (x(j,1)-2.*x(j,n)+x(j,n-1))
|
||||
t2 = (x(j,2)-2.*x(j,n)+x(j,n-2))
|
||||
IF (x(j,1).eq.x(j,n).and.x(j,n).eq.x(j,n-1)) t1=0.
|
||||
IF (x(j,2).eq.x(j,n).and.x(j,n).eq.x(j,n-2)) t2=0.
|
||||
! dx(j,n)=(r1*(x(j,1)-2.*x(j,n)+x(j,n-1)) &
|
||||
! +r2*(x(j,2)-2.*x(j,n)+x(j,n-2)))
|
||||
dx(j,n)=(r1*t1+r2*t2)
|
||||
|
||||
t1 = (x(j,2)-2.*x(j,1)+x(j,n))
|
||||
t2 = (x(j,3)-2.*x(j,1)+x(j,n-1))
|
||||
IF (x(j,2).eq.x(j,1).and.x(j,1).eq.x(j,n)) t1=0.
|
||||
IF (x(j,3).eq.x(j,1).and.x(j,1).eq.x(j,n-1)) t2=0.
|
||||
! dx(j,1)=(r1*(x(j,2)-2.*x(j,1)+x(j,n)) &
|
||||
! +r2*(x(j,3)-2.*x(j,1)+x(j,n-1)))
|
||||
dx(j,1)=(r1*t1+r2*t2)
|
||||
|
||||
t1 = (x(j,3)-2.*x(j,2)+x(j,1))
|
||||
t2 = (x(j,4)-2.*x(j,2)+x(j,n))
|
||||
IF (x(j,3).eq.x(j,2).and.x(j,2).eq.x(j,1)) t1=0.
|
||||
IF (x(j,4).eq.x(j,2).and.x(j,2).eq.x(j,n)) t2=0.
|
||||
! dx(j,2)=(r1*(x(j,3)-2.*x(j,2)+x(j,1)) &
|
||||
! +r2*(x(j,4)-2.*x(j,2)+x(j,n)))
|
||||
dx(j,2)=(r1*t1+r2*t2)
|
||||
|
|
@ -297,8 +1020,6 @@
|
|||
DO j=1,nd
|
||||
t1 = (x(j,i+1)-2.*x(j,i)+x(j,i-1))
|
||||
t2 = (x(j,i+2)-2.*x(j,i)+x(j,i-2))
|
||||
IF (x(j,i+1).eq.x(j,i).and.x(j,i).eq.x(j,i-1)) t1=0.
|
||||
IF (x(j,i+2).eq.x(j,i).and.x(j,i).eq.x(j,i-2)) t2=0.
|
||||
! dx(j,i)=(r1*(x(j,i+1)-2.*x(j,i)+x(j,i-1)) &
|
||||
! +r2*(x(j,i+2)-2.*x(j,i)+x(j,i-2)))
|
||||
dx(j,i)=(r1*t1+r2*t2)
|
||||
|
|
@ -310,6 +1031,109 @@
|
|||
IF (dir.eq.3) CALL ptdslv(dx,n,lzs,wzs,nd) ! z-direction
|
||||
END SUBROUTINE d2fp
|
||||
|
||||
SUBROUTINE par_d2fp(x,xu,xl,dx,h,nd,n,nall,dir)
|
||||
INTEGER,INTENT(IN) :: nall,n,nd,dir
|
||||
REAL*8,INTENT(IN) :: h
|
||||
REAL*8,INTENT(IN),DIMENSION(nd,n) :: x
|
||||
REAL*8,INTENT(OUT),DIMENSION(nd,n) :: dx
|
||||
INTEGER :: i,j
|
||||
REAL*8 :: h2,r1,r2,t1,t2
|
||||
|
||||
REAL*8,DIMENSION(nd,2) :: xu, xl
|
||||
|
||||
h2=1./(h*h)
|
||||
r1=6.
|
||||
r2=3./8.
|
||||
|
||||
!if (myid.eq.0) write(*,*) "parallel d2fp commuication"
|
||||
if (myid.eq.master) then
|
||||
|
||||
call MPI_ISEND (x(1,1), 2*nd, MPI_REAL8, numprocs-1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xu, 2*nd, MPI_REAL8, myid+1, myid+1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
call MPI_ISEND (x(1,n-1), 2*nd, MPI_REAL8, myid+1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xl, 2*nd, MPI_REAL8, numprocs-1, numprocs-1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
elseif (myid.eq.numprocs-1) then
|
||||
|
||||
call MPI_ISEND (x(1,1), 2*nd, MPI_REAL8, myid-1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xu, 2*nd, MPI_REAL8, 0, 0, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
call MPI_ISEND (x(1,n-1), 2*nd, MPI_REAL8, 0, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xl, 2*nd, MPI_REAL8, myid-1, myid-1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
else
|
||||
|
||||
call MPI_ISEND (x(1,1), 2*nd, MPI_REAL8, myid-1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xu, 2*nd, MPI_REAL8, myid+1, myid+1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
call MPI_ISEND (x(1,n-1), 2*nd, MPI_REAL8, myid+1, myid, &
|
||||
MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
|
||||
call MPI_RECV (xl, 2*nd, MPI_REAL8, myid-1, myid-1, &
|
||||
MPI_COMM_TASK, mpi_status, mpi_err)
|
||||
|
||||
endif
|
||||
|
||||
|
||||
DO j=1,nd
|
||||
|
||||
t1 = ( x(j,n)-2.*x(j,n-1)+x(j,n-2))
|
||||
t2 = (xu(j,1)-2.*x(j,n-1)+x(j,n-3))
|
||||
dx(j,n-1)=(r1*t1+r2*t2)*h2
|
||||
|
||||
t1 = (xu(j,1)-2.*x(j,n)+x(j,n-1))
|
||||
t2 = (xu(j,2)-2.*x(j,n)+x(j,n-2))
|
||||
! dx(j,n)=(r1*(x(j,1)-2.*x(j,n)+x(j,n-1)) &
|
||||
! +r2*(x(j,2)-2.*x(j,n)+x(j,n-2)))
|
||||
dx(j,n)=(r1*t1+r2*t2)*h2
|
||||
|
||||
t1 = (x(j,2)-2.*x(j,1)+xl(j,2))
|
||||
t2 = (x(j,3)-2.*x(j,1)+xl(j,1))
|
||||
! dx(j,1)=(r1*(x(j,2)-2.*x(j,1)+x(j,n)) &
|
||||
! +r2*(x(j,3)-2.*x(j,1)+x(j,n-1)))
|
||||
dx(j,1)=(r1*t1+r2*t2)*h2
|
||||
|
||||
t1 = (x(j,3)-2.*x(j,2)+ x(j,1))
|
||||
t2 = (x(j,4)-2.*x(j,2)+xl(j,2))
|
||||
! dx(j,2)=(r1*(x(j,3)-2.*x(j,2)+x(j,1)) &
|
||||
! +r2*(x(j,4)-2.*x(j,2)+x(j,n)))
|
||||
dx(j,2)=(r1*t1+r2*t2)*h2
|
||||
|
||||
ENDDO
|
||||
|
||||
DO i=3,n-2
|
||||
DO j=1,nd
|
||||
|
||||
t1 = (x(j,i+1)-2.*x(j,i)+x(j,i-1))
|
||||
t2 = (x(j,i+2)-2.*x(j,i)+x(j,i-2))
|
||||
! dx(j,i)=(r1*(x(j,i+1)-2.*x(j,i)+x(j,i-1)) &
|
||||
! +r2*(x(j,i+2)-2.*x(j,i)+x(j,i-2)))
|
||||
dx(j,i)=(r1*t1+r2*t2)*h2
|
||||
|
||||
ENDDO
|
||||
ENDDO
|
||||
|
||||
IF (dir.eq.1) CALL par_ptdslv(dx,lxs,wxs,nd,n,nall) ! x-direction
|
||||
IF (dir.eq.2) CALL par_ptdslv(dx,lys,wys,nd,n,nall) ! y-direction
|
||||
IF (dir.eq.3) CALL par_ptdslv(dx,lzs,wzs,nd,n,nall) ! z-direction
|
||||
END SUBROUTINE par_d2fp
|
||||
|
||||
SUBROUTINE tdslv(r,n,l,nd)
|
||||
INTEGER,INTENT(IN) :: n,nd
|
||||
REAL*8,INTENT(INOUT),DIMENSION(nd,n) :: r
|
||||
|
|
@ -354,10 +1178,6 @@
|
|||
dx(j,2)=(x(j,3)-2.*x(j,2)+x(j,1))
|
||||
dx(j,n-1)=(x(j,n)-2.*x(j,n-1)+x(j,n-2))
|
||||
dx(j,n)=(a*x(j,n)+b*x(j,n-1)+c*x(j,n-2)+e*x(j,n-3))
|
||||
IF (x(j,1).eq.x(j,2).and.x(j,2).eq.x(j,3).and.x(j,3).eq.x(j,4)) dx(j,1)=0.
|
||||
IF (x(j,3).eq.x(j,2).and.x(j,2).eq.x(j,1)) dx(j,2)=0.
|
||||
IF (x(j,n).eq.x(j,n-1).and.x(j,n-1).eq.x(j,n-2).and.x(j,n-2).eq.x(j,n-3)) dx(j,n)=0.
|
||||
IF (x(j,n).eq.x(j,n-1).and.x(j,n-1).eq.x(j,n-2)) dx(j,n-1)=0.
|
||||
dx(j,1)=dx(j,1)*h2
|
||||
dx(j,2)=dx(j,2)*h2*r3
|
||||
dx(j,n-1)=dx(j,n-1)*h2*r3
|
||||
|
|
@ -367,8 +1187,6 @@
|
|||
DO j=1,nd
|
||||
t1 = (x(j,i+1)-2.*x(j,i)+x(j,i-1))
|
||||
t2 = (x(j,i+2)-2.*x(j,i)+x(j,i-2))
|
||||
IF (x(j,i+1).eq.x(j,i).and.x(j,i).eq.x(j,i-1)) t1=0.
|
||||
IF (x(j,i+2).eq.x(j,i).and.x(j,i).eq.x(j,i-2)) t2=0.
|
||||
|
||||
! dx(j,i)=(r1*(x(j,i+1)-2.*x(j,i)+x(j,i-1)) &
|
||||
! +r2*(x(j,i+2)-2.*x(j,i)+x(j,i-2)))
|
||||
|
|
|
|||
1061
m_fdm_calc.f90
1061
m_fdm_calc.f90
File diff suppressed because it is too large
Load diff
|
|
@ -64,7 +64,8 @@ contains
|
|||
|
||||
|
||||
! initializing MPI environment
|
||||
call MPI_INIT_THREAD(MPI_THREAD_SERIALIZED, mpi_provide, mpi_err)
|
||||
!call MPI_INIT_THREAD(MPI_THREAD_SERIALIZED, mpi_provide, mpi_err)
|
||||
call MPI_INIT(mpi_err)
|
||||
call MPI_Comm_size(MPI_COMM_WORLD,numprocs_world,mpi_err)
|
||||
call MPI_Comm_rank(MPI_COMM_WORLD,myid_world,mpi_err)
|
||||
|
||||
|
|
|
|||
|
|
@ -64,7 +64,8 @@ contains
|
|||
|
||||
|
||||
! initializing MPI environment
|
||||
call MPI_INIT_THREAD(MPI_THREAD_SERIALIZED, mpi_provide, mpi_err)
|
||||
!call MPI_INIT_THREAD(MPI_THREAD_SERIALIZED, mpi_provide, mpi_err)
|
||||
call MPI_INIT(mpi_err)
|
||||
call MPI_Comm_size(MPI_COMM_WORLD,numprocs_world,mpi_err)
|
||||
call MPI_Comm_rank(MPI_COMM_WORLD,myid_world,mpi_err)
|
||||
|
||||
|
|
|
|||
|
|
@ -114,6 +114,7 @@ module m_parameters
|
|||
real*8 :: ac,bc,c_cut,c_ref,min_wr,prof_wr,hx,hy,diff,pre
|
||||
real*8 :: dt_fullsave,dt_fdmsave,t_fullsave,t_fdmsave
|
||||
real*8 :: fdmcfl,startad,dummyu_,schmidt
|
||||
real*8 :: dm_ratio
|
||||
real*8 :: bgn_save
|
||||
|
||||
! files for statistics
|
||||
|
|
@ -581,6 +582,9 @@ contains
|
|||
read(in,*,ERR=9000,END=9000) fdmcfl
|
||||
write(out,'(a30,3x,e20.10)') 'FDM_CFL =',fdmcfl
|
||||
|
||||
read(in,*,ERR=9000,END=9000) dm_ratio
|
||||
write(out,'(a30,3x,e20.10)') 'Dm_Ratio =',dm_ratio
|
||||
|
||||
! initialize ----------------------
|
||||
t_fdmsave=dt_fdmsave
|
||||
t_fullsave=dt_fullsave
|
||||
|
|
|
|||
|
|
@ -19,8 +19,8 @@ contains
|
|||
! make the default job runlimit to be 6 months
|
||||
job_runlimit = 6 * 30 * 24 * 60
|
||||
|
||||
! make the default job runlimit to be 12 hours
|
||||
job_runlimit = 12 * 60
|
||||
! make the default job runlimit to be 12 hours -> one month
|
||||
job_runlimit = 12 * 60 * 30 * 2
|
||||
|
||||
write(out,'(a30,3x,i20)') 'job_runlimit (default):',job_runlimit
|
||||
|
||||
|
|
|
|||
41
main.f90
41
main.f90
|
|
@ -67,10 +67,7 @@ program x_code
|
|||
! Initializing the LES stuff
|
||||
if (les) call m_les_begin
|
||||
|
||||
if (task.eq.'hydro' .and. myid.eq.master) then
|
||||
call prepare_fdm
|
||||
end if
|
||||
call MPI_BARRIER(MPI_COMM_TASK, mpi_err)
|
||||
call prepare_fdm
|
||||
|
||||
call stat_main
|
||||
|
||||
|
|
@ -132,22 +129,10 @@ program x_code
|
|||
end if
|
||||
end if
|
||||
|
||||
if (fdm_sw.ne.0) then ! Reacting
|
||||
! reacting scalars
|
||||
if (fdm_sw.ne.0) then
|
||||
call gather_4
|
||||
if (myid.eq.master) then
|
||||
call fdm_exe
|
||||
do id_to=1,numprocs-1
|
||||
call MPI_Isend (dummy, 1, MPI_REAL, id_to, 0, MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
end do
|
||||
else
|
||||
call MPI_Irecv (dummy, 1, MPI_REAL, master, 0, MPI_COMM_TASK, mpi_request, mpi_err)
|
||||
do
|
||||
call MPI_Test(mpi_request, flag, mpi_status, mpi_err)
|
||||
if (flag) exit
|
||||
call SLEEPQQ (200)
|
||||
end do
|
||||
end if
|
||||
call MPI_BARRIER(MPI_COMM_TASK, mpi_err)
|
||||
call fdm_exe
|
||||
endif
|
||||
|
||||
! RHS for scalars
|
||||
|
|
@ -202,18 +187,18 @@ program x_code
|
|||
! write the restart file if it's the time
|
||||
if (mod(itime,IPRINT2).eq.0) then
|
||||
call restart_write_parallel
|
||||
!remove hit restart file
|
||||
call system ('rm -f '//run_name//'.64.'//ex_file_ext)
|
||||
if (fdm_sw.ne.0) then ! Reacting
|
||||
call gather_4
|
||||
if (myid.eq.master) then
|
||||
!write itime, savenum pair
|
||||
if(TIME.ge.bgn_save) call fdm_restart_write
|
||||
write (restart_pair, *) file_ext, fullsavenum-1
|
||||
call write_tp_field
|
||||
if (TIME.ge.bgn_save) then
|
||||
call fdm_restart_write
|
||||
if (myid.eq.master) then
|
||||
write (restart_pair, *) file_ext, fullsavenum-1, '\n'
|
||||
!call write_tp_field
|
||||
end if
|
||||
end if
|
||||
call MPI_BARRIER(MPI_COMM_TASK, mpi_err)
|
||||
endif
|
||||
!remove HIT3D restart file except the latest
|
||||
call system ('rm -f '//run_name//'.64.'//ex_file_ext)
|
||||
ex_file_ext = file_ext
|
||||
endif
|
||||
|
||||
|
|
@ -266,7 +251,7 @@ program x_code
|
|||
if (task_split) call fields_to_stats
|
||||
if (mod(itime,iprint1).eq.0) call stat_main
|
||||
if (fdm_sw.eq.0) then ! non-reacting
|
||||
if (mod(itime,iprint2).eq.0) call io_write_4
|
||||
!if (mod(itime,iprint2).eq.0) call io_write_4
|
||||
endif
|
||||
end if
|
||||
end if stats
|
||||
|
|
|
|||
|
|
@ -31,6 +31,7 @@ subroutine write_tp_field
|
|||
do i=1,nx,spx
|
||||
write(tp_field,'(6e18.9)') dx*real(i),dy*real(j),dz*real(k),u_(i,j,k),v_(i,j,k),w_(i,j,k) ! u',v',w'
|
||||
enddo
|
||||
call flush(tp_field)
|
||||
enddo
|
||||
enddo
|
||||
|
||||
|
|
@ -49,13 +50,12 @@ subroutine write_tp_field
|
|||
(1.-y1(2,i,j,ii)/y1(1,i,j,ii)),u_(i,j,ii), &
|
||||
v_(i,j,ii),w_(i,j,ii)
|
||||
enddo
|
||||
call flush(tp_field)
|
||||
enddo
|
||||
enddo
|
||||
|
||||
endif
|
||||
|
||||
call flush(tp_field)
|
||||
|
||||
897 format('ZONE T="time= ',f10.5,'" I= ',i4,' J= ',i4,' K= ',i4,' F=POINT')
|
||||
|
||||
end subroutine write_tp_field
|
||||
|
|
|
|||
|
|
@ -298,7 +298,7 @@ CONTAINS
|
|||
! Definition of the array ialias.
|
||||
! The array ialias is just the number of wavenumbers at (i,j,k) that have
|
||||
! their magnitude higher than nx/3. This is needed in dealiasing procedures.
|
||||
rnx3 = real(nx/3, 8)
|
||||
rnx3 = real(2*nx, 8) / real(3*lx, 8)
|
||||
do k = 1,nz
|
||||
if (abs(aky(k)) .gt. rnx3) ialias(:,:,k) = 1
|
||||
do j = 1,ny
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue