cb1955bb53
Co-authored by: Artem Polykaov <artemp@mellanox.com> Signed-off-by: Tomislav Janjusic <tomislavj@mellanox.com>
127 строки
3.5 KiB
C
127 строки
3.5 KiB
C
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil -*- */
|
|
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2005 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* Copyright (c) 2004-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* Copyright (c) 2019 Sandia National Laboratories. All rights reserved.
|
|
*
|
|
* Copyright (c) 2020 Cisco Systems, Inc. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "opal_config.h"
|
|
|
|
#include <unistd.h>
|
|
|
|
#include "opal/mca/threads/argobots/threads_argobots.h"
|
|
#include "opal/constants.h"
|
|
#include "opal/util/sys_limits.h"
|
|
#include "opal/util/output.h"
|
|
#include "opal/prefetch.h"
|
|
#include "opal/mca/threads/threads.h"
|
|
#include "opal/mca/threads/tsd.h"
|
|
|
|
struct opal_tsd_key_value {
|
|
opal_tsd_key_t key;
|
|
opal_tsd_destructor_t destructor;
|
|
};
|
|
|
|
static opal_mutex_t opal_tsd_lock = OPAL_MUTEX_STATIC_INIT;
|
|
static struct opal_tsd_key_value *opal_tsd_key_values = NULL;
|
|
static int opal_tsd_key_values_count = 0;
|
|
static int opal_tsd_key_values_size = 0;
|
|
|
|
/*
|
|
* Constructor
|
|
*/
|
|
static void opal_thread_construct(opal_thread_t *t)
|
|
{
|
|
t->t_run = 0;
|
|
t->t_handle = ABT_THREAD_NULL;
|
|
}
|
|
|
|
OBJ_CLASS_INSTANCE(opal_thread_t,
|
|
opal_object_t,
|
|
opal_thread_construct, NULL);
|
|
|
|
static inline ABT_thread opal_thread_get_argobots_self(void)
|
|
{
|
|
ABT_thread self;
|
|
ABT_thread_self(&self);
|
|
return self;
|
|
}
|
|
|
|
static void opal_thread_argobots_wrapper(void *arg)
|
|
{
|
|
opal_thread_t *t = (opal_thread_t *)arg;
|
|
t->t_ret = ((void *(*)(void *))t->t_run)(t);
|
|
}
|
|
|
|
opal_thread_t *opal_thread_get_self(void)
|
|
{
|
|
opal_threads_argobots_ensure_init();
|
|
opal_thread_t *t = OBJ_NEW(opal_thread_t);
|
|
t->t_handle = opal_thread_get_argobots_self();
|
|
return t;
|
|
}
|
|
|
|
bool opal_thread_self_compare(opal_thread_t *t)
|
|
{
|
|
opal_threads_argobots_ensure_init();
|
|
return opal_thread_get_argobots_self() == t->t_handle;
|
|
}
|
|
|
|
int opal_thread_join(opal_thread_t *t, void **thr_return)
|
|
{
|
|
int rc = ABT_thread_free(&t->t_handle);
|
|
if (thr_return) {
|
|
*thr_return = t->t_ret;
|
|
}
|
|
t->t_handle = ABT_THREAD_NULL;
|
|
return (ABT_SUCCESS == rc) ? OPAL_SUCCESS : OPAL_ERROR;
|
|
}
|
|
|
|
void opal_thread_set_main()
|
|
{
|
|
}
|
|
|
|
int opal_thread_start(opal_thread_t *t)
|
|
{
|
|
opal_threads_argobots_ensure_init();
|
|
int rc;
|
|
if (OPAL_ENABLE_DEBUG) {
|
|
if (NULL == t->t_run || ABT_THREAD_NULL != t->t_handle) {
|
|
return OPAL_ERR_BAD_PARAM;
|
|
}
|
|
}
|
|
|
|
ABT_xstream self_xstream;
|
|
ABT_xstream_self(&self_xstream);
|
|
rc = ABT_thread_create_on_xstream(self_xstream,
|
|
opal_thread_argobots_wrapper, t,
|
|
ABT_THREAD_ATTR_NULL, &t->t_handle);
|
|
|
|
return (ABT_SUCCESS == rc) ? OPAL_SUCCESS : OPAL_ERROR;
|
|
}
|
|
|
|
OBJ_CLASS_DECLARATION(opal_thread_t);
|
|
|
|
int opal_tsd_key_create(opal_tsd_key_t *key, opal_tsd_destructor_t destructor)
|
|
{
|
|
opal_threads_argobots_ensure_init();
|
|
int rc;
|
|
rc = ABT_key_create(destructor, key);
|
|
return (ABT_SUCCESS == rc) ? OPAL_SUCCESS : OPAL_ERROR;
|
|
}
|