Samuel Thibault 6 years ago
parent
commit
d255c70d6b
1 changed files with 4 additions and 2 deletions
  1. 4 2
      tests/datawizard/specific_node.c

+ 4 - 2
tests/datawizard/specific_node.c

@@ -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)++;
 }