#include "private.h"
#include "xbt.h"
-extern int xargc;
-extern char** xargv;
-
static xbt_dict_t comm_lookup = NULL;
static xbt_dict_t group_lookup = NULL;
static xbt_dict_t request_lookup = NULL;
static xbt_dict_t datatype_lookup = NULL;
static xbt_dict_t op_lookup = NULL;
+static xbt_dict_t win_lookup = NULL;
static int running_processes = 0;
+#if defined(__alpha__) || defined(__sparc64__) || defined(__x86_64__) || defined(__ia64__)
+typedef int integer;
+typedef unsigned int uinteger;
+#else
+typedef long int integer;
+typedef unsigned long int uinteger;
+#endif
+
/* Bindings for MPI special values */
-union u_smpi_common {
+
struct s_smpi_common {
integer _MPI_IN_PLACE;
integer _MPI_BOTTOM;
integer _MPI_STATUS_IGNORE;
integer _MPI_STATUSES_IGNORE;
- } *f77; /* with f2c, remains NULL with gfortran */
- struct s_smpi_common f90; /* with gftortran */
-} smpi_ = { NULL };
+ } smpi_;
+
+
/* Convert between Fortran and C */
-#define F2C_ADDR(addr, val) \
- (((void *)(addr) == (void *)(smpi_.f77 \
- ? &smpi_.f77[smpi_current_rank]._ ## val \
- : &smpi_.f90._ ## val)) \
+#define FORT_ADDR(addr, val) \
+ (((void *)(addr) == (void*) &(smpi_._ ## val)) \
? (val) : (void *)(addr))
-#define F2C_BOTTOM(addr) F2C_ADDR(addr, MPI_BOTTOM)
-#define F2C_IN_PLACE(addr) F2C_ADDR(addr, MPI_IN_PLACE)
-#define F2C_STATUS_IGNORE(addr) F2C_ADDR(addr, MPI_STATUS_IGNORE)
-#define F2C_STATUSES_IGNORE(addr) F2C_ADDR(addr, MPI_STATUSES_IGNORE)
+#define FORT_BOTTOM(addr) FORT_ADDR(addr, MPI_BOTTOM)
+#define FORT_IN_PLACE(addr) FORT_ADDR(addr, MPI_IN_PLACE)
+#define FORT_STATUS_IGNORE(addr) FORT_ADDR(addr, MPI_STATUS_IGNORE)
+#define FORT_STATUSES_IGNORE(addr) FORT_ADDR(addr, MPI_STATUSES_IGNORE)
#define KEY_SIZE (sizeof(int) * 2 + 1)
xbt_dict_remove(op_lookup, get_key(key, op));
}
+static int new_win(MPI_Win win) {
+ static int win_id = 0;
+ char key[KEY_SIZE];
+ xbt_dict_set(win_lookup, get_key(key, win_id), win, NULL);
+ win_id++;
+ return win_id-1;
+}
+
+static MPI_Win get_win(int win) {
+ char key[KEY_SIZE];
+ return win >= 0
+ ? (MPI_Win)xbt_dict_get_or_null(win_lookup, get_key(key, win))
+ : MPI_WIN_NULL;
+}
+
+static void free_win(int win) {
+ char key[KEY_SIZE];
+ xbt_dict_remove(win_lookup, get_key(key, win));
+}
+
+
void mpi_init_(int* ierr) {
if(!comm_lookup){
comm_lookup = xbt_dict_new_homogeneous(NULL);
new_comm(MPI_COMM_WORLD);
group_lookup = xbt_dict_new_homogeneous(NULL);
-
request_lookup = xbt_dict_new_homogeneous(NULL);
-
datatype_lookup = xbt_dict_new_homogeneous(NULL);
+ win_lookup = xbt_dict_new_homogeneous(NULL);
new_datatype(MPI_BYTE);
new_datatype(MPI_CHAR);
+ #if defined(__alpha__) || defined(__sparc64__) || defined(__x86_64__) || defined(__ia64__)
new_datatype(MPI_INT);
new_datatype(MPI_INT);
+ #else
+ new_datatype(MPI_LONG);
+ new_datatype(MPI_LONG);
+ #endif
new_datatype(MPI_INT8_T);
new_datatype(MPI_INT16_T);
new_datatype(MPI_INT32_T);
new_datatype(MPI_DOUBLE);
new_datatype(MPI_C_FLOAT_COMPLEX);
new_datatype(MPI_C_DOUBLE_COMPLEX);
+ #if defined(__alpha__) || defined(__sparc64__) || defined(__x86_64__) || defined(__ia64__)
new_datatype(MPI_2INT);
+ #else
+ new_datatype(MPI_2LONG);
+ #endif
new_datatype(MPI_UINT8_T);
new_datatype(MPI_UINT16_T);
new_datatype(MPI_UINT32_T);
new_op(MPI_BOR);
new_op(MPI_BXOR);
}
- /* smpif2c is responsible for generating a call with the final arguments */
*ierr = MPI_Init(NULL, NULL);
running_processes++;
}
xbt_dict_free(&request_lookup);
xbt_dict_free(&group_lookup);
xbt_dict_free(&comm_lookup);
+ xbt_dict_free(&win_lookup);
}
}
void mpi_isend_(void *buf, int* count, int* datatype, int* dst,
int* tag, int* comm, int* request, int* ierr) {
MPI_Request req;
- buf = (char *) F2C_BOTTOM(buf);
+ buf = (char *) FORT_BOTTOM(buf);
*ierr = MPI_Isend(buf, *count, get_datatype(*datatype), *dst, *tag,
get_comm(*comm), &req);
if(*ierr == MPI_SUCCESS) {
void mpi_irsend_(void *buf, int* count, int* datatype, int* dst,
int* tag, int* comm, int* request, int* ierr) {
MPI_Request req;
- buf = (char *) F2C_BOTTOM(buf);
+ buf = (char *) FORT_BOTTOM(buf);
*ierr = MPI_Irsend(buf, *count, get_datatype(*datatype), *dst, *tag,
get_comm(*comm), &req);
if(*ierr == MPI_SUCCESS) {
int* comm, MPI_Status* status, int* ierr) {
*ierr = MPI_Sendrecv(sendbuf, *sendcount, get_datatype(*sendtype), *dst,
*sendtag, recvbuf, *recvcount,get_datatype(*recvtype), *src, *recvtag,
- get_comm(*comm), F2C_STATUS_IGNORE(status));
+ get_comm(*comm), FORT_STATUS_IGNORE(status));
}
void mpi_recv_init_(void *buf, int* count, int* datatype, int* src, int* tag,
void mpi_irecv_(void *buf, int* count, int* datatype, int* src, int* tag,
int* comm, int* request, int* ierr) {
MPI_Request req;
- buf = (char *) F2C_BOTTOM(buf);
+ buf = (char *) FORT_BOTTOM(buf);
*ierr = MPI_Irecv(buf, *count, get_datatype(*datatype), *src, *tag,
get_comm(*comm), &req);
if(*ierr == MPI_SUCCESS) {
void mpi_wait_(int* request, MPI_Status* status, int* ierr) {
MPI_Request req = find_request(*request);
- *ierr = MPI_Wait(&req, F2C_STATUS_IGNORE(status));
+ *ierr = MPI_Wait(&req, FORT_STATUS_IGNORE(status));
if(req==MPI_REQUEST_NULL){
free_request(*request);
*request=MPI_FORTRAN_REQUEST_NULL;
for(i = 0; i < *count; i++) {
reqs[i] = find_request(requests[i]);
}
- *ierr = MPI_Waitall(*count, reqs, F2C_STATUSES_IGNORE(status));
+ *ierr = MPI_Waitall(*count, reqs, FORT_STATUSES_IGNORE(status));
for(i = 0; i < *count; i++) {
if(reqs[i]==MPI_REQUEST_NULL){
free_request(requests[i]);
void mpi_reduce_(void* sendbuf, void* recvbuf, int* count,
int* datatype, int* op, int* root, int* comm, int* ierr) {
- sendbuf = (char *) F2C_IN_PLACE(sendbuf);
- sendbuf = (char *) F2C_BOTTOM(sendbuf);
- recvbuf = (char *) F2C_BOTTOM(recvbuf);
+ sendbuf = (char *) FORT_IN_PLACE(sendbuf);
+ sendbuf = (char *) FORT_BOTTOM(sendbuf);
+ recvbuf = (char *) FORT_BOTTOM(recvbuf);
*ierr = MPI_Reduce(sendbuf, recvbuf, *count,
get_datatype(*datatype), get_op(*op), *root, get_comm(*comm));
}
void mpi_allreduce_(void* sendbuf, void* recvbuf, int* count, int* datatype,
int* op, int* comm, int* ierr) {
- sendbuf = (char *) F2C_IN_PLACE(sendbuf);
+ sendbuf = (char *) FORT_IN_PLACE(sendbuf);
*ierr = MPI_Allreduce(sendbuf, recvbuf, *count, get_datatype(*datatype),
get_op(*op), get_comm(*comm));
}
void mpi_reduce_scatter_(void* sendbuf, void* recvbuf, int* recvcounts, int* datatype,
int* op, int* comm, int* ierr) {
- sendbuf = (char *) F2C_IN_PLACE(sendbuf);
+ sendbuf = (char *) FORT_IN_PLACE(sendbuf);
*ierr = MPI_Reduce_scatter(sendbuf, recvbuf, recvcounts, get_datatype(*datatype),
get_op(*op), get_comm(*comm));
}
void mpi_scatter_(void* sendbuf, int* sendcount, int* sendtype,
void* recvbuf, int* recvcount, int* recvtype,
int* root, int* comm, int* ierr) {
- recvbuf = (char *) F2C_IN_PLACE(recvbuf);
+ recvbuf = (char *) FORT_IN_PLACE(recvbuf);
*ierr = MPI_Scatter(sendbuf, *sendcount, get_datatype(*sendtype),
recvbuf, *recvcount, get_datatype(*recvtype), *root, get_comm(*comm));
}
void mpi_scatterv_(void* sendbuf, int* sendcounts, int* displs, int* sendtype,
void* recvbuf, int* recvcount, int* recvtype,
int* root, int* comm, int* ierr) {
- recvbuf = (char *) F2C_IN_PLACE(recvbuf);
+ recvbuf = (char *) FORT_IN_PLACE(recvbuf);
*ierr = MPI_Scatterv(sendbuf, sendcounts, displs, get_datatype(*sendtype),
recvbuf, *recvcount, get_datatype(*recvtype), *root, get_comm(*comm));
}
void mpi_gather_(void* sendbuf, int* sendcount, int* sendtype,
void* recvbuf, int* recvcount, int* recvtype,
int* root, int* comm, int* ierr) {
- sendbuf = (char *) F2C_IN_PLACE(sendbuf);
- sendbuf = (char *) F2C_BOTTOM(sendbuf);
- recvbuf = (char *) F2C_BOTTOM(recvbuf);
+ sendbuf = (char *) FORT_IN_PLACE(sendbuf);
+ sendbuf = (char *) FORT_BOTTOM(sendbuf);
+ recvbuf = (char *) FORT_BOTTOM(recvbuf);
*ierr = MPI_Gather(sendbuf, *sendcount, get_datatype(*sendtype),
recvbuf, *recvcount, get_datatype(*recvtype), *root, get_comm(*comm));
}
void mpi_gatherv_(void* sendbuf, int* sendcount, int* sendtype,
void* recvbuf, int* recvcounts, int* displs, int* recvtype,
int* root, int* comm, int* ierr) {
- sendbuf = (char *) F2C_IN_PLACE(sendbuf);
- sendbuf = (char *) F2C_BOTTOM(sendbuf);
- recvbuf = (char *) F2C_BOTTOM(recvbuf);
+ sendbuf = (char *) FORT_IN_PLACE(sendbuf);
+ sendbuf = (char *) FORT_BOTTOM(sendbuf);
+ recvbuf = (char *) FORT_BOTTOM(recvbuf);
*ierr = MPI_Gatherv(sendbuf, *sendcount, get_datatype(*sendtype),
recvbuf, recvcounts, displs, get_datatype(*recvtype), *root, get_comm(*comm));
}
void mpi_allgather_(void* sendbuf, int* sendcount, int* sendtype,
void* recvbuf, int* recvcount, int* recvtype,
int* comm, int* ierr) {
- sendbuf = (char *) F2C_IN_PLACE(sendbuf);
+ sendbuf = (char *) FORT_IN_PLACE(sendbuf);
*ierr = MPI_Allgather(sendbuf, *sendcount, get_datatype(*sendtype),
recvbuf, *recvcount, get_datatype(*recvtype), get_comm(*comm));
}
void mpi_allgatherv_(void* sendbuf, int* sendcount, int* sendtype,
void* recvbuf, int* recvcounts,int* displs, int* recvtype,
int* comm, int* ierr) {
- sendbuf = (char *) F2C_IN_PLACE(sendbuf);
+ sendbuf = (char *) FORT_IN_PLACE(sendbuf);
*ierr = MPI_Allgatherv(sendbuf, *sendcount, get_datatype(*sendtype),
recvbuf, recvcounts, displs, get_datatype(*recvtype), get_comm(*comm));
}
void mpi_test_ (int * request, int *flag, MPI_Status * status, int* ierr){
MPI_Request req = find_request(*request);
- *ierr= MPI_Test(&req, flag, F2C_STATUS_IGNORE(status));
+ *ierr= MPI_Test(&req, flag, FORT_STATUS_IGNORE(status));
if(req==MPI_REQUEST_NULL){
free_request(*request);
*request=MPI_FORTRAN_REQUEST_NULL;
for(i = 0; i < *count; i++) {
reqs[i] = find_request(requests[i]);
}
- *ierr= MPI_Testall(*count, reqs, flag, F2C_STATUSES_IGNORE(statuses));
+ *ierr= MPI_Testall(*count, reqs, flag, FORT_STATUSES_IGNORE(statuses));
for(i = 0; i < *count; i++) {
if(reqs[i]==MPI_REQUEST_NULL){
free_request(requests[i]);
}
void mpi_get_count_(MPI_Status * status, int* datatype, int *count, int* ierr){
- *ierr = MPI_Get_count(F2C_STATUS_IGNORE(status), get_datatype(*datatype), count);
+ *ierr = MPI_Get_count(FORT_STATUS_IGNORE(status), get_datatype(*datatype), count);
}
void mpi_attr_get_(int* comm, int* keyval, void* attr_value, int* flag, int* ierr ){
}
void mpi_win_fence_( int* assert, int* win, int* ierr){
- *ierr = MPI_Win_fence(* assert, *(MPI_Win*)win);
+ *ierr = MPI_Win_fence(* assert, get_win(*win));
}
void mpi_win_free_( int* win, int* ierr){
- *ierr = MPI_Win_free( (MPI_Win*)win);
+ MPI_Win tmp = get_win(*win);
+ *ierr = MPI_Win_free(&tmp);
+ if(*ierr == MPI_SUCCESS) {
+ free_win(*win);
+ }
}
void mpi_win_create_( int *base, MPI_Aint* size, int* disp_unit, int* info, int* comm, int *win, int* ierr){
- *ierr = MPI_Win_create( (void*)base, *size, *disp_unit, *(MPI_Info*)info, get_comm(*comm),(MPI_Win*)win);
+ MPI_Win tmp;
+ *ierr = MPI_Win_create( (void*)base, *size, *disp_unit, *(MPI_Info*)info, get_comm(*comm),&tmp);
+ if(*ierr == MPI_SUCCESS) {
+ *win = new_win(tmp);
+ }
+}
+
+void mpi_win_set_name_ (int* win, char * name, int* ierr, int size){
+ char* tname = xbt_malloc((size+1)*sizeof(char));
+ strncpy(tname, name, size);
+ tname[size]='\0';
+ *ierr = MPI_Win_set_name(get_win(*win), tname);
+ xbt_free(tname);
+}
+
+void mpi_win_get_name_ (int* win, char * name, int* len, int* ierr){
+ *ierr = MPI_Win_get_name(get_win(*win),name,len);
+ if(*len>0) name[*len]=' ';//blank padding, not \0
}
void mpi_info_create_( int *info, int* ierr){
void mpi_get_( int *origin_addr, int* origin_count, int* origin_datatype, int *target_rank,
MPI_Aint* target_disp, int *target_count, int* target_datatype, int* win, int* ierr){
*ierr = MPI_Get( (void*)origin_addr,*origin_count, get_datatype(*origin_datatype),*target_rank,
- *target_disp, *target_count,get_datatype(*target_datatype), *(MPI_Win *)win);
+ *target_disp, *target_count,get_datatype(*target_datatype), get_win(*win));
+}
+
+void mpi_accumulate_( int *origin_addr, int* origin_count, int* origin_datatype, int *target_rank,
+ MPI_Aint* target_disp, int *target_count, int* target_datatype, int* op, int* win, int* ierr){
+ *ierr = MPI_Accumulate( (void*)origin_addr,*origin_count, get_datatype(*origin_datatype),*target_rank,
+ *target_disp, *target_count,get_datatype(*target_datatype), get_op(*op), get_win(*win));
+}
+
+void mpi_put_( int *origin_addr, int* origin_count, int* origin_datatype, int *target_rank,
+ MPI_Aint* target_disp, int *target_count, int* target_datatype, int* win, int* ierr){
+ *ierr = MPI_Put( (void*)origin_addr,*origin_count, get_datatype(*origin_datatype),*target_rank,
+ *target_disp, *target_count,get_datatype(*target_datatype), get_win(*win));
}
new_op(MPI_BOR);
new_op(MPI_BXOR);
}
- /* smpif2c is responsible for generating a call with the final arguments */
+
*ierr = MPI_Init_thread(NULL, NULL,*required, provided);
}
}
}
-void mpi_type_set_name_ (int* datatype, char * name, int* ierr){
-
- *ierr = MPI_Type_set_name(get_datatype(*datatype), name);
+void mpi_type_set_name_ (int* datatype, char * name, int* ierr, int size){
+ char* tname = xbt_malloc((size+1)*sizeof(char));
+ strncpy(tname, name, size);
+ tname[size]='\0';
+ *ierr = MPI_Type_set_name(get_datatype(*datatype), tname);
+ xbt_free(tname);
}
void mpi_type_get_name_ (int* datatype, char * name, int* len, int* ierr){
-
*ierr = MPI_Type_get_name(get_datatype(*datatype),name,len);
+ if(*len>0) name[*len]=' ';
}
void mpi_type_get_attr_ (int* type, int* type_keyval, void *attribute_val, int* flag, int* ierr){
}
void mpi_comm_get_name_ (int* comm, char* name, int* len, int* ierr){
-
*ierr = MPI_Comm_get_name(get_comm(*comm), name, len);
+ if(*len>0) name[*len]=' ';
}
void mpi_comm_compare_ (int* comm1, int* comm2, int *result, int* ierr){
{
*ierr = MPI_Sendrecv_replace(buf, *count, get_datatype(*datatype), *dst, *sendtag, *src,
- *recvtag, get_comm(*comm), F2C_STATUS_IGNORE(status));
+ *recvtag, get_comm(*comm), FORT_STATUS_IGNORE(status));
}
void mpi_testany_ (int* count, int* requests, int *index, int *flag, MPI_Status* status, int* ierr)
for(i = 0; i < *count; i++) {
reqs[i] = find_request(requests[i]);
}
- *ierr = MPI_Testany(*count, reqs, index, flag, F2C_STATUS_IGNORE(status));
+ *ierr = MPI_Testany(*count, reqs, index, flag, FORT_STATUS_IGNORE(status));
if(*index!=MPI_UNDEFINED)
if(reqs[*index]==MPI_REQUEST_NULL){
free_request(requests[*index]);
void mpi_reduce_scatter_block_ (void *sendbuf, void *recvbuf, int* recvcount, int* datatype, int* op, int* comm, int* ierr)
{
- sendbuf = (char *) F2C_IN_PLACE(sendbuf);
+ sendbuf = (char *) FORT_IN_PLACE(sendbuf);
*ierr = MPI_Reduce_scatter_block(sendbuf, recvbuf, *recvcount, get_datatype(*datatype), get_op(*op), get_comm(*comm));
}
reqs[i] = find_request(requests[i]);
indices[i]=0;
}
- *ierr = MPI_Testsome(*incount, reqs, outcount, indices, F2C_STATUSES_IGNORE(statuses));
+ *ierr = MPI_Testsome(*incount, reqs, outcount, indices, FORT_STATUSES_IGNORE(statuses));
for(i=0;i<*incount;i++){
if(indices[i]){
if(reqs[indices[i]]==MPI_REQUEST_NULL){
}
void mpi_probe_ (int* source, int* tag, int* comm, MPI_Status* status, int* ierr) {
- *ierr = MPI_Probe(*source, *tag, get_comm(*comm), F2C_STATUS_IGNORE(status));
+ *ierr = MPI_Probe(*source, *tag, get_comm(*comm), FORT_STATUS_IGNORE(status));
}
void mpi_attr_delete_ (int* comm, int* keyval, int* ierr) {
*ierr = MPI_Exscan(sendbuf, recvbuf, *count, get_datatype(*datatype), get_op(*op), get_comm(*comm));
}
-void mpi_comm_set_name_ (int* comm, char* name, int* ierr){
- *ierr = MPI_Comm_set_name (get_comm(*comm), name);
+void mpi_comm_set_name_ (int* comm, char* name, int* ierr, int size){
+ char* tname = xbt_malloc((size+1)*sizeof(char));
+ strncpy(tname, name, size);
+ tname[size]='\0';
+ *ierr = MPI_Comm_set_name (get_comm(*comm), tname);
+ xbt_free(tname);
}
void mpi_comm_dup_with_info_ (int* comm, int* info, int* newcomm, int* ierr){