/* * Copyright (c) 2015 Mellanox Technologies, Inc. * All rights reserved. * $COPYRIGHT$ * * Additional copyrights may follow * * $HEADER$ */ #include "oshmem_config.h" #include #include "oshmem/constants.h" #include "oshmem/mca/atomic/atomic.h" #include "oshmem/mca/spml/spml.h" #include "oshmem/mca/memheap/memheap.h" #include "oshmem/proc/proc.h" #include "atomic_ucx.h" /* * Initial query function that is invoked during initialization, allowing * this module to indicate what level of thread support it provides. */ int mca_atomic_ucx_init(bool enable_progress_threads, bool enable_threads) { return OSHMEM_SUCCESS; } int mca_atomic_ucx_finalize(void) { return OSHMEM_SUCCESS; } mca_atomic_base_module_t * mca_atomic_ucx_query(int *priority) { mca_atomic_ucx_module_t *module; *priority = mca_atomic_ucx_component.priority; module = OBJ_NEW(mca_atomic_ucx_module_t); if (module) { module->super.atomic_fadd = mca_atomic_ucx_fadd; module->super.atomic_cswap = mca_atomic_ucx_cswap; return &(module->super); } return NULL ; }