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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
#include <mpi.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <assert.h>
#include "mpitest.h"
//#define NUM_X 536870911
#define NUM_X 536870912
#define NUM_Y 1
//#define BIGDT 2147483643
#define BIGDT 2147483647
/*
static char MTEST_Descrip[] = "Nonblocking file read/write for bigtype";
*/
int main(int argc, char **argv)
{
MPI_File fh;
int i, j;
size_t k;
MPI_Datatype inner_type, rem_type, mem_type;
MPI_Datatype int_type, file_type;
int *buf_write, *buf_read;
int rc;
MPI_Aint disp[2];
int block_len[2];
MPI_Datatype type[2];
MPI_Status status;
MPI_Request request;
MTest_Init(&argc, &argv);
if (sizeof(MPI_Aint) <= sizeof(int)) {
/* can't test on this platform... */
goto exit;
}
k = 0;
/* create a large buffer 2 */
buf_write = malloc(NUM_X * NUM_Y * sizeof(int));
buf_read = malloc(NUM_X * NUM_Y * sizeof(int));
if (buf_write == NULL || buf_read == NULL) {
fprintf(stderr, "Not enough memory\n");
exit(1);
}
memset(buf_read, 0, NUM_X * NUM_Y * sizeof(int));
for (i = 0; i < NUM_X; i++) {
for (j = 0; j < NUM_Y; j++) {
buf_write[k] = k;
k++;
}
}
/* Big Datatype (2^31 - 1 bytes) */
MPI_Type_contiguous(BIGDT, MPI_BYTE, &inner_type);
/* Small Datatype (1 byte) */
MPI_Type_contiguous(1, MPI_BYTE, &rem_type);
type[0] = inner_type;
type[1] = rem_type;
block_len[0] = 1;
block_len[1] = 1;
disp[0] = 0;
disp[1] = BIGDT;
/* combine both types */
MPI_Type_create_struct(2, block_len, disp, type, &mem_type);
MPI_Type_commit(&mem_type);
MPI_Type_free(&rem_type);
MPI_Type_free(&inner_type);
MPI_Type_contiguous(4, MPI_BYTE, &int_type);
{
/* This creates a big type that is actually contituous, touching an
* optimization that was at one point buggy */
MPI_Type_vector(1, NUM_X, 1, int_type, &file_type);
}
MPI_Type_commit(&file_type);
MPI_Type_free(&int_type);
rc = MPI_File_open(MPI_COMM_WORLD, "testfile",
MPI_MODE_RDWR | MPI_MODE_CREATE, MPI_INFO_NULL, &fh);
if (rc != MPI_SUCCESS) {
fprintf(stderr, "Can't open file: %s\n", "testfile");
exit(1);
}
rc = MPI_File_set_view(fh, 2144, MPI_BYTE, file_type, "native", MPI_INFO_NULL);
if (rc != MPI_SUCCESS) {
fprintf(stderr, "ERROR SET VIEW\n");
exit(1);
}
/* write everything */
rc = MPI_File_iwrite_at_all(fh, 0, buf_write, 1, mem_type, &request);
if (rc != MPI_SUCCESS) {
fprintf(stderr, "%d ERROR IWRITE AT ALL\n", rc);
exit(1);
}
MPI_Wait(&request, &status);
rc = MPI_File_set_view(fh, 2144, MPI_BYTE, file_type, "native", MPI_INFO_NULL);
if (rc != MPI_SUCCESS) {
fprintf(stderr, "ERROR SET VIEW\n");
exit(1);
}
/* read everything */
rc = MPI_File_iread_at_all(fh, 0, buf_read, 1, mem_type, &request);
if (rc != MPI_SUCCESS) {
fprintf(stderr, "%d ERROR IREAD AT ALL\n", rc);
exit(1);
}
MPI_Wait(&request, &status);
for (k = 0; k < NUM_X * NUM_Y; k++) {
if (buf_read[k] != buf_write[k]) {
fprintf(stderr, "Verfiy Failed index %zu: expected %d found %d\n",
k, buf_write[k], buf_read[k]);
assert(0);
}
}
free(buf_write);
free(buf_read);
MPI_File_close(&fh);
MPI_Type_free(&mem_type);
MPI_Type_free(&file_type);
exit:
MTest_Finalize(0);
return 0;
}
|
