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

Always reset the local worker key to worker0 for datawizard to get the proper wait queue when using multiple workers in the same driver thread

Samuel Thibault преди 8 години
родител
ревизия
5a5eb593d1
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. 1 0
      src/drivers/cuda/driver_cuda.c

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

@@ -727,6 +727,7 @@ int _starpu_cuda_driver_run_once(struct _starpu_worker_set *worker_set)
 #ifdef STARPU_SIMGRID
 	starpu_pthread_wait_reset(&worker0->wait);
 #endif
+	_starpu_set_local_worker_key(worker0);
 
 	/* First poll for completed jobs */
 	idle_tasks = 0;