Actual source code: mpi.c

  1: /*
  2:       This provides a few of the MPI-uni functions that cannot be implemented
  3:     with C macros
  4: */
 5:  #include mpiuni/mpi.h
  7: #error "Wrong mpi.h included! require mpi.h from MPIUNI"
  8: #endif
  9: #if !defined(PETSC_STDCALL)
 10: #define PETSC_STDCALL
 11: #endif
 12: #include <stdio.h>
 13: #if defined(PETSC_HAVE_STDLIB_H)
 14: #include <stdlib.h>
 15: #endif

 17: #define MPI_SUCCESS 0
 18: #define MPI_FAILURE 1
 19: void    *MPIUNI_TMP        = 0;
 20: int     MPIUNI_DATASIZE[10] = {sizeof(int),sizeof(float),sizeof(double),2*sizeof(double),sizeof(char),2*sizeof(int),4*sizeof(double),4,8,2*sizeof(double)};
 21: /*
 22:        With MPI Uni there is only one communicator, which is called 1.
 23: */
 24: #define MAX_ATTR 128

 26: typedef struct {
 27:   void                *extra_state;
 28:   void                *attribute_val;
 29:   int                 active;
 30:   MPI_Delete_function *del;
 31: } MPI_Attr;

 33: static MPI_Attr attr[MAX_ATTR];
 34: static int      num_attr = 1,mpi_tag_ub = 100000000;

 36: #if defined(__cplusplus)
 38: #endif

 40: /* 
 41:    To avoid problems with prototypes to the system memcpy() it is duplicated here
 42: */
 43: int MPIUNI_Memcpy(void *a,const void* b,int n) {
 44:   int  i;
 45:   char *aa= (char*)a;
 46:   char *bb= (char*)b;

 48:   for (i=0; i<n; i++) aa[i] = bb[i];
 49:   return 0;
 50: }

 52: /*
 53:    Used to set the built-in MPI_TAG_UB attribute
 54: */
 55: static int Keyval_setup(void)
 56: {
 57:   attr[0].active        = 1;
 58:   attr[0].attribute_val = &mpi_tag_ub;
 59:   return 0;
 60: }

 62: int MPI_Keyval_create(MPI_Copy_function *copy_fn,MPI_Delete_function *delete_fn,int *keyval,void *extra_state)
 63: {
 64:   if (num_attr >= MAX_ATTR) MPI_Abort(MPI_COMM_WORLD,1);

 66:   attr[num_attr].extra_state = extra_state;
 67:   attr[num_attr].del         = delete_fn;
 68:   *keyval                    = num_attr++;
 69:   return 0;
 70: }

 72: int MPI_Keyval_free(int *keyval)
 73: {
 74:   attr[*keyval].active = 0;
 75:   return MPI_SUCCESS;
 76: }

 78: int MPI_Attr_put(MPI_Comm comm,int keyval,void *attribute_val)
 79: {
 80:   attr[keyval].active        = 1;
 81:   attr[keyval].attribute_val = attribute_val;
 82:   return MPI_SUCCESS;
 83: }
 84: 
 85: int MPI_Attr_delete(MPI_Comm comm,int keyval)
 86: {
 87:   if (attr[keyval].active && attr[keyval].del) {
 88:     (*(attr[keyval].del))(comm,keyval,attr[keyval].attribute_val,attr[keyval].extra_state);
 89:   }
 90:   attr[keyval].active        = 0;
 91:   attr[keyval].attribute_val = 0;
 92:   return MPI_SUCCESS;
 93: }

 95: int MPI_Attr_get(MPI_Comm comm,int keyval,void *attribute_val,int *flag)
 96: {
 97:   if (!keyval) Keyval_setup();
 98:   *flag                   = attr[keyval].active;
 99:   *(void **)attribute_val = attr[keyval].attribute_val;
100:   return MPI_SUCCESS;
101: }

103: int MPI_Comm_create(MPI_Comm comm,MPI_Group group,MPI_Comm *newcomm)
104: {
105:   *newcomm =  comm;
106:   return MPI_SUCCESS;
107: }

109: static int dups = 0;
110: int MPI_Comm_dup(MPI_Comm comm,MPI_Comm *out)
111: {
112:   *out = comm;
113:   dups++;
114:   return 0;
115: }

