浏览代码

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;