Samuel Thibault 6 gadi atpakaļ
vecāks
revīzija
d255c70d6b
1 mainītis faili ar 4 papildinājumiem un 2 dzēšanām
  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)++;
 }