root/oshmem/shmem/fortran/shmem_alltoall_f.c

   1 /*
   2  * Copyright (c) 2013-2018 Mellanox Technologies, Inc.
   3  *                         All rights reserved.
   4  * Copyright (c) 2013 Cisco Systems, Inc.  All rights reserved.
   5  * $COPYRIGHT$
   6  *
   7  * Additional copyrights may follow
   8  *
   9  * $HEADER$
  10  */
  11 
  12 #include "oshmem_config.h"
  13 #include "oshmem/shmem/fortran/bindings.h"
  14 #include "oshmem/include/shmem.h"
  15 #include "oshmem/constants.h"
  16 #include "oshmem/mca/scoll/scoll.h"
  17 #include "oshmem/proc/proc.h"
  18 #include "oshmem/op/op.h"
  19 
  20 #if OSHMEM_PROFILING
  21 #include "oshmem/shmem/fortran/profile/pbindings.h"
  22 SHMEM_GENERATE_WEAK_BINDINGS(SHMEM_ALLTOALL32, shmem_alltoall32)
  23 SHMEM_GENERATE_WEAK_BINDINGS(SHMEM_ALLTOALL64, shmem_alltoall64)
  24 SHMEM_GENERATE_WEAK_BINDINGS(SHMEM_ALLTOALLS32, shmem_alltoalls32)
  25 SHMEM_GENERATE_WEAK_BINDINGS(SHMEM_ALLTOALLS64, shmem_alltoalls64)
  26 #include "oshmem/shmem/fortran/profile/defines.h"
  27 #endif
  28 
  29 SHMEM_GENERATE_FORTRAN_BINDINGS_SUB (void,
  30         SHMEM_ALLTOALL32,
  31         shmem_alltoall32_,
  32         shmem_alltoall32__,
  33         shmem_alltoall32_f,
  34         (FORTRAN_POINTER_T target, FORTRAN_POINTER_T source, MPI_Fint *nlong, MPI_Fint *PE_start, MPI_Fint * logPE_stride, MPI_Fint *PE_size, FORTRAN_POINTER_T pSync),
  35         (target, source, nlong, PE_start, logPE_stride, PE_size, pSync))
  36 
  37 SHMEM_GENERATE_FORTRAN_BINDINGS_SUB (void,
  38         SHMEM_ALLTOALL64,
  39         shmem_alltoall64_,
  40         shmem_alltoall64__,
  41         shmem_alltoall64_f,
  42         (FORTRAN_POINTER_T target, FORTRAN_POINTER_T source, MPI_Fint *nlong, MPI_Fint *PE_start, MPI_Fint * logPE_stride, MPI_Fint *PE_size, FORTRAN_POINTER_T pSync),
  43         (target, source, nlong, PE_start, logPE_stride, PE_size, pSync))
  44 
  45 SHMEM_GENERATE_FORTRAN_BINDINGS_SUB (void,
  46         SHMEM_ALLTOALLS32,
  47         shmem_alltoalls32_,
  48         shmem_alltoalls32__,
  49         shmem_alltoalls32_f,
  50         (FORTRAN_POINTER_T target, FORTRAN_POINTER_T source, MPI_Fint *dst, MPI_Fint *sst, MPI_Fint *nlong, MPI_Fint *PE_start, MPI_Fint * logPE_stride, MPI_Fint *PE_size, FORTRAN_POINTER_T pSync),
  51         (target, source, dst, sst, nlong, PE_start, logPE_stride, PE_size, pSync))
  52 
  53 SHMEM_GENERATE_FORTRAN_BINDINGS_SUB (void,
  54         SHMEM_ALLTOALLS64,
  55         shmem_alltoalls64_,
  56         shmem_alltoalls64__,
  57         shmem_alltoalls64_f,
  58         (FORTRAN_POINTER_T target, FORTRAN_POINTER_T source, MPI_Fint *dst, MPI_Fint *sst, MPI_Fint *nlong, MPI_Fint *PE_start, MPI_Fint * logPE_stride, MPI_Fint *PE_size, FORTRAN_POINTER_T pSync),
  59         (target, source, dst, sst, nlong, PE_start, logPE_stride, PE_size, pSync))
  60 
  61 #define SHMEM_ALLTOALL(F_NAME, T_NAME) void F_NAME(FORTRAN_POINTER_T target, \
  62     FORTRAN_POINTER_T source, \
  63     MPI_Fint *nlong,\
  64     MPI_Fint *PE_start, \
  65     MPI_Fint *logPE_stride, \
  66     MPI_Fint *PE_size, \
  67     FORTRAN_POINTER_T pSync)\
  68 {\
  69     int rc;\
  70     oshmem_group_t *group;\
  71     /* Create group basing PE_start, logPE_stride and PE_size */\
  72     group = oshmem_proc_group_create_nofail(OMPI_FINT_2_INT(*PE_start), \
  73                                             (1 << OMPI_FINT_2_INT(*logPE_stride)), \
  74                                             OMPI_FINT_2_INT(*PE_size));\
  75     oshmem_op_t* op = T_NAME;\
  76     \
  77     /* Call collective broadcast operation */\
  78     rc = group->g_scoll.scoll_alltoall( group, \
  79             FPTR_2_VOID_PTR(target), \
  80             FPTR_2_VOID_PTR(source), \
  81             1, \
  82             1, \
  83             OMPI_FINT_2_INT(*nlong), \
  84             op->dt_size, \
  85             FPTR_2_VOID_PTR(pSync), SCOLL_DEFAULT_ALG );\
  86     oshmem_proc_group_destroy(group);\
  87     RUNTIME_CHECK_RC(rc);\
  88 }
  89 
  90 #define SHMEM_ALLTOALLS(F_NAME, T_NAME) void F_NAME(FORTRAN_POINTER_T target, \
  91     FORTRAN_POINTER_T source, \
  92     MPI_Fint *dst,\
  93     MPI_Fint *sst,\
  94     MPI_Fint *nlong,\
  95     MPI_Fint *PE_start, \
  96     MPI_Fint *logPE_stride, \
  97     MPI_Fint *PE_size, \
  98     FORTRAN_POINTER_T pSync)\
  99 {\
 100     int rc;\
 101     oshmem_group_t *group;\
 102     /* Create group basing PE_start, logPE_stride and PE_size */\
 103     group = oshmem_proc_group_create_nofail(OMPI_FINT_2_INT(*PE_start), \
 104                                             (1 << OMPI_FINT_2_INT(*logPE_stride)), \
 105                                             OMPI_FINT_2_INT(*PE_size));\
 106     oshmem_op_t* op = T_NAME;\
 107     \
 108     /* Call collective broadcast operation */\
 109     rc = group->g_scoll.scoll_alltoall( group, \
 110             FPTR_2_VOID_PTR(target), \
 111             FPTR_2_VOID_PTR(source), \
 112             OMPI_FINT_2_INT(*dst), \
 113             OMPI_FINT_2_INT(*sst), \
 114             OMPI_FINT_2_INT(*nlong), \
 115             op->dt_size, \
 116             FPTR_2_VOID_PTR(pSync), SCOLL_DEFAULT_ALG );\
 117     oshmem_proc_group_destroy(group);\
 118     RUNTIME_CHECK_RC(rc);\
 119 }
 120 
 121 SHMEM_ALLTOALL(shmem_alltoall32_f, oshmem_op_prod_fint4)
 122 SHMEM_ALLTOALL(shmem_alltoall64_f, oshmem_op_prod_fint8)
 123 SHMEM_ALLTOALLS(shmem_alltoalls32_f, oshmem_op_prod_fint4)
 124 SHMEM_ALLTOALLS(shmem_alltoalls64_f, oshmem_op_prod_fint8)