self.c

Go to the documentation of this file.

/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
/*
 * See COPYRIGHT in top-level directory.
 */

#include "abti.h"

int ABT_self_get_type(ABT_unit_type *type)
{
    int abt_errno = ABT_SUCCESS;
    /* If Argobots has not been initialized, set type to ABT_UNIT_TYPE_EXIT. */
    if (gp_ABTI_global == NULL) {
        abt_errno = ABT_ERR_UNINITIALIZED;
        *type = ABT_UNIT_TYPE_EXT;
        goto fn_exit;
    }

    ABTI_local *p_local = ABTI_local_get_local();
    *type = ABTI_self_get_type(p_local);
#ifndef ABT_CONFIG_DISABLE_EXT_THREAD
    /* This is when an external thread called this routine. */
    if (*type == ABT_UNIT_TYPE_EXT) {
        abt_errno = ABT_ERR_INV_XSTREAM;
        goto fn_exit;
    }
#endif

fn_exit:
    return abt_errno;
}

int ABT_self_is_primary(ABT_bool *flag)
{
    int abt_errno = ABT_SUCCESS;
    ABTI_local *p_local = ABTI_local_get_local();
    ABTI_thread *p_thread;

    /* If Argobots has not been initialized, set flag to ABT_FALSE. */
    if (gp_ABTI_global == NULL) {
        abt_errno = ABT_ERR_UNINITIALIZED;
        *flag = ABT_FALSE;
        goto fn_exit;
    }

#ifndef ABT_CONFIG_DISABLE_EXT_THREAD
    /* This is when an external thread called this routine. */
    if (p_local == NULL) {
        abt_errno = ABT_ERR_INV_XSTREAM;
        *flag = ABT_FALSE;
        goto fn_exit;
    }
#endif

    p_thread = p_local->p_thread;
    if (p_thread) {
        *flag =
            (p_thread->type == ABTI_THREAD_TYPE_MAIN) ? ABT_TRUE : ABT_FALSE;
    } else {
        abt_errno = ABT_ERR_INV_THREAD;
        *flag = ABT_FALSE;
    }

fn_exit:
    return abt_errno;
}

int ABT_self_on_primary_xstream(ABT_bool *flag)
{
    int abt_errno = ABT_SUCCESS;
    ABTI_local *p_local = ABTI_local_get_local();
    ABTI_xstream *p_xstream;

    /* If Argobots has not been initialized, set flag to ABT_FALSE. */
    if (gp_ABTI_global == NULL) {
        abt_errno = ABT_ERR_UNINITIALIZED;
        *flag = ABT_FALSE;
        goto fn_exit;
    }

#ifndef ABT_CONFIG_DISABLE_EXT_THREAD
    /* This is when an external thread called this routine. */
    if (p_local == NULL) {
        abt_errno = ABT_ERR_INV_XSTREAM;
        *flag = ABT_FALSE;
        goto fn_exit;
    }
#endif

    p_xstream = p_local->p_xstream;
    ABTI_CHECK_NULL_XSTREAM_PTR(p_xstream);

    /* Return value */
    *flag =
        (p_xstream->type == ABTI_XSTREAM_TYPE_PRIMARY) ? ABT_TRUE : ABT_FALSE;

fn_exit:
    return abt_errno;

fn_fail:
    HANDLE_ERROR_FUNC_WITH_CODE(abt_errno);
    goto fn_exit;
}

