1 /*
2 * Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
3 * University Research and Technology
4 * Corporation. All rights reserved.
5 * Copyright (c) 2004-2005 The University of Tennessee and The University
6 * of Tennessee Research Foundation. All rights
7 * reserved.
8 * Copyright (c) 2004-2008 High Performance Computing Center Stuttgart,
9 * University of Stuttgart. All rights reserved.
10 * Copyright (c) 2004-2005 The Regents of the University of California.
11 * All rights reserved.
12 * Copyright (c) 2006 Cisco Systems, Inc. All rights reserved.
13 * Copyright (c) 2015 Research Organization for Information Science
14 * and Technology (RIST). All rights reserved.
15 * $COPYRIGHT$
16 *
17 * Additional copyrights may follow
18 *
19 * $HEADER$
20 */
21
22 #include "ompi_config.h"
23
24 #include <stdio.h>
25
26 #include "ompi/mpi/c/bindings.h"
27 #include "ompi/runtime/params.h"
28 #include "ompi/communicator/communicator.h"
29 #include "ompi/errhandler/errhandler.h"
30 #include "ompi/memchecker.h"
31
32 #if OMPI_BUILD_MPI_PROFILING
33 #if OPAL_HAVE_WEAK_SYMBOLS
34 #pragma weak MPI_Comm_size = PMPI_Comm_size
35 #endif
36 #define MPI_Comm_size PMPI_Comm_size
37 #endif
38
39 static const char FUNC_NAME[] = "MPI_Comm_size";
40
41
42 int MPI_Comm_size(MPI_Comm comm, int *size)
43 {
44 MEMCHECKER(
45 memchecker_comm(comm);
46 );
47
48 OPAL_CR_NOOP_PROGRESS();
49
50 if ( MPI_PARAM_CHECK ) {
51 OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
52
53 if ( ompi_comm_invalid (comm)) {
54 return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD,
55 MPI_ERR_COMM, FUNC_NAME);
56 }
57
58 if ( NULL == size ) {
59 return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG, FUNC_NAME);
60 }
61 }
62
63 *size = ompi_comm_size((ompi_communicator_t*)comm);
64 return MPI_SUCCESS;
65 }