write_ordb.c 4.03 KB
Newer Older
1
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
/* 
 *
 *   Copyright (C) 1997 University of Chicago. 
 *   See COPYRIGHT notice in top-level directory.
 */

#include "mpioimpl.h"

#ifdef HAVE_WEAK_SYMBOLS

#if defined(HAVE_PRAGMA_WEAK)
#pragma weak MPI_File_write_ordered_begin = PMPI_File_write_ordered_begin
#elif defined(HAVE_PRAGMA_HP_SEC_DEF)
#pragma _HP_SECONDARY_DEF PMPI_File_write_ordered_begin MPI_File_write_ordered_begin
#elif defined(HAVE_PRAGMA_CRI_DUP)
#pragma _CRI duplicate MPI_File_write_ordered_begin as PMPI_File_write_ordered_begin
/* end of weak pragmas */
19
20
21
#elif defined(HAVE_WEAK_ATTRIBUTE)
int MPI_File_write_ordered_begin(MPI_File fh, const void *buf, int count, MPI_Datatype datatype)
    __attribute__((weak,alias("PMPI_File_write_ordered_begin")));
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#endif

/* Include mapping from MPI->PMPI */
#define MPIO_BUILD_PROFILING
#include "mpioprof.h"
#endif

/*@
    MPI_File_write_ordered_begin - Begin a split collective write using shared file pointer

Input Parameters:
. fh - file handle (handle)
. count - number of elements in buffer (nonnegative integer)
. datatype - datatype of each buffer element (handle)

Output Parameters:
. buf - initial address of buffer (choice)

.N fortran
@*/
42
int MPI_File_write_ordered_begin(MPI_File fh, ROMIO_CONST void *buf, int count,
43
44
				 MPI_Datatype datatype)
{
45
46
    int error_code, nprocs, myrank;
    ADIO_Offset incr;
47
    MPI_Count datatype_size;
48
49
50
    int source, dest;
    static char myname[] = "MPI_FILE_WRITE_ORDERED_BEGIN";
    ADIO_Offset shared_fp;
51
    ADIO_File adio_fh;
52
53
    void *e32buf = NULL;
    const void *xbuf=NULL;
54

55
    MPIU_THREAD_CS_ENTER(ALLFUNC,);
56

57
    adio_fh = MPIO_File_resolve(fh);
58
59

    /* --BEGIN ERROR HANDLING-- */
60
61
62
    MPIO_CHECK_FILE_HANDLE(adio_fh, myname, error_code);
    MPIO_CHECK_COUNT(adio_fh, count, myname, error_code);
    MPIO_CHECK_DATATYPE(adio_fh, datatype, myname, error_code);
63

64
    if (adio_fh->split_coll_count)
65
66
67
68
    {
	error_code = MPIO_Err_create_code(MPI_SUCCESS, MPIR_ERR_RECOVERABLE,
					  myname, __LINE__, MPI_ERR_IO, 
					  "**iosplitcoll", 0);
69
	error_code = MPIO_Err_return_file(adio_fh, error_code);
70
71
72
73
	goto fn_exit;
    }
    /* --END ERROR HANDLING-- */

74
    adio_fh->split_coll_count = 1;
75

76
    MPI_Type_size_x(datatype, &datatype_size);
77
    /* --BEGIN ERROR HANDLING-- */
78
79
80
    MPIO_CHECK_INTEGRAL_ETYPE(adio_fh, count, datatype_size, myname, error_code);
    MPIO_CHECK_FS_SUPPORTS_SHARED(adio_fh, myname, error_code);
    MPIO_CHECK_COUNT_SIZE(adio_fh, count, datatype_size, myname, error_code);
81
82
    /* --END ERROR HANDLING-- */

83
    ADIOI_TEST_DEFERRED(adio_fh, myname, &error_code);
84

85
86
    MPI_Comm_size(adio_fh->comm, &nprocs);
    MPI_Comm_rank(adio_fh->comm, &myrank);
87

88
    incr = (count*datatype_size)/adio_fh->etype_size;
89
90
91
92
93
    /* Use a message as a 'token' to order the operations */
    source = myrank - 1;
    dest   = myrank + 1;
    if (source < 0) source = MPI_PROC_NULL;
    if (dest >= nprocs) dest = MPI_PROC_NULL;
94
    MPI_Recv(NULL, 0, MPI_BYTE, source, 0, adio_fh->comm, MPI_STATUS_IGNORE);
95

96
    ADIO_Get_shared_fp(adio_fh, incr, &shared_fp, &error_code);
97
98
99
100
101
102
    /* --BEGIN ERROR HANDLING-- */
    if (error_code != MPI_SUCCESS)
    {
	error_code = MPIO_Err_create_code(MPI_SUCCESS, MPIR_ERR_FATAL,
					  myname, __LINE__, MPI_ERR_INTERN, 
					  "**iosharedfailed", 0);
103
	error_code = MPIO_Err_return_file(adio_fh, error_code);
104
105
106
107
	goto fn_exit;
    }
    /* --END ERROR HANDLING-- */

108
    MPI_Send(NULL, 0, MPI_BYTE, dest, 0, adio_fh->comm);
109

110
    xbuf = buf;
111
    if (adio_fh->is_external32) {
112
	error_code = MPIU_external32_buffer_setup(buf, count, datatype, &e32buf);
113
114
115
116
117
118
	if (error_code != MPI_SUCCESS) 
	    goto fn_exit;

	xbuf = e32buf;
    }

119
120
    ADIO_WriteStridedColl(adio_fh, xbuf, count, datatype, ADIO_EXPLICIT_OFFSET,
			  shared_fp, &adio_fh->split_status, &error_code);
121
122
123

    /* --BEGIN ERROR HANDLING-- */
    if (error_code != MPI_SUCCESS)
124
	error_code = MPIO_Err_return_file(adio_fh, error_code);
125
126
127
    /* --END ERROR HANDLING-- */

fn_exit:
128
    MPIU_THREAD_CS_EXIT(ALLFUNC,);
129
130
131
132

    /* FIXME: Check for error code from WriteStridedColl? */
    return error_code;
}