浏览代码

perf_knobs_03: skip when environment force e.g. cuda devices

Samuel Thibault 5 年之前
父节点
当前提交
0c9013a26e
共有 1 个文件被更改,包括 7 次插入0 次删除
  1. 7 0
      examples/perf_steering/perf_knobs_03.c

+ 7 - 0
examples/perf_steering/perf_knobs_03.c

@@ -56,6 +56,13 @@ int main(int argc, char **argv)
 	ret = starpu_initialize(&conf, &argc, &argv);
 	if (ret == -ENODEV)
 		return 77;
+	if (conf.ncpus != 2
+		|| conf.ncuda != 0
+		|| conf.nopencl != 0
+		|| conf.nmic != 0
+		|| conf.nmpi_ms != 0)
+		/* Environment variables forced otherwise, refuse it */
+		return 77;
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
 
 	unsigned int ncpu  = starpu_cpu_worker_get_count();