This repository has been archived by the owner on Nov 7, 2019. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
io_write_one.f90
82 lines (75 loc) · 2.74 KB
/
io_write_one.f90
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
!=======================================================================
! This is part of the 2DECOMP&FFT library
!
! 2DECOMP&FFT is a software framework for general-purpose 2D (pencil)
! decomposition. It also implements a highly scalable distributed
! three-dimensional Fast Fourier Transform (FFT).
!
! Copyright (C) 2009-2012 Ning Li, the Numerical Algorithms Group (NAG)
!
!=======================================================================
! This file contain common code to be included by subroutines
! 'mpiio_write_one_...' in io.f90
! Using MPI-IO to write a distributed 3D array into a file
if (present(opt_decomp)) then
decomp = opt_decomp
else
call get_decomp_info(decomp)
end if
! determine subarray parameters
sizes(1) = decomp%xsz(1)
sizes(2) = decomp%ysz(2)
sizes(3) = decomp%zsz(3)
if (ipencil == 1) then
subsizes(1) = decomp%xsz(1)
subsizes(2) = decomp%xsz(2)
subsizes(3) = decomp%xsz(3)
starts(1) = decomp%xst(1)-1 ! 0-based index
starts(2) = decomp%xst(2)-1
starts(3) = decomp%xst(3)-1
else if (ipencil == 2) then
subsizes(1) = decomp%ysz(1)
subsizes(2) = decomp%ysz(2)
subsizes(3) = decomp%ysz(3)
starts(1) = decomp%yst(1)-1
starts(2) = decomp%yst(2)-1
starts(3) = decomp%yst(3)-1
else if (ipencil == 3) then
subsizes(1) = decomp%zsz(1)
subsizes(2) = decomp%zsz(2)
subsizes(3) = decomp%zsz(3)
starts(1) = decomp%zst(1)-1
starts(2) = decomp%zst(2)-1
starts(3) = decomp%zst(3)-1
endif
#ifdef T3PIO
call MPI_INFO_CREATE(info, ierror)
gs = ceiling(real(sizes(1),mytype)*real(sizes(2),mytype)* &
real(sizes(3),mytype)/1024./1024.)
call t3pio_set_info(MPI_COMM_WORLD, info, "./", ierror, &
GLOBAL_SIZE=gs, factor=1)
#endif
call MPI_TYPE_CREATE_SUBARRAY(3, sizes, subsizes, starts, &
MPI_ORDER_FORTRAN, data_type, newtype, ierror)
call MPI_TYPE_COMMIT(newtype,ierror)
#ifdef T3PIO
call MPI_FILE_OPEN(MPI_COMM_WORLD, filename, &
MPI_MODE_CREATE+MPI_MODE_WRONLY, info, fh, ierror)
#else
call MPI_FILE_OPEN(MPI_COMM_WORLD, filename, &
MPI_MODE_CREATE+MPI_MODE_WRONLY, MPI_INFO_NULL, &
fh, ierror)
#endif
filesize = 0_MPI_OFFSET_KIND
call MPI_FILE_SET_SIZE(fh,filesize,ierror) ! guarantee overwriting
disp = 0_MPI_OFFSET_KIND
call MPI_FILE_SET_VIEW(fh,disp,data_type, &
newtype,'native',MPI_INFO_NULL,ierror)
call MPI_FILE_WRITE_ALL(fh, var, &
subsizes(1)*subsizes(2)*subsizes(3), &
data_type, MPI_STATUS_IGNORE, ierror)
call MPI_FILE_CLOSE(fh,ierror)
call MPI_TYPE_FREE(newtype,ierror)
#ifdef T3PIO
call MPI_INFO_FREE(info,ierror)
#endif