Browse Source

Merge branch 'fpga' of gitlab.inria.fr:starpu/starpu into fpga

Samuel Thibault 3 years ago
parent
commit
d9f0d943eb
1 changed files with 9 additions and 0 deletions
  1. 9 0
      src/core/workers.c

+ 9 - 0
src/core/workers.c

@@ -294,6 +294,10 @@ static uint32_t _starpu_worker_exists_and_can_execute(struct starpu_task *task,
 				if (task->cl->opencl_funcs[impl] != NULL)
 					test_implementation = 1;
 				break;
+			case STARPU_FPGA_WORKER:
+				if (task->cl->fpga_funcs[impl] != NULL)
+					test_implementation = 1;
+				break;
                         case STARPU_MPI_MS_WORKER:
                                 if (task->cl->cpu_funcs_name[impl] != NULL || task->cl->mpi_ms_funcs[impl] != NULL)
                                         test_implementation = 1;
@@ -352,6 +356,11 @@ uint32_t _starpu_worker_exists(struct starpu_task *task)
 	    _starpu_worker_exists_and_can_execute(task, STARPU_OPENCL_WORKER))
 		return 1;
 #endif
+#if defined(STARPU_USE_FPGA)
+	if ((task->where & STARPU_FPGA) &&
+	    _starpu_worker_exists_and_can_execute(task, STARPU_FPGA_WORKER))
+		return 1;
+#endif
 #ifdef STARPU_USE_MPI_MASTER_SLAVE
 	if ((task->where & STARPU_MPI_MS) &&
 	    _starpu_worker_exists_and_can_execute(task, STARPU_MPI_MS_WORKER))