117: int MPI_Comm_free(MPI_Comm *comm)
118: {
119:   int i;

121:   if (--dups) return MPI_SUCCESS;
122:   for (i=0; i<num_attr; i++) {
123:     if (attr[i].active && attr[i].del) {
124:       (*attr[i].del)(*comm,i,attr[i].attribute_val,attr[i].extra_state);
125:     }
126:     attr[i].active = 0;
127:   }
128:   return MPI_SUCCESS;
129: }

131: int MPI_Abort(MPI_Comm comm,int errorcode)
132: {
133:   abort();
134:   return MPI_SUCCESS;
135: }

137: /* --------------------------------------------------------------------------*/
138: 
139: static int MPI_was_initialized = 0;
140: static int MPI_was_finalized   = 0;

142: int MPI_Init(int *argc, char ***argv)
143: {
144:   if (MPI_was_initialized) return 1;
145:   if (MPI_was_finalized) return 1;
146:   MPI_was_initialized = 1;
147:   return 0;
148: }

150: int MPI_Finalize(void)
151: {
152:   if (MPI_was_finalized) return 1;
153:   if (!MPI_was_initialized) return 1;
154:   MPI_was_finalized = 1;
155:   return 0;
156: }

158: int MPI_Initialized(int *flag)
159: {
160:   *flag = MPI_was_initialized;
161:   return 0;
162: }

164: int MPI_Finalized(int *flag)
165: {
166:   *flag = MPI_was_finalized;
167:   return 0;
168: }

170: /* -------------------     Fortran versions of several routines ------------------ */

