mtapi_boolean_t embb_mtapi_spinlock_acquire(embb_mtapi_spinlock_t * that) {
int expected = 0;
while (0 == embb_atomic_compare_and_swap_int(that, &expected, 1)) {
/* empty */
embb_atomic_fetch_and_add_int(&embb_mtapi_spinlock_spins, 1);
expected = 0;
}
return MTAPI_TRUE;
}
mtapi_boolean_t embb_mtapi_spinlock_acquire_with_spincount(
embb_mtapi_spinlock_t * that,
mtapi_uint_t max_spin_count) {
int expected = 0;
mtapi_uint_t spin_count = max_spin_count;
while (0 == embb_atomic_compare_and_swap_int(that, &expected, 1)) {
embb_atomic_fetch_and_add_int(&embb_mtapi_spinlock_spins, 1);
spin_count--;
if (0 == spin_count) {
return MTAPI_FALSE;
}
expected = 0;
}
return MTAPI_TRUE;
}
static void CL_API_CALL opencl_task_complete(
cl_event ev, cl_int status, void * data) {
EMBB_UNUSED(ev);
EMBB_UNUSED(status);
cl_int err;
EMBB_UNUSED_IN_RELEASE(err);
embb_mtapi_opencl_task_t * opencl_task = (embb_mtapi_opencl_task_t*)data;
if (embb_mtapi_node_is_initialized()) {
embb_mtapi_node_t * node = embb_mtapi_node_get_instance();
if (embb_mtapi_task_pool_is_handle_valid(
node->task_pool, opencl_task->task)) {
embb_mtapi_task_t * local_task =
embb_mtapi_task_pool_get_storage_for_handle(
node->task_pool, opencl_task->task);
err = clWaitForEvents(1, &opencl_task->kernel_finish_event);
assert(CL_SUCCESS == err);
if (NULL != opencl_task->result_buffer) {
err = clReleaseMemObject(opencl_task->result_buffer);
assert(CL_SUCCESS == err);
}
if (NULL != opencl_task->arguments) {
err = clReleaseMemObject(opencl_task->arguments);
assert(CL_SUCCESS == err);
}
if (embb_mtapi_action_pool_is_handle_valid(
node->action_pool, local_task->action)) {
embb_mtapi_action_t * local_action =
embb_mtapi_action_pool_get_storage_for_handle(
node->action_pool, local_task->action);
embb_atomic_fetch_and_add_int(&local_action->num_tasks,
-(int)local_task->attributes.num_instances);
}
embb_mtapi_task_set_state(local_task, MTAPI_TASK_COMPLETED);
}
}
}
void mtapi_group_wait_any(
MTAPI_IN mtapi_group_hndl_t group,
MTAPI_OUT void** result,
MTAPI_IN mtapi_timeout_t timeout,
MTAPI_OUT mtapi_status_t* status) {
mtapi_status_t local_status = MTAPI_ERR_UNKNOWN;
embb_mtapi_log_trace("mtapi_group_wait_any() called\n");
if (embb_mtapi_node_is_initialized()) {
embb_mtapi_node_t* node = embb_mtapi_node_get_instance();
if (embb_mtapi_group_pool_is_handle_valid(node->group_pool, group)) {
embb_mtapi_group_t* local_group =
embb_mtapi_group_pool_get_storage_for_handle(
node->group_pool, group);
embb_mtapi_task_t* local_task;
/* are there any tasks left? */
if (0 == embb_atomic_load_int(&local_group->num_tasks)) {
/* group becomes invalid, so delete it */
mtapi_group_delete(group, &local_status);
local_status = MTAPI_GROUP_COMPLETED;
} else {
embb_mtapi_thread_context_t * context = NULL;
embb_duration_t wait_duration;
embb_time_t end_time;
if (MTAPI_INFINITE < timeout) {
embb_duration_set_milliseconds(
&wait_duration, (unsigned long long)timeout);
embb_time_in(&end_time, &wait_duration);
}
/* find out on which thread we are */
context = embb_mtapi_scheduler_get_current_thread_context(
node->scheduler);
/* wait for any task to arrive */
local_status = MTAPI_SUCCESS;
local_task = embb_mtapi_task_queue_pop(&local_group->queue);
while (MTAPI_NULL == local_task) {
if (MTAPI_INFINITE < timeout) {
embb_time_t current_time;
embb_time_now(¤t_time);
if (embb_time_compare(¤t_time, &end_time) > 0) {
/* timeout! */
local_status = MTAPI_TIMEOUT;
break;
}
}
/* do other work if applicable */
embb_mtapi_scheduler_execute_task_or_yield(
node->scheduler,
node,
context);
/* try to pop a task from the group queue */
local_task = embb_mtapi_task_queue_pop(&local_group->queue);
}
/* was there a timeout, or is there a result? */
if (MTAPI_NULL != local_task) {
/* store result */
if (MTAPI_NULL != result) {
*result = local_task->result_buffer;
}
/* return error code set by the task */
local_status = local_task->error_code;
/* delete task */
embb_mtapi_task_delete(local_task, node->task_pool);
embb_atomic_fetch_and_add_int(&local_group->num_tasks, -1);
}
}
} else {
local_status = MTAPI_ERR_GROUP_INVALID;
}
} else {
embb_mtapi_log_error("mtapi not initialized\n");
local_status = MTAPI_ERR_NODE_NOTINIT;
}
mtapi_status_set(status, local_status);
embb_mtapi_log_trace("mtapi_group_wait_any() returns\n");
}
