Преглед изворни кода

Fix modular-gemm for machines without CPUs

Samuel Thibault пре 6 година
родитељ
комит
be5032d374
1 измењених фајлова са 2 додато и 1 уклоњено
  1. 2 1
      src/sched_policies/modular_gemm.c

+ 2 - 1
src/sched_policies/modular_gemm.c

@@ -140,7 +140,8 @@ static int gemm_push_task(struct starpu_sched_component * component, struct star
 					{
 						if (idworker == workerid)
 						{
-							if (starpu_worker_get_type(workerid) == STARPU_CPU_WORKER
+							if ((starpu_cpu_worker_get_count() == 0 ||
+									starpu_worker_get_type(workerid) == STARPU_CPU_WORKER)
 							 && (starpu_worker_can_execute_task(workerid,task,nimpl)
 							   || starpu_combined_worker_can_execute_task(workerid, task, nimpl)))
 							{