Samuel Thibault 6 年之前
父节点
当前提交
d255c70d6b
共有 1 个文件被更改,包括 4 次插入2 次删除
  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 specific_kernel(void *descr[], void *arg)
 {
 {
 	(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]);
 	unsigned *dataptr = (unsigned*) STARPU_VARIABLE_GET_PTR(descr[0]);
 
 
-	STARPU_ASSERT(dataptr == &data);
+	if (node == STARPU_MAIN_RAM)
+		STARPU_ASSERT(dataptr == &data);
 	(*dataptr)++;
 	(*dataptr)++;
 }
 }