int main(int argc, char **argv)
{
starpu_init(NULL);
starpu_data_malloc_pinned_if_possible((void **)&v, VECTORSIZE*sizeof(unsigned));
starpu_vector_data_register(&v_handle, 0, (uintptr_t)v, VECTORSIZE, sizeof(unsigned));
unsigned nworker = starpu_worker_get_count();
cnt = nworker*N;
unsigned iter, worker;
for (iter = 0; iter < N; iter++)
{
for (worker = 0; worker < nworker; worker++)
{
/* synchronous prefetch */
unsigned node = starpu_worker_get_memory_node(worker);
starpu_data_prefetch_on_node(v_handle, node, 0);
/* execute a task */
struct starpu_task *task = starpu_task_create();
task->cl = &cl;
task->buffers[0].handle = v_handle;
task->buffers[0].mode = select_random_mode();
task->callback_func = callback;
task->callback_arg = NULL;
task->synchronous = 1;
int ret = starpu_task_submit(task, NULL);
if (ret == -ENODEV)
goto enodev;
}
}
pthread_mutex_lock(&mutex);
if (!finished)
pthread_cond_wait(&cond, &mutex);
pthread_mutex_unlock(&mutex);
starpu_shutdown();
return 0;
enodev:
fprintf(stderr, "WARNING: No one can execute this task\n");
/* yes, we do not perform the computation but we did detect that no one
* could perform the kernel, so this is not an error from StarPU */
return 0;
}
int main(int argc, char **argv)
{
unsigned *foo;
starpu_data_handle_t handle;
int ret;
unsigned n, i, size;
ret = starpu_initialize(NULL, &argc, &argv);
if (ret == -ENODEV) return STARPU_TEST_SKIPPED;
STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
#ifdef STARPU_USE_OPENCL
ret = starpu_opencl_load_opencl_from_file("tests/datawizard/scal_opencl.cl", &opencl_program, NULL);
STARPU_CHECK_RETURN_VALUE(ret, "starpu_opencl_load_opencl_from_file");
#endif
n = starpu_worker_get_count();
if (n == 1)
{
starpu_shutdown();
return STARPU_TEST_SKIPPED;
}
size = 10 * n;
foo = (unsigned *) calloc(size, sizeof(*foo));
for (i = 0; i < size; i++)
foo[i] = i;
starpu_vector_data_register(&handle, STARPU_MAIN_RAM, (uintptr_t)foo, size, sizeof(*foo));
/* Broadcast the data to force in-place partitioning */
for (i = 0; i < n; i++)
starpu_data_prefetch_on_node(handle, starpu_worker_get_memory_node(i), 0);
struct starpu_data_filter f =
{
.filter_func = starpu_vector_filter_block,
.nchildren = n,
};
starpu_data_partition(handle, &f);
for (i = 0; i < f.nchildren; i++)
{
struct starpu_task *task = starpu_task_create();
task->handles[0] = starpu_data_get_sub_data(handle, 1, i);
task->cl = &scal_codelet;
task->execute_on_a_specific_worker = 1;
task->workerid = i;
ret = starpu_task_submit(task);
if (ret == -ENODEV) goto enodev;
STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
}
ret = starpu_task_wait_for_all();
STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_wait_for_all");
starpu_data_unpartition(handle, STARPU_MAIN_RAM);
starpu_data_unregister(handle);
starpu_shutdown();
ret = EXIT_SUCCESS;
for (i = 0; i < size; i++)
{
if (foo[i] != i*2)
{
FPRINTF(stderr,"value %u is %u instead of %u\n", i, foo[i], 2*i);
ret = EXIT_FAILURE;
}
}
return ret;
enodev:
starpu_data_unregister(handle);
fprintf(stderr, "WARNING: No one can execute this task\n");
/* yes, we do not perform the computation but we did detect that no one
* could perform the kernel, so this is not an error from StarPU */
starpu_shutdown();
return STARPU_TEST_SKIPPED;
}
