|
@@ -159,21 +159,41 @@ static inline void default_init_sched(unsigned sched_ctx_id)
|
|
|
int min_prio = starpu_sched_ctx_get_min_priority(sched_ctx_id);
|
|
|
int max_prio = starpu_sched_ctx_get_max_priority(sched_ctx_id);
|
|
|
// By default each type of devices uses 1 bucket and no slow factor
|
|
|
+#ifdef STARPU_USE_CPU
|
|
|
starpu_heteroprio_set_nb_prios(sched_ctx_id, STARPU_CPU_IDX, max_prio-min_prio+1);
|
|
|
+#endif
|
|
|
+#ifdef STARPU_USE_CUDA
|
|
|
starpu_heteroprio_set_nb_prios(sched_ctx_id, STARPU_CUDA_IDX, max_prio-min_prio+1);
|
|
|
+#endif
|
|
|
+#ifdef STARPU_USE_OPENCL
|
|
|
starpu_heteroprio_set_nb_prios(sched_ctx_id, STARPU_OPENCL_IDX, max_prio-min_prio+1);
|
|
|
+#endif
|
|
|
+#ifdef STARPU_USE_MIC
|
|
|
starpu_heteroprio_set_nb_prios(sched_ctx_id, STARPU_MIC_IDX, max_prio-min_prio+1);
|
|
|
+#endif
|
|
|
+#ifdef STARPU_USE_SCC
|
|
|
starpu_heteroprio_set_nb_prios(sched_ctx_id, STARPU_SCC_IDX, max_prio-min_prio+1);
|
|
|
+#endif
|
|
|
|
|
|
// Direct mapping
|
|
|
int prio;
|
|
|
for(prio=min_prio ; prio<=max_prio ; prio++)
|
|
|
{
|
|
|
+#ifdef STARPU_USE_CPU
|
|
|
starpu_heteroprio_set_mapping(sched_ctx_id, STARPU_CPU_IDX, prio, prio);
|
|
|
+#endif
|
|
|
+#ifdef STARPU_USE_CUDA
|
|
|
starpu_heteroprio_set_mapping(sched_ctx_id, STARPU_CUDA_IDX, prio, prio);
|
|
|
+#endif
|
|
|
+#ifdef STARPU_USE_OPENCL
|
|
|
starpu_heteroprio_set_mapping(sched_ctx_id, STARPU_OPENCL_IDX, prio, prio);
|
|
|
+#endif
|
|
|
+#ifdef STARPU_USE_MIC
|
|
|
starpu_heteroprio_set_mapping(sched_ctx_id, STARPU_MIC_IDX, prio, prio);
|
|
|
+#endif
|
|
|
+#ifdef STARPU_USE_SCC
|
|
|
starpu_heteroprio_set_mapping(sched_ctx_id, STARPU_SCC_IDX, prio, prio);
|
|
|
+#endif
|
|
|
}
|
|
|
}
|
|
|
|