172: #if defined(PETSC_HAVE_FORTRAN_CAPS)
173: #define mpi_init_             MPI_INIT
174: #define mpi_finalize_         MPI_FINALIZE
175: #define mpi_comm_size_        MPI_COMM_SIZE
176: #define mpi_comm_rank_        MPI_COMM_RANK
177: #define mpi_abort_            MPI_ABORT
178: #define mpi_reduce_           MPI_REDUCE
179: #define mpi_allreduce_        MPI_ALLREDUCE
180: #define mpi_barrier_          MPI_BARRIER
181: #define mpi_bcast_            MPI_BCAST
182: #define mpi_gather_           MPI_GATHER
183: #define mpi_allgather_        MPI_ALLGATHER
184: #define mpi_comm_split_       MPI_COMM_SPLIT
185: #define mpi_scan_             MPI_SCAN
186: #define mpi_send_             MPI_SEND
187: #define mpi_recv_             MPI_RECV
188: #define mpi_reduce_scatter_   MPI_REDUCE_SCATTER
189: #define mpi_irecv_            MPI_IRECV
190: #define mpi_isend_            MPI_ISEND
191: #define mpi_sendrecv_         MPI_SENDRECV
192: #define mpi_test_             MPI_TEST
193: #define mpi_waitall_          MPI_WAITALL
194: #define mpi_waitany_          MPI_WAITANY
195: #define mpi_allgatherv_       MPI_ALLGATHERV
196: #define mpi_alltoallv_        MPI_ALLTOALLV
197: #define mpi_comm_create_      MPI_COMM_CREATE
198: #define mpi_address_          MPI_ADDRESS
199: #define mpi_pack_             MPI_PACK
200: #define mpi_unpack_           MPI_UNPACK
201: #define mpi_pack_size_        MPI_PACK_SIZE
202: #define mpi_type_struct_      MPI_TYPE_STRUCT
203: #define mpi_type_commit_      MPI_TYPE_COMMIT
204: #define mpi_wtime_            MPI_WTIME
205: #define mpi_cancel_           MPI_CANCEL
206: #define mpi_comm_dup_         MPI_COMM_DUP
207: #define mpi_comm_free_        MPI_COMM_FREE
208: #define mpi_get_count_        MPI_GET_COUNT
209: #define mpi_get_processor_name_ MPI_GET_PROCESSOR_NAME
210: #define mpi_initialized_      MPI_INITIALIZED
211: #define mpi_iprobe_           MPI_IPROBE
212: #define mpi_probe_            MPI_PROBE
213: #define mpi_request_free_     MPI_REQUEST_FREE
214: #define mpi_ssend_            MPI_SSEND
215: #define mpi_wait_             MPI_WAIT
216: #define mpi_comm_group_       MPI_COMM_GROUP
217: #define mpi_exscan_           MPI_EXSCAN
218: #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE)
219: #define mpi_init_             mpi_init
220: #define mpi_finalize_         mpi_finalize
221: #define mpi_comm_size_        mpi_comm_size
222: #define mpi_comm_rank_        mpi_comm_rank
223: #define mpi_abort_            mpi_abort
224: #define mpi_reduce_           mpi_reduce
225: #define mpi_allreduce_        mpi_allreduce
226: #define mpi_barrier_          mpi_barrier
227: #define mpi_bcast_            mpi_bcast
228: #define mpi_gather_           mpi_gather
229: #define mpi_allgather_        mpi_allgather
230: #define mpi_comm_split_       mpi_comm_split
231: #define mpi_scan_             mpi_scan
232: #define mpi_send_             mpi_send
233: #define mpi_recv_             mpi_recv
234: #define mpi_reduce_scatter_   mpi_reduce_scatter
235: #define mpi_irecv_            mpi_irecv
236: #define mpi_isend_            mpi_isend
237: #define mpi_sendrecv_         mpi_sendrecv
238: #define mpi_test_             mpi_test
239: #define mpi_waitall_          mpi_waitall
240: #define mpi_waitany_          mpi_waitany
241: #define mpi_allgatherv_       mpi_allgatherv
242: #define mpi_alltoallv_        mpi_alltoallv
243: #define mpi_comm_create_      mpi_comm_create
244: #define mpi_address_          mpi_address
245: #define mpi_pack_             mpi_pack
246: #define mpi_unpack_           mpi_unpack
247: #define mpi_pack_size_        mpi_pack_size
248: #define mpi_type_struct_      mpi_type_struct
249: #define mpi_type_commit_      mpi_type_commit
250: #define mpi_wtime_            mpi_wtime
251: #define mpi_cancel_           mpi_cancel
252: #define mpi_comm_dup_         mpi_comm_dup
253: #define mpi_comm_free_        mpi_comm_free
254: #define mpi_get_count_        mpi_get_count
255: #define mpi_get_processor_name_ mpi_get_processor_name
256: #define mpi_initialized_      mpi_initialized
257: #define mpi_iprobe_           mpi_iprobe
258: #define mpi_probe_            mpi_probe
259: #define mpi_request_free_     mpi_request_free
260: #define mpi_ssend_            mpi_ssend
261: #define mpi_wait_             mpi_wait
262: #define mpi_comm_group_       mpi_comm_group
263: #define mpi_exscan_           mpi_exscan
264: #endif

266: #if defined(PETSC_HAVE_FORTRAN_UNDERSCORE_UNDERSCORE)
267: #define mpi_init_             mpi_init__
268: #define mpi_finalize_         mpi_finalize__
269: #define mpi_comm_size_        mpi_comm_size__
270: #define mpi_comm_rank_        mpi_comm_rank__
271: #define mpi_abort_            mpi_abort__
272: #define mpi_reduce_           mpi_reduce__
273: #define mpi_allreduce_        mpi_allreduce__
274: #define mpi_barrier_          mpi_barrier__
275: #define mpi_bcast_            mpi_bcast__
276: #define mpi_gather_           mpi_gather__
277: #define mpi_allgather_        mpi_allgather__
278: #define mpi_comm_split_       mpi_comm_split__
279: #define mpi_scan_             mpi_scan__
280: #define mpi_send_             mpi_send__
281: #define mpi_recv_             mpi_recv__
282: #define mpi_reduce_scatter_   mpi_reduce_scatter__
283: #define mpi_irecv_            mpi_irecv__
284: #define mpi_isend_            mpi_isend__
285: #define mpi_sendrecv_         mpi_sendrecv__
286: #define mpi_test_             mpi_test__
287: #define mpi_waitall_          mpi_waitall__
288: #define mpi_waitany_          mpi_waitany__
289: #define mpi_allgatherv_       mpi_allgatherv__
290: #define mpi_alltoallv_        mpi_alltoallv__
291: #define mpi_comm_create_      mpi_comm_create__
292: #define mpi_address_          mpi_address__
293: #define mpi_pack_             mpi_pack__
294: #define mpi_unpack_           mpi_unpack__
295: #define mpi_pack_size_        mpi_pack_size__
296: #define mpi_type_struct_      mpi_type_struct__
297: #define mpi_type_commit_      mpi_type_commit__
298: #define mpi_wtime_            mpi_wtime__
299: #define mpi_cancel_           mpi_cancel__
300: #define mpi_comm_dup_         mpi_comm_dup__
301: #define mpi_comm_free_        mpi_comm_free__
302: #define mpi_get_count_        mpi_get_count__
303: #define mpi_get_processor_name_ mpi_get_processor_name__
304: #define mpi_initialized_      mpi_initialized__
305: #define mpi_iprobe_           mpi_iprobe__
306: #define mpi_probe_            mpi_probe__
307: #define mpi_request_free_     mpi_request_free__
308: #define mpi_ssend_            mpi_ssend__
309: #define mpi_wait_             mpi_wait__
310: #define mpi_comm_group_       mpi_comm_group__
311: #define mpi_exscan_           mpi_exscan__
312: #endif


