69 строки
1.7 KiB
C
69 строки
1.7 KiB
C
/*
|
|
* Copyright (c) 2013 Mellanox Technologies, Inc.
|
|
* All rights reserved.
|
|
* Copyright (c) 2016 Research Organization for Information Science
|
|
* and Technology (RIST). All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "oshmem_config.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "oshmem/constants.h"
|
|
#include "oshmem/mca/spml/spml.h"
|
|
#include "oshmem/mca/atomic/atomic.h"
|
|
#include "oshmem/mca/atomic/base/base.h"
|
|
#include "oshmem/mca/memheap/memheap.h"
|
|
#include "oshmem/mca/memheap/base/base.h"
|
|
#include "oshmem/runtime/runtime.h"
|
|
|
|
#include "atomic_mxm.h"
|
|
|
|
int mca_atomic_mxm_swap(shmem_ctx_t ctx,
|
|
void *target,
|
|
void *prev,
|
|
uint64_t value,
|
|
size_t nlong,
|
|
int pe)
|
|
{
|
|
mxm_send_req_t sreq;
|
|
|
|
mca_atomic_mxm_req_init(&sreq, pe, target, nlong);
|
|
memcpy(prev, &value, nlong);
|
|
|
|
sreq.base.data.buffer.ptr = prev;
|
|
sreq.opcode = MXM_REQ_OP_ATOMIC_SWAP;
|
|
|
|
mca_atomic_mxm_post(&sreq);
|
|
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
int mca_atomic_mxm_cswap(shmem_ctx_t ctx,
|
|
void *target,
|
|
uint64_t *prev,
|
|
uint64_t cond,
|
|
uint64_t value,
|
|
size_t nlong,
|
|
int pe)
|
|
{
|
|
mxm_send_req_t sreq;
|
|
|
|
mca_atomic_mxm_req_init(&sreq, pe, target, nlong);
|
|
|
|
*prev = value;
|
|
sreq.op.atomic.value = cond;
|
|
sreq.base.data.buffer.ptr = prev;
|
|
sreq.opcode = MXM_REQ_OP_ATOMIC_CSWAP;
|
|
|
|
mca_atomic_mxm_post(&sreq);
|
|
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|