oshmem: Align OSHMEM API with spec v1.3 (Add nonblocking put/get C functions)
Этот коммит содержится в:
Igor Ivanov
родитель
8464b6147a
Коммит
abe7ba5720
@ -1,5 +1,5 @@
|
|||||||
#
|
#
|
||||||
# Copyright (c) 2013-2015 Mellanox Technologies, Inc.
|
# Copyright (c) 2013-2016 Mellanox Technologies, Inc.
|
||||||
# All rights reserved
|
# All rights reserved
|
||||||
# Copyright (c) 2014 Cisco Systems, Inc. All rights reserved.
|
# Copyright (c) 2014 Cisco Systems, Inc. All rights reserved.
|
||||||
# $COPYRIGHT$
|
# $COPYRIGHT$
|
||||||
@ -40,6 +40,8 @@ OSHMEM_API_SOURCES = \
|
|||||||
shmem_wait.c \
|
shmem_wait.c \
|
||||||
shmem_iget.c \
|
shmem_iget.c \
|
||||||
shmem_iput.c \
|
shmem_iput.c \
|
||||||
|
shmem_get_nb.c \
|
||||||
|
shmem_put_nb.c \
|
||||||
shmem_udcflush.c \
|
shmem_udcflush.c \
|
||||||
shmem_udcflush_line.c \
|
shmem_udcflush_line.c \
|
||||||
shmem_set_cache_inv.c \
|
shmem_set_cache_inv.c \
|
||||||
|
|||||||
@ -1,5 +1,5 @@
|
|||||||
#
|
#
|
||||||
# Copyright (c) 2013 Mellanox Technologies, Inc.
|
# Copyright (c) 2013-2016 Mellanox Technologies, Inc.
|
||||||
# All rights reserved
|
# All rights reserved
|
||||||
# Copyright (c) 2014 Cisco Systems, Inc. All rights reserved.
|
# Copyright (c) 2014 Cisco Systems, Inc. All rights reserved.
|
||||||
# $COPYRIGHT$
|
# $COPYRIGHT$
|
||||||
@ -52,6 +52,8 @@ OSHMEM_API_SOURCES = \
|
|||||||
pshmem_wait.c \
|
pshmem_wait.c \
|
||||||
pshmem_iget.c \
|
pshmem_iget.c \
|
||||||
pshmem_iput.c \
|
pshmem_iput.c \
|
||||||
|
pshmem_get_nb.c \
|
||||||
|
pshmem_put_nb.c \
|
||||||
pshmem_udcflush.c \
|
pshmem_udcflush.c \
|
||||||
pshmem_udcflush_line.c \
|
pshmem_udcflush_line.c \
|
||||||
pshmem_set_cache_inv.c \
|
pshmem_set_cache_inv.c \
|
||||||
|
|||||||
@ -1,5 +1,5 @@
|
|||||||
/*
|
/*
|
||||||
* Copyright (c) 2013-2015 Mellanox Technologies, Inc.
|
* Copyright (c) 2013-2016 Mellanox Technologies, Inc.
|
||||||
* All rights reserved.
|
* All rights reserved.
|
||||||
* $COPYRIGHT$
|
* $COPYRIGHT$
|
||||||
*
|
*
|
||||||
@ -107,6 +107,24 @@
|
|||||||
#define shmem_iput64 pshmem_iput64
|
#define shmem_iput64 pshmem_iput64
|
||||||
#define shmem_iput128 pshmem_iput128
|
#define shmem_iput128 pshmem_iput128
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Non-block data put routines
|
||||||
|
*/
|
||||||
|
#define shmem_char_put_nbi pshmem_char_put_nbi
|
||||||
|
#define shmem_short_put_nbi pshmem_short_put_nbi
|
||||||
|
#define shmem_int_put_nbi pshmem_int_put_nbi
|
||||||
|
#define shmem_long_put_nbi pshmem_long_put_nbi
|
||||||
|
#define shmem_float_put_nbi pshmem_float_put_nbi
|
||||||
|
#define shmem_double_put_nbi pshmem_double_put_nbi
|
||||||
|
#define shmem_longlong_put_nbi pshmem_longlong_put_nbi
|
||||||
|
#define shmem_longdouble_put_nbi pshmem_longdouble_put_nbi
|
||||||
|
#define shmem_put8_nbi pshmem_put8_nbi
|
||||||
|
#define shmem_put16_nbi pshmem_put16_nbi
|
||||||
|
#define shmem_put32_nbi pshmem_put32_nbi
|
||||||
|
#define shmem_put64_nbi pshmem_put64_nbi
|
||||||
|
#define shmem_put128_nbi pshmem_put128_nbi
|
||||||
|
#define shmem_putmem_nbi pshmem_putmem_nbi
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Elemental get routines
|
* Elemental get routines
|
||||||
*/
|
*/
|
||||||
@ -154,6 +172,24 @@
|
|||||||
#define shmem_iget64 pshmem_iget64
|
#define shmem_iget64 pshmem_iget64
|
||||||
#define shmem_iget128 pshmem_iget128
|
#define shmem_iget128 pshmem_iget128
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Non-block data get routines
|
||||||
|
*/
|
||||||
|
#define shmem_char_get_nbi pshmem_char_get_nbi
|
||||||
|
#define shmem_short_get_nbi pshmem_short_get_nbi
|
||||||
|
#define shmem_int_get_nbi pshmem_int_get_nbi
|
||||||
|
#define shmem_long_get_nbi pshmem_long_get_nbi
|
||||||
|
#define shmem_float_get_nbi pshmem_float_get_nbi
|
||||||
|
#define shmem_double_get_nbi pshmem_double_get_nbi
|
||||||
|
#define shmem_longlong_get_nbi pshmem_longlong_get_nbi
|
||||||
|
#define shmem_longdouble_get_nbi pshmem_longdouble_get_nbi
|
||||||
|
#define shmem_get8_nbi pshmem_get8_nbi
|
||||||
|
#define shmem_get16_nbi pshmem_get16_nbi
|
||||||
|
#define shmem_get32_nbi pshmem_get32_nbi
|
||||||
|
#define shmem_get64_nbi pshmem_get64_nbi
|
||||||
|
#define shmem_get128_nbi pshmem_get128_nbi
|
||||||
|
#define shmem_getmem_nbi pshmem_getmem_nbi
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Atomic operations
|
* Atomic operations
|
||||||
*/
|
*/
|
||||||
|
|||||||
100
oshmem/shmem/c/shmem_get_nb.c
Обычный файл
100
oshmem/shmem/c/shmem_get_nb.c
Обычный файл
@ -0,0 +1,100 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2016 Mellanox Technologies, Inc.
|
||||||
|
* All rights reserved.
|
||||||
|
* $COPYRIGHT$
|
||||||
|
*
|
||||||
|
* Additional copyrights may follow
|
||||||
|
*
|
||||||
|
* $HEADER$
|
||||||
|
*/
|
||||||
|
#include "oshmem_config.h"
|
||||||
|
|
||||||
|
#include "oshmem/constants.h"
|
||||||
|
#include "oshmem/include/shmem.h"
|
||||||
|
|
||||||
|
#include "oshmem/runtime/runtime.h"
|
||||||
|
|
||||||
|
#include "oshmem/mca/spml/spml.h"
|
||||||
|
|
||||||
|
/*
|
||||||
|
* These routines retrieve data from a contiguous data object on a remote PE.
|
||||||
|
* The shmem_get() routines transfer nelems elements of the data object at address source
|
||||||
|
* on the remote PE (pe), to the data object at address target on the local PE. These routines
|
||||||
|
* return after the data has been copied to address target on the local pe.
|
||||||
|
*/
|
||||||
|
#define SHMEM_TYPE_GET_NB(type_name, type) \
|
||||||
|
void shmem##type_name##_get_nbi(type *target, const type *source, size_t nelems, int pe) \
|
||||||
|
{ \
|
||||||
|
int rc = OSHMEM_SUCCESS; \
|
||||||
|
size_t size = 0; \
|
||||||
|
\
|
||||||
|
RUNTIME_CHECK_INIT(); \
|
||||||
|
RUNTIME_CHECK_PE(pe); \
|
||||||
|
RUNTIME_CHECK_ADDR(source); \
|
||||||
|
\
|
||||||
|
size = nelems * sizeof(type); \
|
||||||
|
rc = MCA_SPML_CALL(get_nb( \
|
||||||
|
(void *)source, \
|
||||||
|
size, \
|
||||||
|
(void *)target, \
|
||||||
|
pe, NULL)); \
|
||||||
|
RUNTIME_CHECK_RC(rc); \
|
||||||
|
\
|
||||||
|
return ; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#if OSHMEM_PROFILING
|
||||||
|
#include "oshmem/include/pshmem.h"
|
||||||
|
#pragma weak shmem_char_get_nbi = pshmem_char_get_nbi
|
||||||
|
#pragma weak shmem_short_get_nbi = pshmem_short_get_nbi
|
||||||
|
#pragma weak shmem_int_get_nbi = pshmem_int_get_nbi
|
||||||
|
#pragma weak shmem_long_get_nbi = pshmem_long_get_nbi
|
||||||
|
#pragma weak shmem_longlong_get_nbi = pshmem_longlong_get_nbi
|
||||||
|
#pragma weak shmem_float_get_nbi = pshmem_float_get_nbi
|
||||||
|
#pragma weak shmem_double_get_nbi = pshmem_double_get_nbi
|
||||||
|
#pragma weak shmem_longdouble_get_nbi = pshmem_longdouble_get_nbi
|
||||||
|
#pragma weak shmem_get8_nbi = pshmem_get8_nbi
|
||||||
|
#pragma weak shmem_get16_nbi = pshmem_get16_nbi
|
||||||
|
#pragma weak shmem_get32_nbi = pshmem_get32_nbi
|
||||||
|
#pragma weak shmem_get64_nbi = pshmem_get64_nbi
|
||||||
|
#pragma weak shmem_get128_nbi = pshmem_get128_nbi
|
||||||
|
#pragma weak shmem_getmem_nbi = pshmem_getmem_nbi
|
||||||
|
#include "oshmem/shmem/c/profile/defines.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
SHMEM_TYPE_GET_NB(_char, char)
|
||||||
|
SHMEM_TYPE_GET_NB(_short, short)
|
||||||
|
SHMEM_TYPE_GET_NB(_int, int)
|
||||||
|
SHMEM_TYPE_GET_NB(_long, long)
|
||||||
|
SHMEM_TYPE_GET_NB(_longlong, long long)
|
||||||
|
SHMEM_TYPE_GET_NB(_float, float)
|
||||||
|
SHMEM_TYPE_GET_NB(_double, double)
|
||||||
|
SHMEM_TYPE_GET_NB(_longdouble, long double)
|
||||||
|
|
||||||
|
#define SHMEM_TYPE_GETMEM_NB(name, element_size, prefix) \
|
||||||
|
void prefix##name##_nbi(void *target, const void *source, size_t nelems, int pe) \
|
||||||
|
{ \
|
||||||
|
int rc = OSHMEM_SUCCESS; \
|
||||||
|
size_t size = 0; \
|
||||||
|
\
|
||||||
|
RUNTIME_CHECK_INIT(); \
|
||||||
|
RUNTIME_CHECK_PE(pe); \
|
||||||
|
RUNTIME_CHECK_ADDR(source); \
|
||||||
|
\
|
||||||
|
size = nelems * element_size; \
|
||||||
|
rc = MCA_SPML_CALL(get_nb( \
|
||||||
|
(void *)source, \
|
||||||
|
size, \
|
||||||
|
(void *)target, \
|
||||||
|
pe, NULL)); \
|
||||||
|
RUNTIME_CHECK_RC(rc); \
|
||||||
|
\
|
||||||
|
return ; \
|
||||||
|
}
|
||||||
|
|
||||||
|
SHMEM_TYPE_GETMEM_NB(_get8, 1, shmem)
|
||||||
|
SHMEM_TYPE_GETMEM_NB(_get16, 2, shmem)
|
||||||
|
SHMEM_TYPE_GETMEM_NB(_get32, 4, shmem)
|
||||||
|
SHMEM_TYPE_GETMEM_NB(_get64, 8, shmem)
|
||||||
|
SHMEM_TYPE_GETMEM_NB(_get128, 16, shmem)
|
||||||
|
SHMEM_TYPE_GETMEM_NB(_getmem, 1, shmem)
|
||||||
107
oshmem/shmem/c/shmem_put_nb.c
Обычный файл
107
oshmem/shmem/c/shmem_put_nb.c
Обычный файл
@ -0,0 +1,107 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (c) 2016 Mellanox Technologies, Inc.
|
||||||
|
* All rights reserved.
|
||||||
|
* $COPYRIGHT$
|
||||||
|
*
|
||||||
|
* Additional copyrights may follow
|
||||||
|
*
|
||||||
|
* $HEADER$
|
||||||
|
*/
|
||||||
|
#include "oshmem_config.h"
|
||||||
|
|
||||||
|
#include "oshmem/constants.h"
|
||||||
|
#include "oshmem/include/shmem.h"
|
||||||
|
|
||||||
|
#include "oshmem/runtime/runtime.h"
|
||||||
|
|
||||||
|
#include "oshmem/mca/spml/spml.h"
|
||||||
|
|
||||||
|
/*
|
||||||
|
* The nonblocking put routines provide a method for copying data from a contiguous local data
|
||||||
|
* object to a data object on a specified PE.
|
||||||
|
* These routines transfer nelems elements of the data object at address source on the calling
|
||||||
|
* PE, to the data object at address target on the remote PE pe. These routines start the
|
||||||
|
* remote transfer and may return before the data is delivered to the remote PE. The delivery
|
||||||
|
* of data into the data object on the destination PE from different put calls may occur in any
|
||||||
|
* order. Because of this, two successive put operations may deliver data out of order unless a
|
||||||
|
* call to shmem_fence() is introduced between the two calls.
|
||||||
|
* The routines return after posting the operation. The operation is considered complete after a
|
||||||
|
* subsequent call to shmem_quiet. At the completion of shmem_quiet, the data has been copied
|
||||||
|
* into the dest array on the destination PE.
|
||||||
|
*/
|
||||||
|
#define SHMEM_TYPE_PUT_NB(type_name, type) \
|
||||||
|
void shmem##type_name##_put_nbi(type *target, const type *source, size_t len, int pe) \
|
||||||
|
{ \
|
||||||
|
int rc = OSHMEM_SUCCESS; \
|
||||||
|
size_t size = 0; \
|
||||||
|
\
|
||||||
|
RUNTIME_CHECK_INIT(); \
|
||||||
|
RUNTIME_CHECK_PE(pe); \
|
||||||
|
RUNTIME_CHECK_ADDR(target); \
|
||||||
|
\
|
||||||
|
size = len * sizeof(type); \
|
||||||
|
rc = MCA_SPML_CALL(put_nb( \
|
||||||
|
(void *)target, \
|
||||||
|
size, \
|
||||||
|
(void *)source, \
|
||||||
|
pe, NULL)); \
|
||||||
|
RUNTIME_CHECK_RC(rc); \
|
||||||
|
\
|
||||||
|
return ; \
|
||||||
|
}
|
||||||
|
|
||||||
|
#if OSHMEM_PROFILING
|
||||||
|
#include "oshmem/include/pshmem.h"
|
||||||
|
#pragma weak shmem_char_put_nbi = pshmem_char_put_nbi
|
||||||
|
#pragma weak shmem_short_put_nbi = pshmem_short_put_nbi
|
||||||
|
#pragma weak shmem_int_put_nbi = pshmem_int_put_nbi
|
||||||
|
#pragma weak shmem_long_put_nbi = pshmem_long_put_nbi
|
||||||
|
#pragma weak shmem_longlong_put_nbi = pshmem_longlong_put_nbi
|
||||||
|
#pragma weak shmem_float_put_nbi = pshmem_float_put_nbi
|
||||||
|
#pragma weak shmem_double_put_nbi = pshmem_double_put_nbi
|
||||||
|
#pragma weak shmem_longdouble_put_nbi = pshmem_longdouble_put_nbi
|
||||||
|
#pragma weak shmem_put8_nbi = pshmem_put8_nbi
|
||||||
|
#pragma weak shmem_put16_nbi = pshmem_put16_nbi
|
||||||
|
#pragma weak shmem_put32_nbi = pshmem_put32_nbi
|
||||||
|
#pragma weak shmem_put64_nbi = pshmem_put64_nbi
|
||||||
|
#pragma weak shmem_put128_nbi = pshmem_put128_nbi
|
||||||
|
#pragma weak shmem_putmem_nbi = pshmem_putmem_nbi
|
||||||
|
#include "oshmem/shmem/c/profile/defines.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
SHMEM_TYPE_PUT_NB(_char, char)
|
||||||
|
SHMEM_TYPE_PUT_NB(_short, short)
|
||||||
|
SHMEM_TYPE_PUT_NB(_int, int)
|
||||||
|
SHMEM_TYPE_PUT_NB(_long, long)
|
||||||
|
SHMEM_TYPE_PUT_NB(_longlong, long long)
|
||||||
|
SHMEM_TYPE_PUT_NB(_float, float)
|
||||||
|
SHMEM_TYPE_PUT_NB(_double, double)
|
||||||
|
SHMEM_TYPE_PUT_NB(_longdouble, long double)
|
||||||
|
|
||||||
|
#define SHMEM_TYPE_PUTMEM_NB(name, element_size, prefix) \
|
||||||
|
void prefix##name##_nbi(void *target, const void *source, size_t nelems, int pe) \
|
||||||
|
{ \
|
||||||
|
int rc = OSHMEM_SUCCESS; \
|
||||||
|
size_t size = 0; \
|
||||||
|
\
|
||||||
|
RUNTIME_CHECK_INIT(); \
|
||||||
|
RUNTIME_CHECK_PE(pe); \
|
||||||
|
RUNTIME_CHECK_ADDR(target); \
|
||||||
|
\
|
||||||
|
size = nelems * element_size; \
|
||||||
|
rc = MCA_SPML_CALL(put_nb( \
|
||||||
|
(void *)target, \
|
||||||
|
size, \
|
||||||
|
(void *)source, \
|
||||||
|
pe, NULL)); \
|
||||||
|
RUNTIME_CHECK_RC(rc); \
|
||||||
|
\
|
||||||
|
return ; \
|
||||||
|
}
|
||||||
|
|
||||||
|
SHMEM_TYPE_PUTMEM_NB(_put8, 1, shmem)
|
||||||
|
SHMEM_TYPE_PUTMEM_NB(_put16, 2, shmem)
|
||||||
|
SHMEM_TYPE_PUTMEM_NB(_put32, 4, shmem)
|
||||||
|
SHMEM_TYPE_PUTMEM_NB(_put64, 8, shmem)
|
||||||
|
SHMEM_TYPE_PUTMEM_NB(_put128, 16, shmem)
|
||||||
|
SHMEM_TYPE_PUTMEM_NB(_putmem, 1, shmem)
|
||||||
Загрузка…
x
Ссылка в новой задаче
Block a user