315: /* Do not build fortran interface if MPI namespace colision is to be avoided */
316: #if !defined(MPIUNI_AVOID_MPI_NAMESPACE)

318: void PETSC_STDCALL  mpi_init_(int *ierr)
319: {
320:   *MPI_Init((int*)0, (char***)0);
321: }

323: void PETSC_STDCALL  mpi_finalize_(int *ierr)
324: {
325:   *MPI_Finalize();
326: }

328: void PETSC_STDCALL mpi_comm_size_(MPI_Comm *comm,int *size,int *ierr)
329: {
330:   *size = 1;
331:   *0;
332: }

334: void PETSC_STDCALL mpi_comm_rank_(MPI_Comm *comm,int *rank,int *ierr)
335: {
336:   *rank=0;
337:   *ierr=MPI_SUCCESS;
338: }

340: void PETSC_STDCALL mpi_comm_split_(MPI_Comm *comm,int *color,int *key, MPI_Comm *newcomm, int *ierr)
341: {
342:   *newcomm = *comm;
343:   *ierr=MPI_SUCCESS;
344: }

346: void PETSC_STDCALL mpi_abort_(MPI_Comm *comm,int *errorcode,int *ierr)
347: {
348:   abort();
349:   *MPI_SUCCESS;
350: }

352: void PETSC_STDCALL mpi_reduce_(void *sendbuf,void *recvbuf,int *count,int *datatype,int *op,int *root,int *comm,int *ierr)
353: {
354:   MPIUNI_Memcpy(recvbuf,sendbuf,(*count)*MPIUNI_DATASIZE[*datatype]);
355:   *MPI_SUCCESS;
356: }

358: void PETSC_STDCALL mpi_allreduce_(void *sendbuf,void *recvbuf,int *count,int *datatype,int *op,int *comm,int *ierr)
359: {
360:   MPIUNI_Memcpy(recvbuf,sendbuf,(*count)*MPIUNI_DATASIZE[*datatype]);
361:   *MPI_SUCCESS;
362: }

364: void PETSC_STDCALL mpi_barrier_(MPI_Comm *comm,int *ierr)
365: {
366:   *MPI_SUCCESS;
367: }

369: void PETSC_STDCALL mpi_bcast_(void *buf,int *count,int *datatype,int *root,int *comm,int *ierr)
370: {
371:   *MPI_SUCCESS;
372: }


375: void PETSC_STDCALL mpi_gather_(void *sendbuf,int *scount,int *sdatatype, void* recvbuf, int* rcount, int* rdatatype, int *root,int *comm,int *ierr)
376: {
377:   MPIUNI_Memcpy(recvbuf,sendbuf,(*scount)*MPIUNI_DATASIZE[*sdatatype]);
378:   *MPI_SUCCESS;
379: }

381: void PETSC_STDCALL mpi_allgather_(void *sendbuf,int *scount,int *sdatatype, void* recvbuf, int* rcount, int* rdatatype,int *comm,int *ierr)
382: {
383:   MPIUNI_Memcpy(recvbuf,sendbuf,(*scount)*MPIUNI_DATASIZE[*sdatatype]);
384:   *MPI_SUCCESS;
385: }

