Преглед на файлове

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)))
 							{