67 строки
1.7 KiB
C
67 строки
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/op/op.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_add(shmem_ctx_t ctx,
|
|
void *target,
|
|
uint64_t value,
|
|
size_t size,
|
|
int pe)
|
|
{
|
|
mxm_send_req_t sreq;
|
|
static char dummy_buf[8];
|
|
|
|
mca_atomic_mxm_req_init(&sreq, pe, target, size);
|
|
|
|
sreq.op.atomic.value = value;
|
|
sreq.opcode = MXM_REQ_OP_ATOMIC_FADD;
|
|
sreq.base.data.buffer.ptr = dummy_buf;
|
|
|
|
mca_atomic_mxm_post(&sreq);
|
|
|
|
return OSHMEM_SUCCESS;
|
|
}
|
|
|
|
int mca_atomic_mxm_fadd(shmem_ctx_t ctx,
|
|
void *target,
|
|
void *prev,
|
|
uint64_t value,
|
|
size_t size,
|
|
int pe)
|
|
{
|
|
mxm_send_req_t sreq;
|
|
|
|
mca_atomic_mxm_req_init(&sreq, pe, target, size);
|
|
|
|
sreq.op.atomic.value = value;
|
|
sreq.opcode = MXM_REQ_OP_ATOMIC_FADD;
|
|
sreq.base.data.buffer.ptr = prev;
|
|
|
|
mca_atomic_mxm_post(&sreq);
|
|
|
|
return OSHMEM_SUCCESS;
|
|
}
|