Browse Source

don't omit setting the local worker key, to make sure _starpu_fetch_task_input_tail works with the proper device

Samuel Thibault 8 years ago
parent
commit
c14ba419e6
1 changed files with 1 additions and 0 deletions
  1. 1 0
      src/drivers/opencl/driver_opencl.c

+ 1 - 0
src/drivers/opencl/driver_opencl.c

@@ -691,6 +691,7 @@ int _starpu_opencl_driver_run_once(struct _starpu_worker *worker)
 		_STARPU_TRACE_END_PROGRESS(memnode);
 		j = _starpu_get_job_associated_to_task(task);
 
+		_starpu_set_local_worker_key(worker);
 		_starpu_fetch_task_input_tail(task, j, worker);
 		/* Reset it */
 		worker->task_transferring = NULL;