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

Fix execution with CUDA_PIPELINE=0

Samuel Thibault пре 8 година
родитељ
комит
6b3556b030
1 измењених фајлова са 4 додато и 1 уклоњено
  1. 4 1
      src/drivers/cuda/driver_cuda.c

+ 4 - 1
src/drivers/cuda/driver_cuda.c

@@ -801,7 +801,10 @@ int _starpu_cuda_driver_run_once(struct _starpu_worker_set *worker_set)
 			/* No queued task */
 			continue;
 
-		task = worker->current_tasks[worker->first_task];
+		if (worker->pipeline_length)
+			task = worker->current_tasks[worker->first_task];
+		else
+			task = worker->current_task;
 		if (task == worker->task_transferring)
 			/* Next task is still pending transfer */
 			continue;