/*
* Wait for termination
*/
void wait_end_tasks(int rank)
{
int bz;
int nbz = get_nbz();
for (bz = 0; bz < nbz; bz++)
{
if (get_block_mpi_node(bz) == rank)
{
/* Wait for the task producing block "bz" */
starpu_tag_wait(TAG_FINISH(bz));
/* Get the result back to memory */
struct block_description *block = get_block_description(bz);
starpu_data_acquire(block->layers_handle[0], STARPU_R);
starpu_data_acquire(block->layers_handle[1], STARPU_R);
/* the data_acquire here is done to make sure
* the data is sent back to the ram memory, we
* can safely do a data_release, to avoid the
* data_unregister to block later on
*/
starpu_data_release(block->layers_handle[0]);
starpu_data_release(block->layers_handle[1]);
}
}
}
void create_task_update(unsigned iter, unsigned z, int local_rank)
{
STARPU_ASSERT(iter != 0);
struct starpu_task *task = starpu_task_create();
unsigned niter = get_niter();
/* We are going to synchronize with the last tasks */
if (iter == niter)
{
task->use_tag = 1;
task->tag_id = TAG_FINISH(z);
}
unsigned old_layer = (K*(iter-1)) % 2;
unsigned new_layer = (old_layer + 1) % 2;
struct block_description *descr = get_block_description(z);
task->handles[0] = descr->layers_handle[new_layer];
task->handles[1] = descr->layers_handle[old_layer];
task->handles[2] = descr->boundaries_handle[T][new_layer];
task->handles[3] = descr->boundaries_handle[T][old_layer];
task->handles[4] = descr->boundaries_handle[B][new_layer];
task->handles[5] = descr->boundaries_handle[B][old_layer];
task->cl = &cl_update;
task->cl_arg = descr;
if (iter <= BIND_LAST)
task->execute_on_a_specific_worker = get_bind_tasks();
task->workerid = descr->preferred_worker;
int ret = starpu_task_submit(task);
if (ret)
{
FPRINTF(stderr, "Could not submit task update block: %d\n", ret);
if (ret == -ENODEV)
exit(77);
STARPU_ABORT();
}
}