387: void PETSC_STDCALL mpi_scan_(void *sendbuf,void *recvbuf,int *count,int *datatype,int *op,int *comm,int *ierr)
388: {
389:   MPIUNI_Memcpy(recvbuf,sendbuf,(*count)*MPIUNI_DATASIZE[*datatype]);
390:   *MPI_SUCCESS;
391: }

393: void PETSC_STDCALL mpi_send_(void*buf,int *count,int *datatype,int *dest,int *tag,int *comm,int *ierr )
394: {
395:   *MPI_Abort(MPI_COMM_WORLD,0);
396: }

398: void PETSC_STDCALL mpi_recv_(void*buf,int *count,int *datatype,int *source,int *tag,int *comm,int status,int *ierr )
399: {
400:   *MPI_Abort(MPI_COMM_WORLD,0);
401: }

403: void PETSC_STDCALL mpi_reduce_scatter_(void*sendbuf,void*recvbuf,int *recvcounts,int *datatype,int *op,int *comm,int *ierr)
404: {
405:   *MPI_Abort(MPI_COMM_WORLD,0);
406: }

408: void PETSC_STDCALL mpi_irecv_(void*buf,int *count, int *datatype, int *source, int *tag, int *comm, int *request, int *ierr)
409: {
410:   *MPI_Abort(MPI_COMM_WORLD,0);
411: }

413: void PETSC_STDCALL mpi_isend_(void*buf,int *count,int *datatype,int *dest,int *tag,int *comm,int *request, int *ierr)
414: {
415:   *MPI_Abort(MPI_COMM_WORLD,0);
416: }

418: void PETSC_STDCALL mpi_sendrecv_(void*sendbuf,int *sendcount,int *sendtype,int *dest,int *sendtag,void*recvbuf,int *recvcount,int *recvtype,int *source,int *recvtag,int *comm,int *status,int *ierr)
419: {
420:   MPIUNI_Memcpy(recvbuf,sendbuf,(*sendcount)*MPIUNI_DATASIZE[*sendtype]);
421:   *MPI_SUCCESS;
422: }

424: void PETSC_STDCALL mpi_test_(int *request,int *flag,int *status,int *ierr)
425: {
426:   *MPI_Abort(MPI_COMM_WORLD,0);
427: }

429: void PETSC_STDCALL mpi_waitall_(int *count,int *array_of_requests,int *array_of_statuses,int *ierr)
430: {
431:   *MPI_SUCCESS;
432: }

434:   void PETSC_STDCALL mpi_waitany_(int *count,int *array_of_requests,int * index, int *status,int *ierr)
435: {
436:   *MPI_SUCCESS;
437: }

439: void PETSC_STDCALL mpi_allgatherv_(void*sendbuf,int *sendcount,int *sendtype,void*recvbuf,int *recvcounts,int *displs,int *recvtype,int *comm,int *ierr)
440: {
441:   MPIUNI_Memcpy(recvbuf,sendbuf,(*sendcount)*MPIUNI_DATASIZE[*sendtype]);
442:   *MPI_SUCCESS;
443: }

445: void PETSC_STDCALL mpi_alltoallv_(void*sendbuf,int *sendcounts,int *sdispls,int *sendtype,void*recvbuf,int *recvcounts,int *rdispls,int *recvtype,int *comm,int *ierr)
446: {
447:   MPIUNI_Memcpy(recvbuf,sendbuf,(*sendcounts)*MPIUNI_DATASIZE[*sendtype]);
448:   *MPI_SUCCESS;
449: }

451: void PETSC_STDCALL mpi_comm_create_(int *comm,int *group,int *newcomm,int *ierr)
452: {
453:   *newcomm =  *comm;
454:   *MPI_SUCCESS;
455: }

457: void PETSC_STDCALL mpi_address_(void*location,MPIUNI_INTPTR *address,int *ierr)
458: {
459:   *address =  (MPIUNI_INTPTR) location;
460:   *MPI_SUCCESS;
461: }

