ソースを参照

Skip starpupy tests with the graph_test scheduler

It needs needs starpu_do_schedule calls.
Samuel Thibault 4 年 前
コミット
99ae8e2d95
共有1 個のファイルを変更した11 個の追加1 個の削除を含む
  1. 11 1
      starpupy/src/starpu_task_wrapper.c

+ 11 - 1
starpupy/src/starpu_task_wrapper.c

@@ -743,8 +743,11 @@ PyInit_starpupy(void)
 #endif
 	/*starpu initialization*/
 	int ret;
+	struct starpu_conf conf;
+	starpu_conf_init(&conf);
+
 	Py_BEGIN_ALLOW_THREADS
-	ret = starpu_init(NULL);
+	ret = starpu_init(&conf);
 	Py_END_ALLOW_THREADS
 	if (ret!=0)
 	{
@@ -752,6 +755,13 @@ PyInit_starpupy(void)
 		return NULL;
 	}
 
+	if (conf.sched_policy_name && !strcmp(conf.sched_policy_name, "graph_test"))
+	{
+		/* FIXME: should call starpu_do_schedule when appropriate, the graph_test scheduler needs it. */
+		fprintf(stderr,"TODO: The graph_test scheduler needs starpu_do_schedule calls\n");
+		exit(77);
+	}
+
 	/*python asysncio import*/
 	asyncio_module = PyImport_ImportModule("asyncio");