int ABT_self_get_last_pool_id(int *pool_id)
{
    int abt_errno = ABT_SUCCESS;
    ABTI_local *p_local = ABTI_local_get_local();
    ABTI_thread *p_thread;
    ABTI_task *p_task;

    /* If Argobots has not been initialized, set type to ABT_UNIT_TYPE_EXIT. */
    if (gp_ABTI_global == NULL) {
        abt_errno = ABT_ERR_UNINITIALIZED;
        *pool_id = -1;
        goto fn_exit;
    }

#ifndef ABT_CONFIG_DISABLE_EXT_THREAD
    /* This is when an external thread called this routine. */
    if (p_local == NULL) {
        abt_errno = ABT_ERR_INV_XSTREAM;
        *pool_id = -1;
        goto fn_exit;
    }
#endif

    if ((p_thread = p_local->p_thread)) {
        ABTI_ASSERT(p_thread->p_pool);
        *pool_id = (int)(p_thread->p_pool->id);
    } else if ((p_task = p_local->p_task)) {
        ABTI_ASSERT(p_task->p_pool);
        *pool_id = (int)(p_task->p_pool->id);
    } else {
        abt_errno = ABT_ERR_OTHER;
        *pool_id = -1;
    }

fn_exit:
    return abt_errno;
}

int ABT_self_suspend(void)
{
    int abt_errno = ABT_SUCCESS;
    ABTI_local *p_local = ABTI_local_get_local();
#ifdef ABT_CONFIG_DISABLE_EXT_THREAD
    ABTI_thread *p_thread = p_local->p_thread;
#else
    ABTI_thread *p_thread = NULL;

    /* If this routine is called by non-ULT, just return. */
    if (p_local != NULL) {
        p_thread = p_local->p_thread;
    }
#endif
    if (p_thread == NULL) {
        abt_errno = ABT_ERR_INV_THREAD;
        goto fn_fail;
    }

    abt_errno = ABTI_thread_set_blocked(p_thread);
    ABTI_CHECK_ERROR(abt_errno);

    ABTI_thread_suspend(&p_local, p_thread);

fn_exit:
    return abt_errno;

fn_fail:
    HANDLE_ERROR_FUNC_WITH_CODE(abt_errno);
    goto fn_exit;
}

int ABT_self_set_arg(void *arg)
{
    int abt_errno = ABT_SUCCESS;
    ABTI_local *p_local = ABTI_local_get_local();
    ABTI_thread *p_thread;
    ABTI_task *p_task;

    /* When Argobots has not been initialized */
    if (gp_ABTI_global == NULL) {
        abt_errno = ABT_ERR_UNINITIALIZED;
        goto fn_exit;
    }

    /* When an external thread called this routine */
    if (p_local == NULL) {
        abt_errno = ABT_ERR_INV_XSTREAM;
        goto fn_exit;
    }

    if ((p_thread = p_local->p_thread)) {
        ABTD_thread_context_set_arg(&p_thread->ctx, arg);
    } else if ((p_task = p_local->p_task)) {
        p_task->p_arg = arg;
    } else {
        abt_errno = ABT_ERR_OTHER;
        goto fn_fail;
    }

fn_exit:
    return abt_errno;

fn_fail:
    HANDLE_ERROR_FUNC_WITH_CODE(abt_errno);
    goto fn_exit;
}

int ABT_self_get_arg(void **arg)
{
    int abt_errno = ABT_SUCCESS;
    ABTI_local *p_local = ABTI_local_get_local();
    ABTI_thread *p_thread;
    ABTI_task *p_task;

    /* When Argobots has not been initialized */
    if (gp_ABTI_global == NULL) {
        abt_errno = ABT_ERR_UNINITIALIZED;
        *arg = NULL;
        goto fn_exit;
    }

#ifndef ABT_CONFIG_DISABLE_EXT_THREAD
    /* When an external thread called this routine */
    if (p_local == NULL) {
        abt_errno = ABT_ERR_INV_XSTREAM;
        *arg = NULL;
        goto fn_exit;
    }
#endif

    if ((p_thread = p_local->p_thread)) {
        *arg = ABTD_thread_context_get_arg(&p_thread->ctx);
    } else if ((p_task = p_local->p_task)) {
        *arg = p_task->p_arg;
    } else {
        *arg = NULL;
        abt_errno = ABT_ERR_OTHER;
        goto fn_fail;
    }

fn_exit:
    return abt_errno;

fn_fail:
    HANDLE_ERROR_FUNC_WITH_CODE(abt_errno);
    goto fn_exit;
}