463: void PETSC_STDCALL mpi_pack_(void*inbuf,int *incount,int *datatype,void*outbuf,int *outsize,int *position,int *comm,int *ierr)
464: {
465:   *MPI_Abort(MPI_COMM_WORLD,0);
466: }

468: void PETSC_STDCALL mpi_unpack_(void*inbuf,int *insize,int *position,void*outbuf,int *outcount,int *datatype,int *comm,int *ierr)
469: {
470:   *MPI_Abort(MPI_COMM_WORLD,0);
471: }

473: void PETSC_STDCALL mpi_pack_size_(int *incount,int *datatype,int *comm,int *size,int *ierr)
474: {
475:   *MPI_Abort(MPI_COMM_WORLD,0);
476: }

478: void PETSC_STDCALL mpi_type_struct_(int *count,int *array_of_blocklengths,int * array_of_displaments,int *array_of_types,int *newtype,int *ierr)
479: {
480:   *MPI_Abort(MPI_COMM_WORLD,0);
481: }

483: void PETSC_STDCALL mpi_type_commit_(int *datatype,int *ierr)
484: {
485:   *MPI_SUCCESS;
486: }

488: double PETSC_STDCALL mpi_wtime_(void)
489: {
490:   return 0.0;
491: }

493: void PETSC_STDCALL mpi_cancel_(int *request,int *ierr)
494: {
495:   *MPI_SUCCESS;
496: }

498: void PETSC_STDCALL mpi_comm_dup_(int *comm,int *out,int *ierr)
499: {
500:   *out = *comm;
501:   *MPI_SUCCESS;
502: }

504: void PETSC_STDCALL mpi_comm_free_(int *comm,int *ierr)
505: {
506:   *MPI_SUCCESS;
507: }

509: void PETSC_STDCALL mpi_get_count_(int *status,int *datatype,int *count,int *ierr)
510: {
511:   *MPI_Abort(MPI_COMM_WORLD,0);
512: }

514: /* duplicate from fortranimpl.h */
515: #if defined(PETSC_HAVE_FORTRAN_MIXED_STR_ARG)
516: #define PETSC_MIXED_LEN(len) ,int len
517: #define PETSC_END_LEN(len)
518: #else
519: #define PETSC_MIXED_LEN(len)
520: #define PETSC_END_LEN(len)   ,int len
521: #endif

523: void PETSC_STDCALL mpi_get_processor_name_(char *name PETSC_MIXED_LEN(len),int *result_len,int *ierr PETSC_END_LEN(len))
524: {
525:   MPIUNI_Memcpy(name,"localhost",9*sizeof(char));
526:   *result_len = 9;
527:   *MPI_SUCCESS;
528: }

530: void PETSC_STDCALL mpi_initialized_(int *flag,int *ierr)
531: {
532:   *flag = MPI_was_initialized;
533:   *MPI_SUCCESS;
534: }

536: void PETSC_STDCALL mpi_iprobe_(int *source,int *tag,int *comm,int *glag,int *status,int *ierr)
537: {
538:   *MPI_SUCCESS;
539: }

541: void PETSC_STDCALL mpi_probe_(int *source,int *tag,int *comm,int *flag,int *status,int *ierr)
542: {
543:   *MPI_SUCCESS;
544: }

546: void PETSC_STDCALL mpi_request_free_(int *request,int *ierr)
547: {
548:   *MPI_SUCCESS;
549: }

551: void PETSC_STDCALL mpi_ssend_(void*buf,int *count,int *datatype,int *dest,int *tag,int *comm,int *ierr)
552: {
553:   *MPI_Abort(MPI_COMM_WORLD,0);
554: }

556: void PETSC_STDCALL mpi_wait_(int *request,int *status,int *ierr)
557: {
558:   *MPI_SUCCESS;
559: }

561: void PETSC_STDCALL mpi_comm_group_(int*comm,int*group,int *ierr)
562: {
563:   *MPI_SUCCESS;
564: }

566: void PETSC_STDCALL mpi_exscan_(void*sendbuf,void*recvbuf,int*count,int*datatype,int*op,int*comm,int*ierr)
567: {
568:   *MPI_SUCCESS;
569: }

571: #endif /* MPIUNI_AVOID_MPI_NAMESPACE */

573: #if defined(__cplusplus)
574: }
575: #endif