@@ -54,7 +54,7 @@ static void test_lazy_allocation()
starpu_insert_task(&cl,
STARPU_W, handle,
- STARPU_VALUE, &count, sizeof(float),
+ STARPU_VALUE, &count, sizeof(size_t),
0);
/* Acquire the handle, forcing a local allocation. */
@@ -34,8 +34,9 @@ static void task(void **buffers, void *args)
}
static starpu_codelet cl = {
- .where = STARPU_CPU,
+ .where = STARPU_CPU | STARPU_CUDA,
.cpu_func = task,
+ .cuda_func = task,
.nbuffers = 1
};