|
@@ -35,10 +35,12 @@ unsigned data, data2;
|
|
|
void specific_kernel(void *descr[], void *arg)
|
|
|
{
|
|
|
(void)arg;
|
|
|
- STARPU_ASSERT(starpu_node_get_kind(starpu_task_get_current_data_node(0)) == STARPU_CPU_RAM);
|
|
|
+ int node = starpu_task_get_current_data_node(0);
|
|
|
+ STARPU_ASSERT(starpu_node_get_kind(node) == STARPU_CPU_RAM);
|
|
|
unsigned *dataptr = (unsigned*) STARPU_VARIABLE_GET_PTR(descr[0]);
|
|
|
|
|
|
- STARPU_ASSERT(dataptr == &data);
|
|
|
+ if (node == STARPU_MAIN_RAM)
|
|
|
+ STARPU_ASSERT(dataptr == &data);
|
|
|
(*dataptr)++;
|
|
|
}
|
|
|
|